A Fast and Efficient Modern BERT based Text-Conditioned Diffusion Model for Medical Image Segmentation
This work addresses the need for cost-effective segmentation in medical imaging, though it is incremental as it builds on existing diffusion and transformer methods.
The paper tackled the problem of expensive pixel-wise labels in medical image segmentation by proposing FastTextDiff, a label-efficient diffusion-based model that integrates medical text annotations, achieving improved segmentation accuracy and training efficiency over traditional diffusion models.
In recent times, denoising diffusion probabilistic models (DPMs) have proven effective for medical image generation and denoising, and as representation learners for downstream segmentation. However, segmentation performance is limited by the need for dense pixel-wise labels, which are expensive, time-consuming, and require expert knowledge. We propose FastTextDiff, a label-efficient diffusion-based segmentation model that integrates medical text annotations to enhance semantic representations. Our approach uses ModernBERT, a transformer capable of processing long clinical notes, to tightly link textual annotations with semantic content in medical images. Trained on MIMIC-III and MIMIC-IV, ModernBERT encodes clinical knowledge that guides cross-modal attention between visual and textual features. This study validates ModernBERT as a fast, scalable alternative to Clinical BioBERT in diffusion-based segmentation pipelines and highlights the promise of multi-modal techniques for medical image analysis. By replacing Clinical BioBERT with ModernBERT, FastTextDiff benefits from FlashAttention 2, an alternating attention mechanism, and a 2-trillion-token corpus, improving both segmentation accuracy and training efficiency over traditional diffusion-based models.