7 papers
Segmentor-Guided Counterfactual Fine-Tuning for Locally Coherent and Targeted Image Synthesis
Tian Xia, Matthew Sinclair, Andreas Schuh +8
Counterfactual image generation is a powerful tool for augmenting training data, de-biasing datasets, and modeling disease. Current approaches rely on external classifiers or regre…
Flow Stochastic Segmentation Networks
Fabio De Sousa Ribeiro, Omar Todd, Charles Jones +3
We introduce the Flow Stochastic Segmentation Network (Flow-SSN), a generative segmentation model family featuring discrete-time autoregressive and modern continuous-time flow vari…
Exploring the interplay of label bias with subgroup size and separability: A case study in mammographic density classification
Emma A. M. Stanley, Raghav Mehta, Mélanie Roschewitz +2
Systematic mislabelling affecting specific subgroups (i.e., label bias) in medical imaging datasets represents an understudied issue concerning the fairness of medical AI systems.…
UNSURF: Uncertainty Quantification for Cortical Surface Reconstruction of Clinical Brain MRIs
Raghav Mehta, Karthik Gopinath, Ben Glocker +1
We propose UNSURF, a novel uncertainty measure for cortical surface reconstruction of clinical brain MRI scans of any orientation, resolution, and contrast. It relies on the discre…
Where are we with calibration under dataset shift in image classification?
Mélanie Roschewitz, Raghav Mehta, Fabio de Sousa Ribeiro +1
We conduct an extensive study on the state of calibration under real-world dataset shift for image classification. Our work provides important insights on the choice of post-hoc an…
CF-Seg: Counterfactuals meet Segmentation
Raghav Mehta, Fabio De Sousa Ribeiro, Tian Xia +4
Segmenting anatomical structures in medical images plays an important role in the quantitative assessment of various diseases. However, accurate segmentation becomes significantly…