Task Augmentation-Based Meta-Learning Segmentation Method for Retinopathy.

Journal: IEEE transactions on pattern analysis and machine intelligence
Published Date:

Abstract

Deep learning (DL) requires large amounts of labeled data, which is extremely time-consuming and laborintensive to obtain for medical image segmentation tasks. Metalearning focuses on developing learning strategies that enable quick adaptation to new tasks with limited labeled data. However, rich-class medical image segmentation datasets for constructing meta-learning multi-tasks are currently unavailable. In addition, data collected from various healthcare sites and devices may present significant distribution differences, potentially degrading model's performance. In this paper, we propose a task augmentation-based meta-learning method for retinal image segmentation (TAMS) to meet labor-intensive annotation demand. A retinal Lesion Simulation Algorithm (LSA) is proposed to automatically generate multi-class retinal disease datasets with pixel-level segmentation labels, such that metalearning tasks can be augmented without collecting data from various sources. In addition, a novel simulation function library is designed to control generation process and ensure interpretability. Moreover, a generative simulation network (GSNet) with an improved adversarial training strategy is introduced to maintain high-quality representations of complex retinal diseases. TAMS is evaluated on three different OCT and CFP image datasets, and comprehensive experiments have demonstrated that TAMS achieves superior segmentation performance than state-of-the-art models.

Authors

  • Jingtao Wang
  • Muhammad Mateen
    Department of Computer Science, Air University Multan Campus, Multan 60000, Pakistan.
  • Dehui Xiang
  • Weifang Zhu
  • Fei Shi
    Medical Image Processing, Analysis, and Visualization (MIVAP) Lab, School of Electronics and Information Engineering, Soochow University, Suzhou, China.
  • Jing Huang
    Department of Nephrology, The Second Affiliated Hospital of Guangxi Medical University, Nanning, Guangxi, China.
  • Kai Sun
    Department of Materials Science and Engineering, Jinan University.
  • Jun Dai
    Shanghai Veterinary Research Institute, Chinese Academy of Agricultural Sciences, Shanghai, China.
  • Jingchen Xu
  • Su Zhang
  • Xinjian Chen
    Medical Image Processing, Analysis, and Visualization (MIVAP) Lab, School of Electronics and Information Engineering, Soochow University, Suzhou, China.

Keywords

No keywords available for this article.