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
来源
MEDICAL IMAGE COMPUTING AND COMPUTER ASSISTED INTERVENTION - MICCAI 2024, PT VIII | 2024年 / 15008卷
关键词
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 条
  • [41] UNETR: Transformers for 3D Medical Image Segmentation
    Hatamizadeh, Ali
    Tang, Yucheng
    Nath, Vishwesh
    Yang, Dong
    Myronenko, Andriy
    Landman, Bennett
    Roth, Holger R.
    Xu, Daguang
    2022 IEEE WINTER CONFERENCE ON APPLICATIONS OF COMPUTER VISION (WACV 2022), 2022, : 1748 - 1758
  • [42] Dual stream fusion module integrating 2D and 3D for 3D semi-supervised medical image segmentation
    He, Zhiquan
    Ouyang, Yating
    Wen, Yang
    SIGNAL IMAGE AND VIDEO PROCESSING, 2025, 19 (04)
  • [43] Weakly-Supervised 3D Medical Image Segmentation Using Geometric Prior and Contrastive Similarity
    Du, Hao
    Dong, Qihua
    Xu, Yan
    Liao, Jing
    IEEE TRANSACTIONS ON MEDICAL IMAGING, 2023, 42 (10) : 2936 - 2947
  • [44] Triple-task mutual consistency for semi-supervised 3D medical image segmentation
    Chen Y.
    Ma Y.
    Mei X.
    Zhang L.
    Fu Z.
    Ma J.
    Computers in Biology and Medicine, 2024, 175
  • [45] Segmentation of 3D Image of a Rock Sample Supervised by 2D Mineralogical Image
    Igor, Varfolomeev
    Yakimchuk, Ivan
    Sharchilev, Boris
    PROCEEDINGS 3RD IAPR ASIAN CONFERENCE ON PATTERN RECOGNITION ACPR 2015, 2015, : 346 - 350
  • [46] Inter-Slice Context Residual Learning for 3D Medical Image Segmentation
    Zhang, Jianpeng
    Xie, Yutong
    Wang, Yan
    Xia, Yong
    IEEE TRANSACTIONS ON MEDICAL IMAGING, 2021, 40 (02) : 661 - 672
  • [47] Robust Semi-Supervised 3D Medical Image Segmentation With Diverse Joint-Task Learning and Decoupled Inter-Student Learning
    Zhou, Quan
    Yu, Bin
    Xiao, Feng
    Ding, Mingyue
    Wang, Zhiwei
    Zhang, Xuming
    IEEE TRANSACTIONS ON MEDICAL IMAGING, 2024, 43 (06) : 2317 - 2331
  • [48] Image Understands Point Cloud: Weakly Supervised 3D Semantic Segmentation via Association Learning
    Sun, Tianfang
    Zhang, Zhizhong
    Tan, Xin
    Qu, Yanyun
    Xie, Yuan
    IEEE TRANSACTIONS ON IMAGE PROCESSING, 2024, 33 : 1838 - 1852
  • [49] A new, general method of 3D model generation for active shape image segmentation
    Lim, Seong-Jae
    Udupa, Jayaram K.
    Souza, Andre
    Jeong, Yong-Yeon
    Ho, Yo-Sung
    Torigian, Drew A.
    MEDICAL IMAGING 2006: IMAGE PROCESSING, PTS 1-3, 2006, 6144
  • [50] A Multiple Geometric Deformable Model Framework for Homeomorphic 3D Medical Image Segmentation
    Fan, Xian
    Bazin, Pierre-Louis
    Bogovic, John
    Bai, Ying
    Prince, Jerry L.
    2008 IEEE COMPUTER SOCIETY CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION WORKSHOPS, VOLS 1-3, 2008, : 472 - 478