Diagnosis with Confidence: Deep Learning for Reliable Classification of Squamous Lesions of the Upper Aerodigestive Tract

Abstract

BackgroundDiagnosis of head and neck (HN) squamous dysplasias and carcinomas is critical for patient care cure and follow-up. It can be challenging, especially for grading intraepithelial lesions. Despite recent simplification in the last WHO grading system, the inter- and intra-observer variability remains substantial, particularly for non-specialized pathologists, exhibiting the need for new tools to support pathologists. MethodsIn this study we investigated the potential of deep learning to assist the pathologist with automatic and reliable classification of HN lesions following the 2022 WHO classification system. We created, for the first time, a large-scale database of histological samples (>2000 slides) intended for developing an automatic diagnostic tool. We developed and trained a weakly supervised model performing classification from whole slide images (WSI). We evaluated our model on both internal and external test sets and we defined and validated a new confidence score to assess the predictions which can be used to identify difficult cases. ResultsOur model demonstrated high classification accuracy across all lesion types on both internal and external test sets (respectively average AUC: 0.878 (95% CI:[0.834-0.918]) and 0.886 (95% CI: [0.813-0.947])) and the confidence score allowed for accurate differentiation between reliable and uncertain predictions. ConclusionsOur results demonstrate that the model, associated with confidence measurements, can help in the difficult task of classifying head and neck squamous lesions by limiting variability and detecting ambiguous cases, taking us one step closer to a wider adoption of AI-based assistive tools.