Semi-supervised image segmentation using a residual-driven mean teacher and an exponential Dice loss

被引:3
作者
Mei, Chenyang [1 ]
Yang, Xiaoguo [2 ]
Zhou, Mi [2 ,3 ]
Zhang, Shaodan [1 ]
Chen, Hao [1 ]
Yang, Xiaokai [2 ,3 ]
Wang, Lei [1 ]
机构
[1] Wenzhou Med Univ, Eye Hosp, Sch Ophthalmol & Optometry, Wenzhou 325027, Peoples R China
[2] Shanghai Univ, Wenzhou Peoples Hosp, Affiliated Hosp 3, Dept Neurol, Wenzhou 325041, Peoples R China
[3] Shanghai Univ, Sch Med, Shanghai 200444, Peoples R China
关键词
Semi-supervised segmentation; Mean teacher; Exponential moving average; Dice loss; Optic cup;
D O I
10.1016/j.artmed.2023.102757
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Semi -supervised segmentation plays an important role in computer vision and medical image analysis and can alleviate the burden of acquiring abundant expert -annotated images. In this paper, we developed a residualdriven semi -supervised segmentation method (termed RDMT) based on the classical mean teacher (MT) framework by introducing a novel model -level residual perturbation and an exponential Dice (eDice) loss. The introduced perturbation was integrated into the exponential moving average (EMA) scheme to enhance the performance of the MT, while the eDice loss was used to improve the detection sensitivity of a given network to object boundaries. We validated the developed method by applying it to segment 3D Left Atrium (LA) and 2D optic cup (OC) from the public LASC and REFUGE datasets based on the V -Net and U -Net, respectively. Extensive experiments demonstrated that the developed method achieved the average Dice score of 0.8776 and 0.7751, when trained on 10% and 20% labeled images, respectively for the LA and OC regions depicted on the LASC and REFUGE datasets. It significantly outperformed the MT and can compete with several existing semi -supervised segmentation methods (i.e., HCMT, UAMT, DTC and SASS).
引用
收藏
页数:11
相关论文
共 37 条
  • [1] Butoi VI, 2023, Arxiv, DOI [arXiv:2304.06131, DOI 10.48550/ARXIV.2304.06131]
  • [2] Chen C., 2018, arXiv
  • [3] Exploring deep feature-blending capabilities to assist glaucoma screening
    Haider, Adnan
    Arsalan, Muhammad
    Park, Chanhum
    Sultan, Haseeb
    Park, Kang Ryoung
    [J]. APPLIED SOFT COMPUTING, 2023, 133
  • [4] Artificial Intelligence-based computer-aided diagnosis of glaucoma using retinal fundus images
    Haider, Adnan
    Arsalan, Muhammad
    Lee, Min Beom
    Owais, Muhammad
    Mahmood, Tahir
    Sultan, Haseeb
    Park, Kang Ryoung
    [J]. EXPERT SYSTEMS WITH APPLICATIONS, 2022, 207
  • [5] End-to-end multi-task learning for simultaneous optic disc and cup segmentation and glaucoma classification in eye fundus images
    Hervella, Alvaro S.
    Rouco, Jose
    Novo, Jorge
    Ortega, Marcos
    [J]. APPLIED SOFT COMPUTING, 2022, 116
  • [6] nnU-Net: a self-configuring method for deep learning-based biomedical image segmentation
    Isensee, Fabian
    Jaeger, Paul F.
    Kohl, Simon A. A.
    Petersen, Jens
    Maier-Hein, Klaus H.
    [J]. NATURE METHODS, 2021, 18 (02) : 203 - +
  • [7] Jiao RS, 2022, Arxiv, DOI arXiv:2207.14191
  • [8] Kirillov A, 2023, Arxiv, DOI [arXiv:2304.02643, 10.48550/arXiv.2304.02643, DOI 10.48550/ARXIV.2304.02643]
  • [9] Laine S, 2017, Arxiv, DOI [arXiv:1610.02242, DOI 10.48550/ARXIV.1610.02242]
  • [10] Li S., 2020, C MED IM COMP COMP A