Fine-Grain Classification Method of Non-small Cell Lung Cancer Based on Progressive Jigsaw and Graph Convolutional Network.

Zhengguang Cao,Wei Jia
DOI: https://doi.org/10.1007/978-981-99-8546-3_33
2024-01-01
Abstract:The classification of non-small cell lung cancer (NSCLC) is a challenging task that faces two main problems that limit its performance. The first is that the feature regions used to discriminate NSCLC types are usually scattered, requiring classification models with the ability to accurately locate these feature regions and capture contextual information. Secondly, there are a large number of redundant regions in NSCLC pathology images that interfere with classification, but existing classification methods make it difficult to deal with them effectively. To solve these problems, we propose a fine-grained classification network, Progressive Jigsaw and Graph Convolutional Network (PJGC-Net). The network consists of two modules. For the first problem, we designed the GCN-Based multi-scale puzzle generation (GMPG) module to achieve fine-grained learning by training separate networks for different scale images, which can help the network identify and localize feature regions used to discriminate NSCLC types as well as their contextual information. For the second problem, we propose the jigsaw supervised progressive training (JSPT) module, which removes a large number of redundant regions and helps the proposed model focus on the effective discriminative regions. Our experimental results and visualization results on the LC25000 dataset demonstrate the feasibility of this method, and our method consistently outperforms other existing classification methods.
What problem does this paper attempt to address?