A Hierarchical Multi-Task Approach to Gastrointestinal Image Analysis

TL;DR

This paper presents a hierarchical multi-task deep learning approach for gastrointestinal image analysis, improving detection and classification of lesions and polyps in endoscopic videos, with high accuracy demonstrated on challenge data.

Contribution

It introduces a novel hierarchical CNN model that jointly performs multi-category classification and polyp segmentation in GI images, advancing automated endoscopic analysis.

Findings

Achieved over 91% MCC and Micro-F1 scores in classification

Secured a 92.30 F1 score in polyp segmentation

Provided effective solutions for GI lesion detection and segmentation

Abstract

A large number of different lesions and pathologies can affect the human digestive system, resulting in life-threatening situations. Early detection plays a relevant role in the successful treatment and the increase of current survival rates to, e.g., colorectal cancer. The standard procedure enabling detection, endoscopic video analysis, generates large quantities of visual data that need to be carefully analyzed by an specialist. Due to the wide range of color, shape, and general visual appearance of pathologies, as well as highly varying image quality, such process is greatly dependent on the human operator experience and skill. In this work, we detail our solution to the task of multi-category classification of images from the gastrointestinal (GI) human tract within the 2020 Endotect Challenge. Our approach is based on a Convolutional Neural Network minimizing a hierarchical error function that takes into account not only the finding category, but also its location within the GI tract (lower/upper tract), and the type of finding (pathological finding/therapeutic intervention/anatomical landmark/mucosal views' quality). We also describe in this paper our solution for the challenge task of polyp segmentation in colonoscopies, which was addressed with a pretrained double encoder-decoder network. Our internal cross-validation results show an average performance of 91.25 Mathews Correlation Coefficient (MCC) and 91.82 Micro-F1 score for the classification task, and a 92.30 F1 score for the polyp segmentation task. The organization provided feedback on the performance in a hidden test set for both tasks, which resulted in 85.61 MCC and 86.96 F1 score for classification, and 91.97 F1 score for polyp segmentation. At the time of writing no public ranking for this challenge had been released.