Semi Supervised Breast MRI Density Segmentation Integrating Fine and Rough Annotations

Tianyu Xie, Yue Sun, Hongxu Yang, Shuo Li, Jinhong Song, Qimin Yang, Hao Chen (Corresponding author), Mingxiang Wu (Corresponding author), Tao Tan (Corresponding author-nrf)

Research output: Contribution to journalArticleAcademicpeer-review

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 languageEnglish
Article number10750144
Pages (from-to)690-699
Number of pages10
JournalIEEE Transactions on Artificial Intelligence
Volume6
Issue number3
Early online date11 Nov 2024
DOIs
Publication statusPublished - 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

Fingerprint

Dive into the research topics of 'Semi Supervised Breast MRI Density Segmentation Integrating Fine and Rough Annotations'. Together they form a unique fingerprint.

Cite this