CA-Net: Leveraging Contextual Features for Lung Cancer Prediction

Mingzhou Liu,Fandong Zhang,Xinwei Sun,Yizhou Yu,Yizhou Wang
DOI: https://doi.org/10.1007/978-3-030-87240-3_3
2021-01-01
Abstract:In the early diagnosis of lung cancer, an important step is classifying malignancy/benignity for each lung nodule. For this classification, the nodule's features (e.g., shape, margin) have traditionally been the main focus. Recently, the contextual features attract increasing attention, due to the complementary information they provide. Clinically, such contextual features refer to the features of nodule's surrounding structures, such that (together with nodule's features) they can expose discriminate patterns for the malignant/benign, such as vascular convergence and fissural attachment. To leverage such contextual features, we propose a Context Attention Network (CA-Net) which extracts both nodule's and contextual features and then effectively fuses them during malignancy/benignity classification. To accurately identify the contextual features that contain structures distorted/attached by the nodule, we take the nodule's features as a reference via an attention mechanism. Further, we propose a feature fusion module that can adaptively adjust the weights of nodule's and contextual features across nodules. The utility of our proposed method is demonstrated by a noticeable margin over the 1st place on Data Science Bowl 2017 dataset in Kaggle's competition.
What problem does this paper attempt to address?