NeoUNet: Towards Accurate Colon Polyp Segmentation and Neoplasm Detection

• Published in: Advances in Visual Computing Vol. 13018; pp. 15 - 28
• Main Authors: Ngoc Lan, Phan, An, Nguyen Sy, Hang, Dao Viet, Long, Dao Van, Trung, Tran Quang, Thuy, Nguyen Thi, Sang, Dinh Viet
• Format: Book Chapter
• Language: English
• Published: Switzerland Springer International Publishing AG 2021; Springer International Publishing
• Series: Lecture Notes in Computer Science
• Subjects: Colonoscopy; Deep learning; Polyp segmentation
• ISBN: 9783030904357; 3030904350
• ISSN: 0302-9743; 1611-3349
• DOI: 10.1007/978-3-030-90436-4_2

Automatic polyp segmentation has proven to be immensely helpful for endoscopy procedures, reducing the missing rate of adenoma detection for endoscopists while increasing efficiency. However, classifying a polyp as being neoplasm or not and segmenting it at the pixel level is still a challenging task for doctors to perform in a limited time. In this work, we propose a fine-grained formulation for the polyp segmentation problem. Our formulation aims to not only segment polyp regions, but also identify those at high risk of malignancy with high accuracy. We then present a UNet-based neural network architecture called NeoUNet, along with a hybrid loss function to solve this problem. Experiments show highly competitive results for NeoUNet on our benchmark dataset compared to existing polyp segmentation models.