Abstract
This article introduces an enhanced teacher-student model featuring a novel Vnet architecture that integrates high-pass and low-pass filters to improve the segmentation of breast MRI images. The model effectively utilizes finely annotated, roughly annotated, and unannotated data to achieve precise breast tissue density segmentation. The teacher-student framework incorporates three specialized Vnet networks, each tailored to different types of annotations. By integrating cosine contrast loss functions between finely and roughly annotated models, and innovatively applying high-pass and low-pass filters within the Vnet architecture, the segmentation performance is significantly enhanced. This hybrid filtering approach enables the model to capture both fine-grained and coarse structural details, leading to more accurate segmentation across various MRI image datasets. Experimental results demonstrate the superiority of the proposed method, achieving Dice values of 0.833 on the finely annotated Shenzhen dataset and 0.780 on the Duke dataset, using 15 finely annotated, 15 roughly annotated, and 58 unlabeled samples provided by Shenzhen People's Hospital. These findings underscore its potential clinical application in breast density assessment.
Original language | English |
---|---|
Article number | 10750144 |
Pages (from-to) | 690-699 |
Number of pages | 10 |
Journal | IEEE Transactions on Artificial Intelligence |
Volume | 6 |
Issue number | 3 |
Early online date | 11 Nov 2024 |
DOIs | |
Publication status | Published - Mar 2025 |
Bibliographical note
Publisher Copyright:© 2024 IEEE.
Funding
This work is supported by Macao Polytechnic University Grant (RP/FCA-15/2022), (RP/FCSD-01/2022) and Technology Development Fund of Macau SAR (Grant number 0105/2022/A). Tianyu Xie, Yue Sun, Jinhong Song, Qimin Yang and Tao Tan are with the Faculty of Applied Sciences, Macao Polytechnic University, Macao, China. Yue Sun and Hongxu Yang are with the Eindhoven University of Technology, Eindhoven, the Netherlands. Shuo Li is with the Department of Computer and Data Science and Department of Biomedical Engineering, Case Western Reserve University, Cleveland, USA. Hao Chen is with Jiangsu JITRI Sioux Technologies Co., Ltd., Suzhou, China and the Department of Biomechanical Engineering, University of Twente, Enschede, Netherlands. *Corresponding author: Mingxiang Wu, Hao Chen, and Tao Tan.
Keywords
- Breast MRI
- Contrastive Loss
- Density Segmentation
- Hybrid Filtering
- Teacher-Student Model
- Vnet Networks
- density segmentation
- contrastive loss
- hybrid filtering
- Vnet networks
- teacher-student model