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 条
  • [31] 3D Medical Image Segmentation by Multiple-Surface Active Volume Models
    Shen, Tian
    Huang, Xiaolei
    MEDICAL IMAGE COMPUTING AND COMPUTER-ASSISTED INTERVENTION - MICCAI 2009, PT II, PROCEEDINGS, 2009, 5762 : 1059 - 1066
  • [32] Medical Instrument Segmentation in 3D US by Hybrid Constrained Semi-Supervised Learning
    Yang, Hongxu
    Shan, Caifeng
    Bouwman, Arthur
    Dekker, Lukas R. C.
    Kolen, Alexander F.
    de With, Peter H. N.
    IEEE JOURNAL OF BIOMEDICAL AND HEALTH INFORMATICS, 2022, 26 (02) : 762 - 773
  • [33] Complementary consistency semi-supervised learning for 3D left atrial image segmentation
    Huang, Hejun
    Chen, Zuguo
    Chen, Chaoyang
    Lu, Ming
    Zou, Ying
    COMPUTERS IN BIOLOGY AND MEDICINE, 2023, 165
  • [34] ACTIVE CONTOUR BASED ON 3D STRUCTURE TENSOR APPLIED IN MEDICAL IMAGE SEGMENTATION
    Zhang, Ping
    Cui, Zhaohua
    Xue, Hale
    Zou, Dexuan
    Guo, Li
    INTERNATIONAL JOURNAL OF BIOMATHEMATICS, 2013, 6 (04)
  • [35] Exploiting Supervised Learning for 3D Model Semantic Segmentation Using Multispectral Data
    Ioannakis, George
    Arnaoutoglou, Fotis
    Koutsoudis, Anestis
    Chamzas, Christodoulos
    2019 6TH INTERNATIONAL CONFERENCE ON SIGNAL PROCESSING AND INTEGRATED NETWORKS (SPIN), 2019, : 965 - 968
  • [36] Multivariate Statistical Model for 3D Image Segmentation with Application to Medical Images
    Nigel M. John
    Mansur R. Kabuka
    Mohamed O. Ibrahim
    Journal of Digital Imaging, 2003, 16 : 365 - 377
  • [37] CS-LBF Improved Model for 3D Medical Image Segmentation
    Zhao, Wencang
    Zhang, Junbo
    SENSORS, MEASUREMENT AND INTELLIGENT MATERIALS, PTS 1-4, 2013, 303-306 : 2272 - 2279
  • [38] Multivariate statistical model for 3D image segmentation with application to medical images
    John, NM
    Kabuka, MR
    Ibrahim, MO
    JOURNAL OF DIGITAL IMAGING, 2003, 16 (04) : 365 - 377
  • [39] A hybrid framework for 3D medical image segmentation
    Chen, T
    Metaxas, D
    MEDICAL IMAGE ANALYSIS, 2005, 9 (06) : 547 - 565
  • [40] Supervised Learning for Semantic Segmentation of 3D LiDAR Data
    Mei, Jilin
    Chen, Jiayu
    Yao, Wen
    Zhao, Xijun
    Zhao, Huijing
    2019 30TH IEEE INTELLIGENT VEHICLES SYMPOSIUM (IV19), 2019, : 1491 - 1498