Research on Cardiac Medical Image Segmentation Based on an Improved U-Net Model
This paper presents a cardiac MRI image segmentation model based on an improved U-Net architecture. Accurate segmentation of cardiac MRI images is critical for the diagnosis and treatment of cardiovascular diseases, yet existing U-Net models exhibit limitations in handling complex cardiac structures and multi-scale features. To address these challenges, this paper proposes two key enhancements. First, we propose a Multi-Dimensional Context Attention module, designed to improve the integration of global and local information within the skip connections, thereby enhancing segmentation accuracy for intricate cardiac structures. Furthermore, we propose a Reverse Feature Modulation module, which generates reverse masks and dynamically adjusts feature weights across different classes using adaptive weighting, effectively mitigating class imbalance issues in multi-class segmentation tasks and improving focus on difficult-to-segment regions. Experimental results demonstrate that the proposed model outperforms the standard U-Net on the ACDC dataset, achieving significant improvements in evaluation metrics such as the Dice coefficient. These enhancements underscore the model's efficacy and robustness in complex cardiac image segmentation tasks, offering new technical support for automated cardiovascular disease diagnosis.
Authors
- Yuqing Wu (ORCID: https://orcid.org/0000-0003-4883-5982)
- Kecheng Zhou
- Jian Wang (ORCID: https://orcid.org/0000-0002-9482-5227)
- Rui Wang (ORCID: https://orcid.org/0000-0003-3350-4312)
- Jun Wang (ORCID: https://orcid.org/0000-0001-9548-0411)
- Xiaoxin Zhang (ORCID: https://orcid.org/0000-0003-4299-7059)
Institutions
- Chinese Academy of Sciences (CN)
- Shenyang University of Chemical Technology (CN)
Publication Details
- Published
- 2024-11-22
- DOI
- https://doi.org/10.1109/icicml63543.2024.10957994
- Citations
- 1
- Primary Topic
- Medical Imaging and Analysis
- Type
- article
- Field-Weighted Citation Impact
- 0.18