Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

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.

Quoc-Huy Trinh• 2023

Related benchmarks

TaskDatasetResultRank
Polyp SegmentationKvasir-SEG (test)
mIoU0.921
116
Skin Lesion SegmentationISIC 2018
Dice Coefficient90.41
94
Polyp SegmentationCVC-ColonDB
mDice93.53
90
2D skin lesion segmentationISIC 2017
mIoU79.88
60
Skin Lesion SegmentationISIC 2016
Dice Score (D)90.23
40
Polyp SegmentationCVC-300
Jaccard Index (J)87.85
9
Thyroid Nodule SegmentationDDTI 83 (test)
Jaccard Index80.76
9
Breast Lesion SegmentationBUSI (test)
Jaccard Index75.97
9
Showing 8 of 8 rows

Other info

Code

Follow for update