Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

MedSymmFlow: Bridging Generative Modeling and Classification in Medical Imaging through Symmetrical Flow Matching

About

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.

Francisco Caetano, Lemar Abdi, Christiaan Viviers, Amaan Valiuddin, Fons van der Sommen• 2025

Related benchmarks

TaskDatasetResultRank
ClassificationPneumoniaMNIST
Accuracy89.4
84
ClassificationRetinaMNIST
ACC54
46
8-class classificationBloodMNIST
Accuracy99
33
Medical Image ClassificationDermaMNIST
AUC92.5
31
Showing 4 of 4 rows

Other info

Follow for update