Classification of artefacts in endoscopic images using deep neural network

Abstract

Early cancer diagnosis by endoscopy is a challenging and time challenging process in the medical field thus requiring endoscopists to first acquire substantial experience and good technique. In addition, the presence of artefacts like saturation, bubbles and blood among others during the endoscopic process, are often misinterpreted as lesions leading to the wrong diagnosis and treatment. Lately, we have witnessed how the intervention of medical imaging with convolution neural networks (CNN) have brought promising results in medical applications. Therefore, we have applied deep neural networks to detect and classify artefacts, which interfere with the diagnosis of gastric cancer. Training CNN models from scratch require considerable number of labelled dataset, which is not usually available in the medical field. Thus, we have performed data augmentation on the EAD 2019 and Kvasir-V2 dataset leading to a total of 9852 images for six classes of artefacts. We then applied transfer learning using three pretrained neural network architectures namely: InceptionV3, InceptionResNetV2 and VGG16. The weights of the models are updated accordingly. The models are enhanced using Adam Optimisation and by varying the learning rates. We achieved a testing accuracy of 68.15 % with the original dataset trained by the InceptionResnetV2 model and 77.65% with the augmented dataset trained by the InceptionV3 models. Our experiments show the effectiveness of using CNN to detect artifacts during endoscopic procedures.