Ambiguous Medical Image Segmentation Using Diffusion Schrödinger Bridge
By: Lalith Bharadwaj Baru , Kamalaker Dadi , Tapabrata Chakraborti and more
Potential Business Impact:
Helps doctors see tiny sickness spots better.
Accurate segmentation of medical images is challenging due to unclear lesion boundaries and mask variability. We introduce \emph{Segmentation Sch\"{o}dinger Bridge (SSB)}, the first application of Sch\"{o}dinger Bridge for ambiguous medical image segmentation, modelling joint image-mask dynamics to enhance performance. SSB preserves structural integrity, delineates unclear boundaries without additional guidance, and maintains diversity using a novel loss function. We further propose the \emph{Diversity Divergence Index} ($D_{DDI}$) to quantify inter-rater variability, capturing both diversity and consensus. SSB achieves state-of-the-art performance on LIDC-IDRI, COCA, and RACER (in-house) datasets.
Similar Papers
MixBridge: Heterogeneous Image-to-Image Backdoor Attack through Mixture of Schrödinger Bridges
Cryptography and Security
Makes AI image generators create hidden bad pictures.
Dynamic Diffusion Schrödinger Bridge in Astrophysical Observational Inversions
Instrumentation and Methods for Astrophysics
Helps predict how stars form in space.
Dequantified Diffusion-Schr{ö}dinger Bridge for Density Ratio Estimation
Machine Learning (CS)
Helps computers understand different data better.