Stepwise Feature Fusion: Local Guides Global
About
Colonoscopy, currently the most efficient and recognized colon polyp detection technology, is necessary for early screening and prevention of colorectal cancer. However, due to the varying size and complex morphological features of colonic polyps as well as the indistinct boundary between polyps and mucosa, accurate segmentation of polyps is still challenging. Deep learning has become popular for accurate polyp segmentation tasks with excellent results. However, due to the structure of polyps image and the varying shapes of polyps, it easy for existing deep learning models to overfitting the current dataset. As a result, the model may not process unseen colonoscopy data. To address this, we propose a new State-Of-The-Art model for medical image segmentation, the SSFormer, which uses a pyramid Transformer encoder to improve the generalization ability of models. Specifically, our proposed Progressive Locality Decoder can be adapted to the pyramid Transformer backbone to emphasize local features and restrict attention dispersion. The SSFormer achieves statet-of-the-art performance in both learning and generalization assessment.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Polyp Segmentation | CVC-ClinicDB (test) | DSC92.88 | 196 | |
| Polyp Segmentation | Kvasir | Dice Score93.6 | 128 | |
| Medical Image Segmentation | BUSI (test) | Dice78.76 | 121 | |
| Polyp Segmentation | ETIS | Dice Score79.6 | 108 | |
| Medical Image Segmentation | ISIC 2018 | Dice Score92.42 | 92 | |
| Polyp Segmentation | Kvasir-SEG (test) | mIoU0.7348 | 87 | |
| Polyp Segmentation | CVC-ClinicDB | Dice Coefficient94.5 | 81 | |
| Multi-organ Segmentation | Synapse multi-organ CT (test) | DSC78.01 | 81 | |
| Medical Image Segmentation | Kvasir-SEG (test) | mIoU89.05 | 78 | |
| Polyp Segmentation | ColonDB | mDice77.2 | 74 |