Extraction and Classification of Lung Nodule Features Using Fusion Method of Self-Supervised Encoder Learning and Supervised Transfer Learning

Siqiao Yi,Huiyan Jiang,Yang Zhou,Guoyu Tong,Rongfei Zeng,Xianhua Han,Yan Pei
DOI: https://doi.org/10.1109/scisisis61014.2024.10760163
2024-01-01
Abstract:Classifying lung nodules as benign or malignant using CT images is crucial for the diagnosis of early-stage lung cancer and can significantly enhance survival rates. This paper proposes a method for extracting and classifying features of lung nodules through self-supervised encoder learning and supervised transfer learning to improve the distinguishing characteristics of lung nodules. Initially, the extracted nodule patches are perturbed to generate two corresponding augmented patches, which are utilized to learn global features using a contrastive learning network. Subsequently, one of the augmented patches is divided into non-overlapping segments and masked with a random mask. Reconstruction learning captures fine-grained local features through the reconstruction task. Furthermore, adversarial training directs the reconstruction learning process to yield more realistic information. Finally, the parameters of the pre-trained encoder are employed in a supervised classification task to categorize nodules as malignant or benign. Validation of the proposed method on the public LIDC-IDRI dataset demonstrates an accuracy of 89.0%, representing a 5.5% improvement over the state-of-the-art method.
What problem does this paper attempt to address?