MedSymmFlow: Bridging Generative Modeling and Classification in Medical Imaging through Symmetrical Flow Matching
By: Francisco Caetano , Lemar Abdi , Christiaan Viviers and more
Potential Business Impact:
Helps doctors see how sure they are about medical pictures.
Reliable medical image classification requires accurate predictions and well-calibrated uncertainty estimates, especially in high-stakes clinical settings. This work presents MedSymmFlow, a generative-discriminative hybrid model built on Symmetrical Flow Matching, designed to unify classification, generation, and uncertainty quantification in medical imaging. MedSymmFlow leverages a latent-space formulation that scales to high-resolution inputs and introduces a semantic mask conditioning mechanism to enhance diagnostic relevance. Unlike standard discriminative models, it naturally estimates uncertainty through its generative sampling process. The model is evaluated on four MedMNIST datasets, covering a range of modalities and pathologies. The results show that MedSymmFlow matches or exceeds the performance of established baselines in classification accuracy and AUC, while also delivering reliable uncertainty estimates validated by performance improvements under selective prediction.
Similar Papers
Symmetrical Flow Matching: Unified Image Generation, Segmentation, and Classification with Score-Based Generative Models
CV and Pattern Recognition
Makes one AI do many picture tasks at once.
Aleatoric Uncertainty Medical Image Segmentation Estimation via Flow Matching
CV and Pattern Recognition
Shows where doctors disagree on scans.
Flow Matching for Medical Image Synthesis: Bridging the Gap Between Speed and Quality
CV and Pattern Recognition
Makes AI create better medical pictures faster.