An Ensemble Framework of Deep Neural Networks for Colorectal Polyp Classification

Farah Younas,Muhammad Usman,Wei Qi Yan
DOI: https://doi.org/10.1007/s11042-022-14177-0
IF: 2.577
2023-01-01
Multimedia Tools and Applications
Abstract:Colorectal cancer (CRC) is caused by malignant polyps which must be resected and examined for accurate classification. Biopsy, the manual workflow of polyp classification is time-intensive task and requires an automated solution. The objective of this study is to develop an accurate virtual biopsy tool for polyp classification. Moreover, automated assessment of polyps is a challenging task due to the similarities in their patterns, and in contrast to existing studies on binary classification, the outcome of multi-class classification requires evaluation through advanced evaluation measures. The proposed method combined the strength of individual weak learner for an accurate weighted-average ensemble deep learning classification. At first, base-classifiers were pretrained on the ImageNet database. Second, an average ensemble was built and evaluated for enhancing the performance, an appropriate combination of weights was chosen through grid search and assigned to the models. The performance evaluation of the proposed method in terms of F1-micro (0.80), F1-macro (0.81), F1-weighted (0.84) metrics, model reliability using Cohen’s Kappa Coefficient (0.60) and Mathew Correlation Co-efficient value (0.49) for binary dataset shows the superiority over existing models. The higher rates of precision and recall show potential usage of the proposed system in the development of a virtual biopsy tool.
What problem does this paper attempt to address?