A diffusion probabilistic model with multi-scale conditional fusion for enhanced medical image segmentation
摘要
This study introduces a novel diffusion probabilistic model, called Residual Attention Medical Diffusion Model (ResAtMedDiff), designed to enhance the segmentation of medical images. Our primary contribution is a novel multi-scale conditional fusion mechanism. This is achieved through a dual-network architecture comprising a dynamic encoder, Multi-scale Enhanced U-Net Encoder (MEUNet), which extracts hierarchical anatomical features from the source image, and a robust codec, Residual-attention U-Net (RsAtUNet). These multi-scale features are then integrated into the diffusion model’s denoising process to provide fine-grained guidance, significantly improving feature representation and segmentation precision over existing methods. ResAtMedDiff demonstrates significant improvements in segmentation accuracy, achieving a 6.5% increase in the Dice similarity coefficient and an 11.9% improvement in the Intersection over Union metric compared to state-of-the-art methods. The application of this model is validated for invasive breast cancer segmentation in magnetic resonance imaging (MRI). The results highlight the potential of this model to enhance diagnostic accuracy, offering practical advancements for clinical applications. The code is available for open source at https://github.com/xiaopeng113/ResAtMedDiff.