A multi-center analysis of deep learning methods for video polyp detection and segmentation
Organizations: Computer Engineering Department, Arab Academy for Science and Technology,Egypt · AI in Medicine and Surgery Group, School of Computer Science, University of Leeds, Leeds, LS2 9JT, UK · Department of Artificial Intelligence and Knowledge Systems, University of Würzburg, Germany · CRO Centro Riferimento Oncologico IRCCS, Aviano, Italy ·
Abstract
Colonic polyps are well-recognized precursors to colorectal cancer (CRC), typically detected during colonoscopy. However, the variability in appearance, location, and size of these polyps complicates their detection and removal, leading to challenges in effective surveillance, intervention, and subsequently CRC prevention. The processes of colonoscopy surveillance and polyp removal are highly reliant on the expertise of gastroenterologists and occur within the complexities of the colonic structure. As a result, there is a high rate of missed detections and incomplete removal of colonic polyps, which can adversely impact patient outcomes. Recently, automated methods that use machine learning have been developed to enhance polyps detection and segmentation, thus helping clinical processes and reducing missed rates. These advancements highlight the potential for improving diagnostic accuracy in real-time applications, which ultimately facilitates more effective patient management. Furthermore, integrating sequence data and temporal information could significantly enhance the precision of these methods by capturing the dynamic nature of polyp growth and the changes that occur over time. To rigorously investigate these challenges, data scientists and experts gastroenterologists collaborated to compile a comprehensive dataset that spans multiple centers and diverse populations. This initiative aims to underscore the critical importance of incorporating sequence data and temporal information in the development of robust automated detection and segmentation methods. This study evaluates the applicability of deep learning techniques developed in real-time clinical colonoscopy tasks using sequence data, highlighting the critical role of temporal relationships between frames in improving diagnostic precision.
Figures & tables
| Team Name | Algorithm | Backbone | Nature | Choice Basis | Data Aug. | Loss | Opt. | Code |
| Task I: Polyp detection | ||||||||
| IMed | Two Stage + Detector | ResNest + TCT | Aggregation | Context Speed | Yes | Cascade Loss | SGD + Warmup | N/A |
| SDS_RBS | YOLO | YOLOv5l6 YOLOv65x6 | Ensemble | Accuracy Speed | Yes | CIoU | SGD | N/A |
| WürzVision | TRA | ResNet50 | Temporal Attention | F1-Score | Yes | IoU | SGD | [d2] |
| Task II: Polyp segmentation | ||||||||
| Arrah_htic | XP-Net | U-Net | Knowledge distillation | Accuracy Speed | Yes | DSC Tversky | Adam | [s2] |
| Average precision, AP | AP across scales | |||||
| Teams/Method | ||||||
| AP mean | AP 50 | AP 75 | AP small | AP medium | AP large | |
| Arrahhtic ∗ | 0.117 | 0.208 | 0.093 | 0.001 | 0.095 | 0.274 |
| IMED | 0.116 | 0.185 | 0.141 | 0.000 | 0.102 | 0.256 |
| SDS-RBS | 0.334 | 0.481 | 0.367 | 0.038 | 0.338 | 0.677 |
| UCU ML Lab ∗ | 0.146 | 0.266 | 0.143 | 0.006 | 0.184 | 0.184 |
| Teams | JC | DSC | F2 | PPV | Rec | Acc | NSD | MASD | d_hfd |
| Arrahhtic | 0.438 0.42 | 0.481 0.44 | 0.495 0.45 | 0.696 0.39 | 0.635 0.44 | 0.989 0.01 | 0.357 0.39 | 68.091 144.90 | 0.388 0.22 |
| HeHIK | 0.720 0.35 | 0.765 0.35 | 0.766 0.36 | 0.897 0.21 | 0.807 0.33 | 0.995 0.01 | 0.675 0.35 | 12.496 59.07 | 0.301 0.22 |
| iMed | 0.522 0.42 | 0.568 0.43 | 0.560 0.43 | 0.860 0.28 | 0.601 0.43 | 0.989 0.02 | 0.443 0.40 | 32.282 103.65 | 0.392 0.28 |
| lswangxmu | 0.747 0.34 | 0.787 0.34 | 0.796 0.35 | 0.893 0.21 | 0.843 0.32 | 0.995 0.01 | 0.715 0.35 | 8.917 38.75 | 0.282 0.21 |
| SDS-RBS | 0.718 0.37 | 0.754 0.37 | 0.744 0.38 | 0.936 0.19 | 0.764 0.36 | 0.995 0.01 | 0.683 0.37 | 11.008 42.92 | 0.287 0.22 |
| UCUML | 0.500 0.41 | 0.550 0.43 | 0.556 0.43 | 0.606 0.43 | 0.794 0.34 | 0.986 0.03 | 0.448 0.38 | 53.947 140.81 | 0.253 0.15 |
| Teams | GPU | Training | Inference | R time | R algo. | R overall |
| (hrs) | (fps) | |||||
| Task I: Polyp detection | ||||||
| IMED | 2 Tesla V100 32G | 1 | 72 | 1 | 4 | 3 |
| SDS_RBS | 1 NVIDIA RTX3090 | 5 | 9.10 | 3 | 1 | 1 |
| WürzVision | NVIDIA QUADRO RTX 8000 | 14 | 24 | 2 | 2 | 2 |
| Task II: Polyp Segmentation | ||||||
Medical Image Analysis
Medical image analysis uses computational methods to interpret clinical images, measure anatomy and disease, and support diagnosis and treatment.
See all 291 papers