Attention Layer-Based Multidimensional Feature Extraction for Diagnosis of Lung Cancer

Abstract

At present, early lung cancer screening is mainly based on radiologists’ experience in diagnosing benign and malignant pulmonary nodules by lung CT images. On the other hand, intraoperative rapid freezing pathology needs to analyse the invasive adenocarcinoma nodules with the worst recovery in adenocarcinoma. Moreover, rapid freezing pathology has a low diagnostic accuracy for small-diameter nodules. Because of the above problems, an algorithm for diagnosing invasive adenocarcinoma nodules in ground-glass pulmonary nodules is based on CT images. According to the nodule space information and plane features, sample data of different dimensions are designed, namely, 3D space and 2D plane feature samples. The network structure is designed based on the attention mechanism and residual learning unit; 2D and 3D neural networks are along built. By fusing the feature vectors extracted from networks of different dimensions, the diagnosis results of invasive adenocarcinoma nodules are finally obtained. The algorithm was studied on 1760 ground-glass nodules with 5-20 mm diameter collected from a city chest hospital with surgical and pathological results. There were 340 nodules with invasive adenocarcinoma and 340 with noninvasive adenocarcinoma. A total of 1420 invasive nodule samples were cross-validated on this example dataset. The classification accuracy of the algorithm was 82.7%, the sensitivity was 82.9%, and the specificity was 82.6%.