An exploratory study of self-supervised pre-training on partially supervised multi-label classification on chest X-ray images

被引:0
|
作者
Dong, Nanqing [1 ]
Kampffmeyer, Michael [2 ]
Su, Haoyang [1 ]
Xing, Eric [3 ,4 ]
机构
[1] Shanghai Artificial Intelligence Lab, Shanghai 200232, Peoples R China
[2] UiT Arctic Univ Norway, Dept Phys & Technol, N-9019 Tromso, Norway
[3] Carnegie Mellon Univ, Machine Learning Dept, Pittsburgh, PA 15213 USA
[4] Mohamed Bin Zayed Univ Artificial Intelligence, Masdar City, Abu Dhabi, U Arab Emirates
关键词
Self-supervised learning; Partially supervised learning; Data scarcity; Multi-label classification;
D O I
10.1016/j.asoc.2024.111855
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
This paper serves as the first empirical study on self -supervised pre -training on partially supervised learning, an emerging yet unexplored learning paradigm with missing annotations. This is particularly important in the medical imaging domain, where label scarcity is the main challenge of practical applications. To promote the awareness of partially supervised learning, we leverage partially supervised multi -label classification on chest X-ray images as an instance task to illustrate the challenges of the problem of interest. Through a series of simulated experiments, the empirical findings validate that solving multiple pretext tasks jointly in the pretraining stage can significantly improve the downstream task performance under the partially supervised setup. Further, we propose a new pretext task, reverse vicinal risk minimization, and demonstrate that it provides a more robust and efficient alternative to existing pretext tasks for the instance task of interest.
引用
收藏
页数:11
相关论文
共 50 条
  • [1] MULTI-TASK SELF-SUPERVISED PRE-TRAINING FOR MUSIC CLASSIFICATION
    Wu, Ho-Hsiang
    Kao, Chieh-Chi
    Tang, Qingming
    Sun, Ming
    McFee, Brian
    Bello, Juan Pablo
    Wang, Chao
    2021 IEEE INTERNATIONAL CONFERENCE ON ACOUSTICS, SPEECH AND SIGNAL PROCESSING (ICASSP 2021), 2021, : 556 - 560
  • [2] Self-Supervised Pre-training for Time Series Classification
    Shi, Pengxiang
    Ye, Wenwen
    Qin, Zheng
    2021 INTERNATIONAL JOINT CONFERENCE ON NEURAL NETWORKS (IJCNN), 2021,
  • [3] Multi-Label Self-Supervised Learning with Scene Images
    Zhu, Ke
    Fu, Minghao
    Wu, Jianxin
    2023 IEEE/CVF INTERNATIONAL CONFERENCE ON COMPUTER VISION, ICCV, 2023, : 6671 - 6680
  • [4] Learning Consistent Semantic Representation for Chest X-ray via Anatomical Localization in Self-Supervised Pre-Training
    Chu, Surong
    Ren, Xueting
    Ji, Guohua
    Zhao, Juanjuan
    Shi, Jinwei
    Wei, Yangyang
    Pei, Bo
    Qiang, Yan
    IEEE JOURNAL OF BIOMEDICAL AND HEALTH INFORMATICS, 2025, 29 (03) : 2100 - 2112
  • [5] HANDLING SEVERE DATA IMBALANCE IN CHEST X-RAY IMAGE CLASSIFICATION WITH TRANSFER LEARNING USING SWAV SELF-SUPERVISED PRE-TRAINING
    Muljo, Hery Harjono
    Pardamean, Bens
    Elwirehardja, Gregorius Natanael
    Hidayat, Alam Ahmad
    Sudigyo, Digdo
    Rahutomo, Reza
    Cenggoro, Tjeng Wawan
    COMMUNICATIONS IN MATHEMATICAL BIOLOGY AND NEUROSCIENCE, 2023,
  • [6] Self-supervised Mean Teacher for Semi-supervised Chest X-Ray Classification
    Liu, Fengbei
    Tian, Yu
    Cordeiro, Filipe R.
    Belagiannis, Vasileios
    Reid, Ian
    Carneiro, Gustavo
    MACHINE LEARNING IN MEDICAL IMAGING, MLMI 2021, 2021, 12966 : 426 - 436
  • [7] Self-supervised ECG pre-training
    Liu, Han
    Zhao, Zhenbo
    She, Qiang
    BIOMEDICAL SIGNAL PROCESSING AND CONTROL, 2021, 70
  • [8] The Effectiveness of Self-supervised Pre-training for Multi-modal Endometriosis Classification
    Butler, David
    Wang, Hu
    Zhang, Yuan
    To, Minh-Son
    Condous, George
    Leonardi, Mathew
    Knox, Steven
    Avery, Jodie
    Hull, M. Louise
    Carneiro, Gustavo
    2023 45TH ANNUAL INTERNATIONAL CONFERENCE OF THE IEEE ENGINEERING IN MEDICINE & BIOLOGY SOCIETY, EMBC, 2023,
  • [9] Representation Recovering for Self-Supervised Pre-training on Medical Images
    Yan, Xiangyi
    Naushad, Junayed
    Sun, Shanlin
    Han, Kun
    Tang, Hao
    Kong, Deying
    Ma, Haoyu
    You, Chenyu
    Xie, Xiaohui
    2023 IEEE/CVF WINTER CONFERENCE ON APPLICATIONS OF COMPUTER VISION (WACV), 2023, : 2684 - 2694
  • [10] Stabilizing Label Assignment for Speech Separation by Self-supervised Pre-training
    Huang, Sung-Feng
    Chuang, Shun-Po
    Liu, Da-Rong
    Chen, Yi-Chen
    Yang, Gene-Ping
    Lee, Hung-yi
    INTERSPEECH 2021, 2021, : 3056 - 3060