Self-supervised multi-task learning for medical image analysis

被引:4
|
作者
Yu, Huihui [1 ,2 ]
Dai, Qun [1 ,2 ]
机构
[1] Nanjing Univ Aeronaut & Astronaut, Coll Comp Sci & Technol, Nanjing 211106, Peoples R China
[2] MIIT, Key Lab Pattern Anal & Machine Intelligence, Nanjing 211106, Peoples R China
基金
中国国家自然科学基金;
关键词
Medical image analysis; Self -supervised multi -task learning; Uniformity regularization; Chest X-ray image;
D O I
10.1016/j.patcog.2024.110327
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Deep learning is crucial for preliminary screening and diagnostic assistance based on medical image analysis. However, limited annotated data and complex anatomical structures challenge existing models as they struggle to capture anatomical context information effectively. In response, we propose a novel self-supervised multi-task learning framework (SSMT), which integrates two key modules: a discriminative-based module and a generativebased module. These modules collaborate through multiple proxy tasks, encouraging models to learn global discriminative representations and local fine-grained representations. Additionally, we introduce an efficient uniformity regularization to further enhance the learned representations. To demonstrate the effectiveness of SSMT, we conduct extensive experiments on six public Chest X-ray image datasets. Our results highlight that SSMT not only outperforms existing state-of-the-art methods but also achieves comparable performance to the supervised model in challenging downstream tasks. The ablation study demonstrates collaboration between the key components of SSMT, showcasing its potential for advancing medical image analysis.
引用
下载
收藏
页数:12
相关论文
共 50 条
  • [31] Self-supervised learning and semi-supervised learning for multi-sequence medical image classification
    Wang, Yueyue
    Song, Danjun
    Wang, Wentao
    Rao, Shengxiang
    Wang, Xiaoying
    Wang, Manning
    NEUROCOMPUTING, 2022, 513 : 383 - 394
  • [32] Self-supervised learning for medical image analysis: Discriminative, restorative, or adversarial?
    Haghighi, Fatemeh
    Taher, Mohammad Reza Hosseinzadeh
    Gotway, Michael B.
    Liang, Jianming
    MEDICAL IMAGE ANALYSIS, 2024, 94
  • [33] ProteinGLUE multi-task benchmark suite for self-supervised protein modeling
    Henriette Capel
    Robin Weiler
    Maurits Dijkstra
    Reinier Vleugels
    Peter Bloem
    K. Anton Feenstra
    Scientific Reports, 12
  • [34] GMSS: Graph-Based Multi-Task Self-Supervised Learning for EEG Emotion Recognition
    Li, Yang
    Chen, Ji
    Li, Fu
    Fu, Boxun
    Wu, Hao
    Ji, Youshuo
    Zhou, Yijin
    Niu, Yi
    Shi, Guangming
    Zheng, Wenming
    IEEE TRANSACTIONS ON AFFECTIVE COMPUTING, 2023, 14 (03) : 2512 - 2525
  • [35] Self-Supervised Learning Based on Spatial Awareness for Medical Image Analysis
    Nguyen, Xuan-Bac
    Lee, Guee Sang
    Kim, Soo Hyung
    Yang, Hyung Jeong
    IEEE ACCESS, 2020, 8 (08): : 162973 - 162981
  • [36] SSMTL plus plus : Revisiting self-supervised multi-task learning for video anomaly detection
    Barbalau, Antonio
    Ionescu, Radu Tudor
    Georgescu, Mariana-Iuliana
    Dueholm, Jacob
    Ramachandra, Bharathkumar
    Nasrollahi, Kamal
    Khan, Fahad Shahbaz
    Moeslund, Thomas B.
    Shah, Mubarak
    COMPUTER VISION AND IMAGE UNDERSTANDING, 2023, 229
  • [37] ProteinGLUE multi-task benchmark suite for self-supervised protein modeling
    Capel, Henriette
    Weiler, Robin
    Dijkstra, Maurits
    Vleugels, Reinier
    Bloem, Peter
    Feenstra, K. Anton
    SCIENTIFIC REPORTS, 2022, 12 (01)
  • [38] 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
  • [39] Cross-Domain Self-supervised Multi-task Feature Learning using Synthetic Imagery
    Ren, Zhongzheng
    Lee, Yong Jae
    2018 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2018, : 762 - 771
  • [40] Self-supervised Adversarial Multi-task Learning for Vocoder-based Monaural Speech Enhancement
    Du, Zhihao
    Lei, Ming
    Han, Jiqing
    Zhang, Shiliang
    INTERSPEECH 2020, 2020, : 3271 - 3275