A novel collaborative self-supervised learning method for radiomic data.

Journal: NeuroImage
Published Date:

Abstract

The computer-aided disease diagnosis from radiomic data is important in many medical applications. However, developing such a technique relies on labeling radiological images, which is a time-consuming, labor-intensive, and expensive process. In this work, we present the first novel collaborative self-supervised learning method to solve the challenge of insufficient labeled radiomic data, whose characteristics are different from text and image data. To achieve this, we present two collaborative pretext tasks that explore the latent pathological or biological relationships between regions of interest and the similarity and dissimilarity of information between subjects. Our method collaboratively learns the robust latent feature representations from radiomic data in a self-supervised manner to reduce human annotation efforts, which benefits the disease diagnosis. We compared our proposed method with other state-of-the-art self-supervised learning methods on a simulation study and two independent datasets. Extensive experimental results demonstrated that our method outperforms other self-supervised learning methods on both classification and regression tasks. With further refinement, our method will have the potential advantage in automatic disease diagnosis with large-scale unlabeled data available.

Authors

  • Zhiyuan Li
    School of Clinical Medicine, General Hospital of Ningxia Medical University, Yinchuan, China.
  • Hailong Li
    College of Energy, Xiamen University, Xiamen, 361005 People's Republic of China.
  • Anca L Ralescu
    Department of Computer Science, University of Cincinnati, Cincinnati, OH, USA.
  • Jonathan R Dillman
    Department of Radiology, Division of Thoracoabdominal Imaging, Cincinnati Children's Hospital Medical Center, University of Cincinnati College of Medicine, 3333 Burnet Ave., Cincinnati, OH, 45229-3039, USA. jonathan.dillman@cchmc.org.
  • Nehal A Parikh
    Perinatal Institute, Department of Pediatrics, Cincinnati Children's Hospital Medical Center, Cincinnati, OH, United States; Department of Pediatrics, University of Cincinnati College of Medicine, Cincinnati, OH, United States; Pediatric Neuroimaging Research Consortium, Cincinnati Children's Hospital Medical Center, Cincinnati, OH, United States.
  • Lili He
    Department of Food Science, University of Massachusetts Amherst, United States of America. Electronic address: lilihe@foodsci.umass.edu.