Whole Slide Cervical Cancer Screening Using Graph Attention Network and Supervised Contrastive Learning

Xin Zhang,Maosong Cao,Sheng Wang,Jiayin Sun,Xiangshan Fan,Qian Wang,Lichi Zhang
DOI: https://doi.org/10.1007/978-3-031-16434-7_20
2022-01-01
Abstract:Cervical cancer is one of the primary factors that endanger women's health, and Thin-prep cytologic test (TCT) has been widely applied for early screening. Automatic whole slide image (WSI) classification is highly demanded, as it can significantly reduce the workload of pathologists. Current methods are mainly based on suspicious lesion patch extraction and classification, which ignore the intrinsic relationships between suspicious patches and neglect the other patches apart from the suspicious patches, and therefore limit their robustness and generalizability. Here we propose a novel method to solve the problem, which is based on graph attention network (GAT) and supervised contrastive learning. First, for each WSI, we extract and rank a large number of representative patches based on suspicious cell detection. Then, we select the top-K and bottom-K suspicious patches to construct two graphs seperately. Next, we introduce GAT to aggregate the features from each node, and use supervised contrastive learning to obtain valuable representations of graphs. Specifically, we design a novel contrastive loss so that the latent distances between two graphs are enlarged for positive WSIs and reduced for negative WSIs. Experimental results show that the proposed GAT method outperforms conventional methods, and also demonstrate the effectiveness of supervised contrastive learning.
What problem does this paper attempt to address?