TY - JOUR
T1 - Semi Supervised Breast MRI Density Segmentation Integrating Fine and Rough Annotations
AU - Xie, Tianyu
AU - Sun, Yue
AU - Yang, Hongxu
AU - Li, Shuo
AU - Song, Jinhong
AU - Yang, Qimin
AU - Chen, Hao
AU - Wu, Mingxiang
AU - Tan, Tao
N1 - Publisher Copyright:
© 2024 IEEE.
PY - 2024
Y1 - 2024
N2 - 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.
AB - 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.
KW - Breast MRI
KW - Contrastive Loss
KW - Density Segmentation
KW - Hybrid Filtering
KW - Teacher-Student Model
KW - Vnet Networks
UR - http://www.scopus.com/inward/record.url?scp=85209688943&partnerID=8YFLogxK
U2 - 10.1109/TAI.2024.3491693
DO - 10.1109/TAI.2024.3491693
M3 - Article
AN - SCOPUS:85209688943
SN - 2691-4581
JO - IEEE Transactions on Artificial Intelligence
JF - IEEE Transactions on Artificial Intelligence
ER -