Search papers, labs, and topics across Lattice.
MedFlowSeg is introduced as a conditional flow matching framework for medical image segmentation, framing the task as learning a time-dependent vector field to transport a simple prior to the target segmentation. The framework incorporates a Dual-Branch Spatial Attention module for multi-scale structural information and a Frequency-Aware Attention module that models cross-domain interactions between spatial and spectral representations. Experiments across multiple medical imaging modalities demonstrate state-of-the-art performance with reduced computational cost compared to diffusion-based methods.
Ditch slow diffusion models for medical image segmentation: MedFlowSeg achieves state-of-the-art accuracy with a computationally efficient, one-step deterministic approach.
Flow matching has recently emerged as a principled framework for learning continuous-time transport maps, enabling efficient deterministic generation without relying on stochastic diffusion processes. While generative modeling has shown promise for medical image segmentation, particularly in capturing uncertainty and complex anatomical variability, existing approaches are predominantly built upon diffusion models, which incur substantial computational overhead due to iterative sampling and are often constrained by UNet-based parameterizations. In this work, we introduce MedFlowSeg, a conditional flow matching framework that formulates medical image segmentation as learning a time-dependent vector field that transports a simple prior distribution to the target segmentation distribution. This formulation enables one-step deterministic inference while preserving the expressiveness of generative modeling. We further develop a dual-conditioning mechanism to incorporate structured priors into the learned flow. Specifically, we propose a Dual-Branch Spatial Attention module that injects multi-scale structural information into the flow field, and a Frequency-Aware Attention module that models cross-domain interactions between spatial and spectral representations via discrepancy-aware fusion and time-dependent modulation. Together, these components provide an effective parameterization of conditional flows that capture both global anatomical structure and fine-grained boundary details. We provide extensive empirical validation across multiple medical imaging modalities, demonstrating that MedFlowSeg achieves state-of-the-art performance while significantly reducing computational cost compared to diffusion-based methods. Our results highlight the potential of flow matching as a theoretically grounded and computationally efficient alternative for generative medical image segmentation.
Mila