Automatic Polyp Segmentation with Multiple Kernel Dilated Convolution Network

MKDCNet: Multitple Kernel Dilated Convolution Network

The detection and removal of precancerous polyps through colonoscopy is the primary technique for the prevention of colorectal cancer worldwide. However, the miss rate of colorectal polyp varies significantly among the endoscopists. It is well known that a computer-aided diagnosis (CAD) system can assist endoscopists in detecting colon polyps and minimize the variation among endoscopists. In this study, we introduce a novel deep learning architecture, named {\textbf{MKDCNet}}, for automatic polyp segmentation robust to significant changes in polyp data distribution. MKDCNet is simply an encoder-decoder neural network that uses the pre-trained \textit{ResNet50} as the encoder and novel \textit{multiple kernel dilated convolution (MKDC)} block that expands the field of view to learn more robust and heterogeneous representation. Extensive experiments on four publicly available polyp datasets and cell nuclei dataset show that the proposed MKDCNet outperforms the state-of-the-art methods when trained and tested on the same dataset as well when tested on unseen polyp datasets from different distributions. With rich results, we demonstrated the robustness of the proposed architecture. From an efficiency perspective, our algorithm can process at ($\approx45$) frames per second on RTX 3090 GPU. MKDCNet can be a strong benchmark for building real-time systems for clinical colonoscopies.

Block digram of the proposed MKDCNet

Block diagram of the Multiple Kernel Dilated Convolution Network

Results (Quantitative results)

Qualitative result comparison of the model trained and tested on several medical image segmentation dataset

Results (Qualitative results and Heatmap)

Qualitative results comparison along with the heatmap on the Kvasir-SEG, BKAI-IGH, and 2018 Data Science Bowl datasets. The heatmaps provide insight into the intermediate feature maps from the multi scale feature fusion block. The heatmap shows the \textit{region of interest} and its statistical significance and the color intensity shows the \textit{effect}. The \textit{red} and \textit{yellow} colors denote the most significant feature and the \textit{blue} color denote the least significance feature.

Citation

Please cite our work if you find it useful.

@INPROCEEDINGS{tomarMKDCNet,
  author={N.Tomar and A. Srivastava and U. Bagci and D. Jha},
  title={Automatic Polyp Segmentation with Multiple Kernel Dilated Convolution Network}, 
  year={2022}}