Project-MONAI / Project-MONAI/MONAI

Feat Request: Conformal prediction

Open
#8,935 1 comment 0 reactions 0 assignees View on GitHub

Nobody has claimed this yet.

Dominant language
Python
Stars
8.7k
Forks
1.6k
Avg merge
5d 1h
Merged PRs (30d)
20

Description

Is your feature request related to a problem? Please describe.

I want to be able to give a prediction set with a coverage guarantee beyond the current calibration and TTA.

Describe the solution you'd like
-- Collect softmax on a held out calibration, compute non conformity scores (LAC:1-softmax), take the empirical quintile as threshold and at inference return score(y) <= threshold.
-- Then calibrate a single threshold that bounds an image level loss, put a per voxel uncertainty mask

Potentially:
inferers/conformal_predictor.py
metrics/conformal_risk.py

Describe alternatives you've considered
External libraries like TorchCP

Additional context
https://arxiv.org/pdf/2208.02814

Contributor guide

Open the contributing guide

First steps

  1. Read the whole issue, then the project's contributing guide.
  2. Comment on the issue to say you are picking it up — it saves two people doing the same work.
  3. Fork the repository and make your change on a branch.
  4. Open a pull request that references the issue number.

Research direction

Start by reading the linked conformal prediction paper and reviewing the proposed inferers/conformal_predictor.py and metrics/conformal_risk.py entry points alongside MONAI's existing calibration and TTA behavior. The work is complete when calibrated prediction sets and the requested image-level or per-voxel uncertainty behavior are defined and integrated with verifiable coverage guarantees.

Written by the indexing model from the issue text.

Assessment

Tech stack
python, pytorch
Domain
computer-vision, machine-learning
Issue type
Feature
Difficulty
5/5
Estimated time
Over a week
Activity status
Quiet
Clarity
Mostly clear
Newbie friendliness
38/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.