Meta-Polyp: a baseline for efficient Polyp segmentation
About
In recent years, polyp segmentation has gained significant importance, and many methods have been developed using CNN, Vision Transformer, and Transformer techniques to achieve competitive results. However, these methods often face difficulties when dealing with out-of-distribution datasets, missing boundaries, and small polyps. In 2022, Meta-Former was introduced as a new baseline for vision, which not only improved the performance of multi-task computer vision but also addressed the limitations of the Vision Transformer and CNN family backbones. To further enhance segmentation, we propose a fusion of Meta-Former with UNet, along with the introduction of a Multi-scale Upsampling block with a level-up combination in the decoder stage to enhance the texture, also we propose the Convformer block base on the idea of the Meta-former to enhance the crucial information of the local feature. These blocks enable the combination of global information, such as the overall shape of the polyp, with local information and boundary information, which is crucial for the decision of the medical segmentation. Our proposed approach achieved competitive performance and obtained the top result in the State of the Art on the CVC-300 dataset, Kvasir, and CVC-ColonDB dataset. Apart from Kvasir-SEG, others are out-of-distribution datasets. The implementation can be found at: https://github.com/huyquoctrinh/MetaPolyp-CBMS2023.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Polyp Segmentation | Kvasir-SEG (test) | mIoU0.921 | 116 | |
| Skin Lesion Segmentation | ISIC 2018 | Dice Coefficient90.41 | 94 | |
| Polyp Segmentation | CVC-ColonDB | mDice93.53 | 90 | |
| 2D skin lesion segmentation | ISIC 2017 | mIoU79.88 | 60 | |
| Skin Lesion Segmentation | ISIC 2016 | Dice Score (D)90.23 | 40 | |
| Polyp Segmentation | CVC-300 | Jaccard Index (J)87.85 | 9 | |
| Thyroid Nodule Segmentation | DDTI 83 (test) | Jaccard Index80.76 | 9 | |
| Breast Lesion Segmentation | BUSI (test) | Jaccard Index75.97 | 9 |