FM-ABS: Promptable Foundation Model Drives Active Barely Supervised Learning for 3D Medical Image Segmentation

被引:0
|
作者
Xu, Zhe [1 ,2 ,3 ]
Chen, Cheng [2 ,3 ]
Lu, Donghuan [4 ]
Sun, Jinghan [6 ]
Wei, Dong [4 ]
Zheng, Yefeng [4 ]
Li, Quanzheng [2 ,3 ,5 ]
Tong, Raymond Kai-yu [1 ]
机构
[1] Chinese Univ Hong Kong, Dept Biomed Engn, Hong Kong, Peoples R China
[2] Massachusetts Gen Hosp, Ctr Adv Med Comp & Anal, Boston, MA 02114 USA
[3] Harvard Med Sch, Boston, MA 02115 USA
[4] Tencent YouTu Lab, Jarvis Res Ctr, Shenzhen, Peoples R China
[5] Massachusetts Gen Brigham, Data Sci Off, Boston, MA 02116 USA
[6] Xiamen Univ, Xiamen, Peoples R China
关键词
Barely Supervised; Foundation Model; Cross Labeling;
D O I
10.1007/978-3-031-72111-3_28
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Semi-supervised learning (SSL) has significantly advanced 3D medical image segmentation by effectively reducing the need for laborious dense labeling from radiologists. Traditionally focused on model-centric advancements, we anticipate that the SSL landscape will shift due to the emergence of open-source generalist foundation models, e.g., Segment Anything Model (SAM). These generalists have shown remarkable zero-shot segmentation capabilities with manual prompts, allowing a promising data-centric perspective for future SSL, particularly in pseudo and expert labeling strategies for enhancing the data pool. To this end, we propose the Foundation Model-driven Active Barely Supervised (FM-ABS) learning paradigm for developing customized 3D specialist segmentation models with shoestring annotation budgets, i.e., merely labeling three slices per scan. Specifically, building upon the basic mean-teacher framework, FM-ABS accounts for the intrinsic characteristics of 3D imaging and modernizes the SSL paradigm with two key data-centric designs: (i) specialist-generalist collaboration where the in-training specialist model delivers class-specific prompts to interact with the frozen class-agnostic generalist model across multiple views to acquire noisy-yet-effective pseudo labels, and (ii) expert-model collaboration that advocates active cross-labeling with notably low annotation efforts to progressively provide the specialist model with informative and efficient supervision in a human-in-the-loop manner, which benefits the automatic object-specific prompt generation in turn. Extensive experiments on two benchmark datasets show the promising results of our approach over recent SSL methods under extremely limited (barely) labeling budgets.
引用
收藏
页码:294 / 304
页数:11
相关论文
共 50 条
  • [21] COMBINING ATLAS AND ACTIVE CONTOUR FOR AUTOMATIC 3D MEDICAL IMAGE SEGMENTATION
    Gao, Yi
    Tannenbaum, Allen
    2011 8TH IEEE INTERNATIONAL SYMPOSIUM ON BIOMEDICAL IMAGING: FROM NANO TO MACRO, 2011, : 1401 - 1404
  • [22] A novel approach for active contour initialization in 3D medical image segmentation
    Peng, Yanjun
    Zhang, Shuai
    MEASUREMENT TECHNOLOGY AND ITS APPLICATION, PTS 1 AND 2, 2013, 239-240 : 1004 - 1010
  • [23] 3D medical image segmentation technique
    El-said, Shaimaa Ahmed
    INTERNATIONAL JOURNAL OF BIOMEDICAL ENGINEERING AND TECHNOLOGY, 2015, 17 (03) : 232 - 251
  • [24] Curriculum Self-Supervised Learning for 3D CT Cardiac Image Segmentation
    Taher, Mohammad Reza Hosseinzadeh
    Ikuta, Masaki
    Soni, Ravi
    MACHINE LEARNING FOR HEALTH, ML4H, VOL 225, 2023, 225 : 145 - 156
  • [25] 3D medical image segmentation based on semi-supervised learning using deep co-training
    Yang, Jingdong
    Li, Haoqiu
    Wang, Han
    Han, Man
    APPLIED SOFT COMPUTING, 2024, 159
  • [26] Pseudo-label guided selective mutual learning for semi-supervised 3D medical image segmentation
    Hang, Wenlong
    Dai, Peng
    Pan, Chengao
    Liang, Shuang
    Zhang, Qingfeng
    Wu, Qiang
    Jin, Yukun
    Wang, Qiong
    Qin, Jing
    BIOMEDICAL SIGNAL PROCESSING AND CONTROL, 2025, 100
  • [27] Fully 3D Active Surface with Machine Learning for PET Image Segmentation
    Comelli, Albert
    JOURNAL OF IMAGING, 2020, 6 (11)
  • [28] Distributing Deep Learning Hyperparameter Tuning for 3D Medical Image Segmentation
    Berral, Josep Ll
    Aranda, Oriol
    Luis Dominguez, Juan
    Torres, Jordi
    2022 IEEE 36TH INTERNATIONAL PARALLEL AND DISTRIBUTED PROCESSING SYMPOSIUM WORKSHOPS (IPDPSW 2022), 2022, : 1045 - 1052
  • [29] SEMI-SUPERVISED AND SELF-SUPERVISED COLLABORATIVE LEARNING FOR PROSTATE 3D MR IMAGE SEGMENTATION
    Osman, Yousuf Babiker M.
    Li, Cheng
    Huang, Weijian
    Elsayed, Nazik
    Ying, Leslie
    Zheng, Hairong
    Wang, Shanshan
    2023 IEEE 20TH INTERNATIONAL SYMPOSIUM ON BIOMEDICAL IMAGING, ISBI, 2023,
  • [30] SEMI-SUPERVISED 3D MEDICAL IMAGE SEGMENTATION VIA BOUNDARY-AWARE CONSISTENT HIDDEN REPRESENTATION LEARNING
    Liu, Linhu
    Tian, Jiang
    Cheng, Xiangqian
    Shi, Zhongchao
    Fan, Jianping
    Rui, Yong
    2022 IEEE INTERNATIONAL CONFERENCE ON IMAGE PROCESSING, ICIP, 2022, : 836 - 840