Using DUCK-Net for Polyp Image Segmentation
About
This paper presents a novel supervised convolutional neural network architecture, "DUCK-Net", capable of effectively learning and generalizing from small amounts of medical images to perform accurate segmentation tasks. Our model utilizes an encoder-decoder structure with a residual downsampling mechanism and a custom convolutional block to capture and process image information at multiple resolutions in the encoder segment. We employ data augmentation techniques to enrich the training set, thus increasing our model's performance. While our architecture is versatile and applicable to various segmentation tasks, in this study, we demonstrate its capabilities specifically for polyp segmentation in colonoscopy images. We evaluate the performance of our method on several popular benchmark datasets for polyp segmentation, Kvasir-SEG, CVC-ClinicDB, CVC-ColonDB, and ETIS-LARIBPOLYPDB showing that it achieves state-of-the-art results in terms of mean Dice coefficient, Jaccard index, Precision, Recall, and Accuracy. Our approach demonstrates strong generalization capabilities, achieving excellent performance even with limited training data. The code is publicly available on GitHub: https://github.com/RazvanDu/DUCK-Net
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Polyp Segmentation | Kvasir-SEG (test) | mIoU0.9051 | 87 | |
| Polyp Segmentation | CVC-ColonDB (test) | Mean Dice0.8945 | 62 | |
| Polyp Segmentation | CVC-300 (test) | mDice0.7598 | 25 | |
| Polyp Segmentation | PolypGen Data Centre 4 | Dice Coefficient55.79 | 14 | |
| Polyp Segmentation | Data Centre PolypGen 6 | Dice55.73 | 14 | |
| Polyp Segmentation | PolypGen Data Centre 2 | Dice51.29 | 14 | |
| Polyp Segmentation | PolypGen 16 Video Sequences | Dice Coefficient63.68 | 14 | |
| Polyp Segmentation | PolypGen 15 Video Sequences | Dice36.54 | 14 | |
| Polyp Segmentation | Kvasir-SEG CVC-ClinicDB (test) | FPR6.78 | 14 | |
| Polyp Segmentation | PolypGen 2 video sequences (test) | Dice Coefficient44.29 | 14 |