• Acta Optica Sinica
  • Vol. 40, Issue 18, 1810003 (2020)
Hong Huang1、*, Chao Peng1, Ruoyu Wu1, Junli Tao2, and Jiuquan Zhang2
Author Affiliations
  • 1Key Laboratory of Optoelectronic Technology & Systems, Ministry of Education, Chongqing University, Chongqing 400044, China;
  • 2Department of Radiology, Chongqing University Cancer Hospital, Chongqing 400030, China
  • show less
    DOI: 10.3788/AOS202040.1810003 Cite this Article Set citation alerts
    Hong Huang, Chao Peng, Ruoyu Wu, Junli Tao, Jiuquan Zhang. Self-Supervised Transfer Learning of Pulmonary Nodule Classification Based on Partially Annotated CT Images[J]. Acta Optica Sinica, 2020, 40(18): 1810003 Copy Citation Text show less

    Abstract

    The process of training a deep learning model requires many annotation samples, even though annotation data is difficult to obtain in the medical field. A self-supervised learning algorithm combined with partial annotation data is proposed as a solution to this problem, in order to improve classification performance of 3D pulmonary nodules. Based on the traditional self-supervised training network structure, a multitask learning network structure is designed to address a large amount of unannotated data and a small amount of annotated data obtained from medical image processing tasks. First, the proposed algorithm trains the unannotated data, and then explores the annotation data to continuously train the model. Thus, this algorithm manages to share partial network structures and parameters between the annotated and unannotated data. Compared to traditional self-supervised learning methods, the proposed algorithm can learn to recognize the discriminant features of pulmonary nodules to ensure the model's capacity to generalize, therefore, model transfer learning can also perform better when applied to the classification of pulmonary nodules. The classification accuracy of the proposed algorithm on LIDC-IDRI dataset is 0.886, and the area under the curve (AUC) is 0.929. The results of the investigation indicate that the proposed algorithm can effectively improve classification performance of pulmonary nodules.
    Hong Huang, Chao Peng, Ruoyu Wu, Junli Tao, Jiuquan Zhang. Self-Supervised Transfer Learning of Pulmonary Nodule Classification Based on Partially Annotated CT Images[J]. Acta Optica Sinica, 2020, 40(18): 1810003
    Download Citation