Self-supervised multi-task learning for medical image analysis

被引:10
|
作者
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] MT-SLVR: Multi-Task Self-Supervised Learning for Transformation In(Variant) Representations
    Heggan, Calum
    Hospedales, Tim
    Budgett, Sam
    Yaghoobi, Mehrdad
    INTERSPEECH 2023, 2023, : 4399 - 4403
  • [32] TransMEF: A Transformer-Based Multi-Exposure Image Fusion Framework Using Self-Supervised Multi-Task Learning
    Qu, Linhao
    Liu, Shaolei
    Wang, Manning
    Song, Zhijian
    THIRTY-SIXTH AAAI CONFERENCE ON ARTIFICIAL INTELLIGENCE / THIRTY-FOURTH CONFERENCE ON INNOVATIVE APPLICATIONS OF ARTIFICIAL INTELLIGENCE / THE TWELVETH SYMPOSIUM ON EDUCATIONAL ADVANCES IN ARTIFICIAL INTELLIGENCE, 2022, : 2126 - 2134
  • [33] Self-supervised learning for medical image analysis using image context restoration
    Chen, Liang
    Bentley, Paul
    Mori, Kensaku
    Misawa, Kazunari
    Fujiwara, Michitaka
    Rueckert, Daniel
    MEDICAL IMAGE ANALYSIS, 2019, 58
  • [34] 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
  • [35] 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
  • [36] 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
  • [37] 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
  • [38] 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
  • [39] 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)
  • [40] Exploring multi-task learning and data augmentation in dementia detection with self-supervised pretrained models
    Chen, Minchuan
    Miao, Chenfeng
    Ma, Jun
    Wang, Shaojun
    Xiao, Jing
    INTERSPEECH 2023, 2023, : 5037 - 5041