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 条
  • [21] MULTI-TASK SELF-SUPERVISED VISUAL REPRESENTATION LEARNING FOR MONOCULAR ROAD SEGMENTATION
    Cho, Jaehoon
    Kim, Youngjung
    Jung, Hyungjoo
    Oh, Changjae
    Youn, Jaesung
    Sohn, Kwanghoon
    2018 IEEE INTERNATIONAL CONFERENCE ON MULTIMEDIA AND EXPO (ICME), 2018,
  • [22] scPretrain: multi-task self-supervised learning for cell-type classification
    Zhang, Ruiyi
    Luo, Yunan
    Ma, Jianzhu
    Zhang, Ming
    Wang, Sheng
    BIOINFORMATICS, 2022, 38 (06) : 1607 - 1614
  • [23] Dive into the details of self-supervised learning for medical image analysis
    Zhang, Chuyan
    Zheng, Hao
    Gu, Yun
    MEDICAL IMAGE ANALYSIS, 2023, 89
  • [24] Self-supervised learning for medical image analysis: a comprehensive review
    Rani, Veenu
    Kumar, Munish
    Gupta, Aastha
    Sachdeva, Monika
    Mittal, Ajay
    Kumar, Krishan
    EVOLVING SYSTEMS, 2024, 15 (04) : 1607 - 1633
  • [25] Multi-task Self-supervised Few-Shot Detection
    Zhang, Guangyong
    Duan, Lijuan
    Wang, Wenjian
    Gong, Zhi
    Ma, Bian
    PATTERN RECOGNITION AND COMPUTER VISION, PRCV 2023, PT XII, 2024, 14436 : 107 - 119
  • [26] MULTI-TASK CURRICULUM LEARNING FOR SEMI-SUPERVISED MEDICAL IMAGE SEGMENTATION
    Wang, Kaiping
    Zhan, Bo
    Luo, Yanmei
    Zhou, Jiliu
    Wu, Xi
    Wang, Yan
    2021 IEEE 18TH INTERNATIONAL SYMPOSIUM ON BIOMEDICAL IMAGING (ISBI), 2021, : 925 - 928
  • [27] Multi-Task Collaborative Network: Bridge the Supervised and Self-Supervised Learning for EEG Classification in RSVP Tasks
    Li, Hongxin
    Tang, Jingsheng
    Li, Wenqi
    Dai, Wei
    Liu, Yaru
    Zhou, Zongtan
    IEEE TRANSACTIONS ON NEURAL SYSTEMS AND REHABILITATION ENGINEERING, 2024, 32 : 638 - 651
  • [28] Weakly-Supervised Medical Image Segmentation Based on Multi-task Learning
    Xie, Xuanhua
    Fan, Huijie
    Yu, Zhencheng
    Bai, Haijun
    Tang, Yandong
    INTELLIGENT ROBOTICS AND APPLICATIONS (ICIRA 2022), PT II, 2022, 13456 : 395 - 404
  • [29] 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
  • [30] 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