Acta Optica Sinica, Volume. 40, Issue 18, 1810003(2020)

Self-Supervised Transfer Learning of Pulmonary Nodule Classification Based on Partially Annotated CT Images

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

    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.

    Tools

    Get Citation

    Copy Citation Text

    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

    EndNote(RIS)BibTexPlain Text
    Save article for my favorites
    Paper Information

    Category: Image Processing

    Received: Apr. 17, 2020

    Accepted: Jun. 11, 2020

    Published Online: Aug. 27, 2020

    The Author Email: Huang Hong (hhuang@cqu.edu.cn)

    DOI:10.3788/AOS202040.1810003

    Topics