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 for medical image analysis: Discriminative, restorative, or adversarial?
    Haghighi, Fatemeh
    Taher, Mohammad Reza Hosseinzadeh
    Gotway, Michael B.
    Liang, Jianming
    [J]. MEDICAL IMAGE ANALYSIS, 2024, 94
  • [32] ProteinGLUE multi-task benchmark suite for self-supervised protein modeling
    Henriette Capel
    Robin Weiler
    Maurits Dijkstra
    Reinier Vleugels
    Peter Bloem
    K. Anton Feenstra
    [J]. Scientific Reports, 12
  • [33] 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
    [J]. IEEE TRANSACTIONS ON AFFECTIVE COMPUTING, 2023, 14 (03) : 2512 - 2525
  • [34] 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
    [J]. COMPUTER VISION AND IMAGE UNDERSTANDING, 2023, 229
  • [35] Self-Supervised Learning Based on Spatial Awareness for Medical Image Analysis
    Nguyen, Xuan-Bac
    Lee, Guee Sang
    Kim, Soo Hyung
    Yang, Hyung Jeong
    [J]. IEEE ACCESS, 2020, 8 : 162973 - 162981
  • [36] ProteinGLUE multi-task benchmark suite for self-supervised protein modeling
    Capel, Henriette
    Weiler, Robin
    Dijkstra, Maurits
    Vleugels, Reinier
    Bloem, Peter
    Feenstra, K. Anton
    [J]. SCIENTIFIC REPORTS, 2022, 12 (01)
  • [37] 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
    [J]. 2021 IEEE INTERNATIONAL CONFERENCE ON ACOUSTICS, SPEECH AND SIGNAL PROCESSING (ICASSP 2021), 2021, : 556 - 560
  • [38] Self-supervised Adversarial Multi-task Learning for Vocoder-based Monaural Speech Enhancement
    Du, Zhihao
    Lei, Ming
    Han, Jiqing
    Zhang, Shiliang
    [J]. INTERSPEECH 2020, 2020, : 3271 - 3275
  • [39] Cross-Domain Self-supervised Multi-task Feature Learning using Synthetic Imagery
    Ren, Zhongzheng
    Lee, Yong Jae
    [J]. 2018 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2018, : 762 - 771
  • [40] Pano-SfMLearner: Self-Supervised Multi-Task Learning of Depth and Semantics in Panoramic Videos
    Liu, Mengyi
    Wang, Shuhui
    Guo, Yulan
    He, Yuan
    Xue, Hui
    [J]. IEEE SIGNAL PROCESSING LETTERS, 2021, 28 : 832 - 836