Debesh Jha, Pia H. Smedsrud, Michael A. Riegler, Dag Johansen, Thomas de Lange, Pal Halvorsen, Havard D. Johansen
Accurate computer-aided polyp detection and segmentation during colonoscopy examinations can help endoscopists resect abnormal tissue and thereby decrease chances of polyps growing into cancer. Towards developing a fully automated model for pixel-wise polyp segmentation, we propose ResUNet++, which is an improved ResUNet architecture for colonoscopic image segmentation. Our experimental evaluations show that the suggested architecture produces good segmentation results on publicly available datasets. Furthermore, ResUNet++ significantly outperforms U-Net and ResUNet, two key state-of-the-art deep learning architectures, by achieving high evaluation scores with a dice coefficient of 81.33%, and a mean Intersection over Union (mIoU) of 79.27% for the Kvasir-SEG dataset and a dice coefficient of 79.55%, and a mIoU of 79.62% with CVC-612 dataset.
| Task | Dataset | Metric | Value | Model |
|---|---|---|---|---|
| Medical Image Segmentation | Kvasir-SEG | mean Dice | 0.8133 | ResUNet++ |
| Medical Image Segmentation | ETIS-LARIBPOLYPDB | mIoU | 0.7534 | ResUNet++ |
| Medical Image Segmentation | ETIS-LARIBPOLYPDB | mean Dice | 0.6364 | ResUNet++ |
| Medical Image Segmentation | CVC-VideoClinicDB | Dice | 0.8798 | ResUNet++ |
| Medical Image Segmentation | CVC-VideoClinicDB | Recall | 0.7749 | ResUNet++ |
| Medical Image Segmentation | CVC-VideoClinicDB | mIoU | 0.873 | ResUNet++ |
| Medical Image Segmentation | CVC-VideoClinicDB | precision | 0.6702 | ResUNet++ |
| Medical Image Segmentation | ASU-Mayo Clinic dataset | DSC | 0.8743 | ResUNet++ |
| Medical Image Segmentation | ASU-Mayo Clinic dataset | Precision | 0.4896 | ResUNet++ |
| Medical Image Segmentation | ASU-Mayo Clinic dataset | Recall | 0.6534 | ResUNet++ |
| Medical Image Segmentation | ASU-Mayo Clinic dataset | mIoU | 0.8569 | ResUNet++ |
| Medical Image Segmentation | KvasirCapsule-SEG | DSC | 0.9499 | ResUNet+ |
| Medical Image Segmentation | KvasirCapsule-SEG | mIoU | 0.9087 | ResUNet+ |
| Medical Image Segmentation | CVC-ClinicDB | mean Dice | 0.7955 | ResUNet++ |
| Semantic Segmentation | Kvasir-SEG | mDice | 0.8133 | ResUNet++ |
| Semantic Segmentation | Kvasir-SEG | mIoU | 0.7927 | ResUNet++ |
| 10-shot image generation | Kvasir-SEG | mDice | 0.8133 | ResUNet++ |
| 10-shot image generation | Kvasir-SEG | mIoU | 0.7927 | ResUNet++ |