HyperAIHyperAI

Command Palette

Search for a command to run...

5 months ago

Multi Kernel Positional Embedding ConvNeXt for Polyp Segmentation

Trong-Hieu Nguyen Mau; Quoc-Huy Trinh; Nhat-Tan Bui; Minh-Triet Tran; Hai-Dang Nguyen

Multi Kernel Positional Embedding ConvNeXt for Polyp Segmentation

Abstract

Medical image segmentation is the technique that helps doctor view and has a precise diagnosis, particularly in Colorectal Cancer. Specifically, with the increase in cases, the diagnosis and identification need to be faster and more accurate for many patients; in endoscopic images, the segmentation task has been vital to helping the doctor identify the position of the polyps or the ache in the system correctly. As a result, many efforts have been made to apply deep learning to automate polyp segmentation, mostly to ameliorate the U-shape structure. However, the simple skip connection scheme in UNet leads to deficient context information and the semantic gap between feature maps from the encoder and decoder. To deal with this problem, we propose a novel framework composed of ConvNeXt backbone and Multi Kernel Positional Embedding block. Thanks to the suggested module, our method can attain better accuracy and generalization in the polyps segmentation task. Extensive experiments show that our model achieves the Dice coefficient of 0.8818 and the IOU score of 0.8163 on the Kvasir-SEG dataset. Furthermore, on various datasets, we make competitive achievement results with other previous state-of-the-art methods.

Code Repositories

Benchmarks

BenchmarkMethodologyMetrics
medical-image-segmentation-on-endotect-polypPEFNet
DSC: 0.8565
mIoU: 0.7967
medical-image-segmentation-on-kvasir-segPEFNet
mIoU: 0.8163
mean Dice: 0.8818
polyp-segmentation-on-kvasir-segPEFNet
mDice: 0.8818
mIoU: 0.8163

Build AI with AI

From idea to launch — accelerate your AI development with free AI co-coding, out-of-the-box environment and best price of GPUs.

AI Co-coding
Ready-to-use GPUs
Best Pricing
Get Started

Hyper Newsletters

Subscribe to our latest updates
We will deliver the latest updates of the week to your inbox at nine o'clock every Monday morning
Powered by MailChimp
Multi Kernel Positional Embedding ConvNeXt for Polyp Segmentation | Papers | HyperAI