Colorectal Cancer Segmentation with Adaptive Augmentation and Multi-Resolution Ensemble Models
Organizations: Graduate School of Informatics, METU, Ankara, Turkey
Abstract
Colorectal cancer (CRC) is the second most deadly and third most common cancer, and the leading cause of death among gastrointestinal cancers. Early diagnosis is crucial for the treatment of this cancer and increasing the survival rates. Although CRC is more common in developed regions, its occurrence is also increasing in developing regions as well. CRC diagnosis relies on histopathology assessment post-biopsy. Automated deep learning algorithms can significantly reduce diagnosis time, enhancing efficiency and supporting timely clinical decisions. We present an automated segmentation pipeline for whole-slide histopathology images that labels tumor grades 1-3 and normal mucosa. It utilizes dense prediction transformers with various encoder backbones, overlapping patches, and test-time augmentation. An adaptive augmentation policy, guided by large language models, further improves training. Top models were ensembled via soft voting, and mask refining post-processing steps, Gaussian blurring, morphological closing, and connected components analysis. On a colorectal cancer grade dataset, our method improved the F1 score from 62.92 to 69.84. Code is available here: github.com/caglarmert/ICIP2025
Figures & tables
| Baseline | Ours | |||||||
|---|---|---|---|---|---|---|---|---|
| Model | F1 | Precision | Recall | Model | F1 | Precision | Recall | |
| SegFormer | 45.07 | 56.44 | 50.52 | UNet | 56.79 | 60.20 | 54.70 | |
| DeepLabv3+ | 52.31 | 51.52 | 62.35 | FCN | 59.81 | 57.37 | 64.68 | |
| UNet | 54.23 | 50.59 | 64.88 | SegFormer | 65.60 | 63.91 | 68.76 | |
| ConvNext | 56.24 | 54.70 | 61.42 | DPT | 67.39 | 63.44 | 72.31 | |
| Swin | 62.92 | 60.94 | 69.63 | Ensemble | 69.73 | 68.63 | 71.14 | |
| Normal | Polyp |
| Method F1 Score IoU Precision Recall U-Net 41.1 26.3 58.6 32.8 Seg-Net 77.7 68.4 89.5 75.8 MedT 67.6 56.2 87.4 61.0 Ours-1 93.8 88.3 94.0 93.6 Ours-2 93.8 88.3 94.5 93.1 | Method F1 Score IoU Precision Recall U-Net 96.5 93.6 91.6 94.7 Seg-Net 93.7 88.6 91.0 96.5 MedT 77.1 64.3 68.7 92.0 Ours-1 96.2 92.6 95.9 96.4 Ours-2 96.6 93.4 96.5 96.6 |
| High-grade IN | Low-grade IN |
| Method F1 Score IoU Precision Recall U-Net 89.5 81.6 84.7 96.1 Seg-Net 89.4 81.2 88.1 91.3 MedT 82.4 70.7 74.0 93.8 Ours-1 93.1 87.1 91.0 95.3 Ours-2 94.1 88.8 93.9 94.3 | Method F1 Score IoU Precision Recall U-Net 91.1 84.9 87.9 95.3 Seg-Net 92.4 86.4 88.0 97.7 MedT 88.9 80.8 87.6 91.6 Ours-1 96.3 92.8 95.8 96.7 Ours-2 96.4 93.1 95.9 96.9 |
| Adenocarcinoma | Serrated adenoma |
| Method F1 Score IoU Precision Recall U-Net 88.7 80.8 85.0 95.0 Seg-Net 86.5 77.5 79.2 97.7 MedT 73.5 59.5 68.2 86.4 Ours-1 90.6 82.9 91.4 89.9 Ours-2 91.1 83.7 90.7 91.5 | Method F1 Score IoU Precision Recall U-Net 93.8 88.6 89.9 98.3 Seg-Net 90.7 83.2 85.9 96.3 MedT 67.0 50.9 89.6 54.4 Ours-1 93.8 88.2 94.9 92.6 Ours-2 95.5 91.4 96.7 94.3 |
| Category | Configuration | F1 Score | IoU | Precision | Recall |
|---|---|---|---|---|---|
| Loss Function | CE | 61.69 | 47.91 | 60.31 | 63.79 |
| Focal | 65.53 | 51.44 | 61.42 | 72.19 | |
| Lovasz | 65.91 | 51.77 | 64.95 | 70.49 | |
| Jaccard | 66.82 | 52.53 | 63.60 | 71.06 | |
| Tversky | 66.87 | 52.47 | 69.28 | 67.51 | |
| Dice | 67.39 | 53.15 | 63.44 | 72.31 |