Single-View Contrastive Learning for Laryngeal Leukoplakia Classification With NBI Laryngoscopy Images.

Abstract

Laryngeal cancer is the second most common upper respiratory tract cancer. Early and accurate diagnosis can improve the cure rate of patients. Laryngoscopy with NBI is a commonly used tool that can help endoscopists diagnose laryngeal diseases. However, the fine classification of laryngeal leukoplakia using NBI images is challenging for computer-aided diagnosis. In this article, we propose a single-view contrastive learning network to locate lesion regions, construct sample pairs for contrastive learning, and provide pseudo-labels to unlabeled data in order to achieve fine classification under small samples. Firstly, we pretrain the backbone network using the original NBI images. Secondly, in order to augment the number of samples for contrastive learning, we design different patch generation methods based on an attention-guided network. The original NBI images are cropped into small patches for the purpose of generating lesion-related regions and complementary samples. The pseudo-labels of these small patches are obtained by applying the pre-trained backbone network. Finally, we combine the contrastive loss function and the cross-entropy loss function for jointly training the backbone network and contrastive learning network. Our NBI dataset is classified into six categories: normal tissue, inflammatory keratosis, mild dysplasia, moderate dysplasia, severe dysplasia, and squamous cell carcinoma. Experimental results demonstrate that our model achieves an accuracy of 96.12%, which is higher than the current mainstream models. Our model also achieves high specificity and sensitivity. The code is available at https://github.com/hans-bbt/single-view-contrastive-learning.