IRv2-Net: A Deep Learning Framework for Enhanced Polyp Segmentation Performance Integrating InceptionResNetV2 and UNet Architecture with Test Time Augmentation Techniques
Author | Ahamed, Md Faysal |
Author | Syfullah, Md Khalid |
Author | Sarkar, Ovi |
Author | Islam, Md Tohidul |
Author | Nahiduzzaman, Md |
Author | Islam, Md Rabiul |
Author | Khandakar, Amith |
Author | Ayari, Mohamed Arselene |
Author | Chowdhury, Muhammad E.H. |
Available date | 2024-04-22T09:28:34Z |
Publication Date | 2023-09-07 |
Publication Name | Sensors |
Identifier | http://dx.doi.org/10.3390/s23187724 |
Citation | Ahamed, M. F., Syfullah, M. K., Sarkar, O., Islam, M. T., Nahiduzzaman, M., Islam, M. R., ... & Chowdhury, M. E. (2023). Irv2-net: A deep learning framework for enhanced polyp segmentation performance integrating inceptionresnetv2 and unet architecture with test time augmentation techniques. Sensors, 23(18), 7724. |
Abstract | Colorectal polyps in the colon or rectum are precancerous growths that can lead to a more severe disease called colorectal cancer. Accurate segmentation of polyps using medical imaging data is essential for effective diagnosis. However, manual segmentation by endoscopists can be time-consuming, error-prone, and expensive, leading to a high rate of missed anomalies. To solve this problem, an automated diagnostic system based on deep learning algorithms is proposed to find polyps. The proposed IRv2-Net model is developed using the UNet architecture with a pre-trained InceptionResNetV2 encoder to extract most features from the input samples. The Test Time Augmentation (TTA) technique, which utilizes the characteristics of the original, horizontal, and vertical flips, is used to gain precise boundary information and multi-scale image features. The performance of numerous state-of-the-art (SOTA) models is compared using several metrics such as accuracy, Dice Similarity Coefficients (DSC), Intersection Over Union (IoU), precision, and recall. The proposed model is tested on the Kvasir-SEG and CVC-ClinicDB datasets, demonstrating superior performance in handling unseen real-time data. It achieves the highest area coverage in the area under the Receiver Operating Characteristic (ROC-AUC) and area under Precision-Recall (AUC-PR) curves. The model exhibits excellent qualitative testing outcomes across different types of polyps, including more oversized, smaller, over-saturated, sessile, or flat polyps, within the same dataset and across different datasets. Our approach can significantly minimize the number of missed rating difficulties. Lastly, a graphical interface is developed for producing the mask in real-time. The findings of this study have potential applications in clinical colonoscopy procedures and can serve based on further research and development. |
Sponsor | The publication of this article is supported by Qatar National Library (QNL). |
Language | en |
Publisher | Multidisciplinary Digital Publishing Institute (MDPI) |
Subject | colonoscopy CVC-ClinicDB IRv2-Net Kvasir-SEG polyps segmentation test time augmentation |
Type | Article |
Issue Number | 18 |
Volume Number | 23 |
ESSN | 1424-8220 |
Files in this item
This item appears in the following Collection(s)
-
Civil & Architectural Engineering [774 items ]
-
Electrical Engineering [2611 items ]
-
Technology Innovation and Engineering Education Unit [63 items ]