Gastrointestinal Endoscopic Image Classification using Transfer Learning

Surajsingh Dookhee
DOI: https://doi.org/10.1109/SCC59637.2023.10527487
2023-12-01
Abstract:The diagnosis of gastrointestinal abnormalities is time-consuming as the wireless capsule endoscopy transmits an extensive number of images of the digestive tract. In this context, convolutional neural networks can be an efficient solution to detect early signs of intestinal diseases such as polyps and ulcerative colitis. This can also be helpful to enhance patients’ treatment progress and make informed decisions about surgical planning. However, most of the studies focus on performance but do not address the problem of overfitting. Therefore, the aim of this study is to propose a method that demonstrates a higher validation accuracy than training accuracy. The performance of 11 pre-trained convolutional neural networks was evaluated on the Kvasir v2 dataset to classify 8,000 endoscopic images into 8 classes. Data augmentation method was adopted, and results showed that DenseNet169 achieved the highest performance of 87.25% when trained with Adam optimiser. The proposed method generalizes well on the validation data and could be used to improve the task of medical professionals and effectively monitor abnormalities.
Computer Science,Medicine
What problem does this paper attempt to address?