Command Palette
Search for a command to run...
Transfer Learning in Polyp and Endoscopic Tool Segmentation from Colonoscopy Images
{Bjørn-Jostein Singstad Nefeli Panagiota Tzavara}
Abstract
Colorectal cancer is one of the deadliest and most widespread types of cancer in the world. Colonoscopy is the procedure used to detect and diagnose polyps from the colon, but today's detection rate shows a significant error rate that affects diagnosis and treatment. An automatic image segmentation algorithm may help doctors to improve the detection rate of pathological polyps in the colon. Furthermore, segmenting endoscopic tools in images taken during colonoscopy may contribute towards robotic assisted surgery. In this study, we trained and validated both pre-trained and not pre-trained segmentation models on two different data sets, containing images of polyps and endoscopic tools. Finally, we applied the models on two separate test sets and the best polyp model got a dice score 0.857 and the test instrument model got a dice score 0.948. Moreover, we found that pre-training of the models increased the performance in segmenting polyps and endoscopic tools.
Benchmarks
| Benchmark | Methodology | Metrics |
|---|---|---|
| medical-image-segmentation-on-hyper-kvasir | efficientnetb1 | Dice score: 0.857 Intersection over Union: 0.800 |
| medical-image-segmentation-on-kvasir | efficientnetb1 | DSC: 0.948 Dice Score: 0.948 Intersection over Union: 0.911 |
Build AI with AI
From idea to launch — accelerate your AI development with free AI co-coding, out-of-the-box environment and best price of GPUs.