A deep CNN model for anomaly detection and localization in wireless capsule endoscopy images
Jazyk angličtina Země Spojené státy americké Médium print-electronic
Typ dokumentu časopisecké články, práce podpořená grantem
PubMed
34455302
DOI
10.1016/j.compbiomed.2021.104789
PII: S0010-4825(21)00583-7
Knihovny.cz E-zdroje
- Klíčová slova
- Anomaly detection, Attention mechanism, Deep convolutional neural network, Localization, Wireless capsule endoscopy,
- MeSH
- algoritmy MeSH
- kapslová endoskopie * MeSH
- neuronové sítě MeSH
- počítačové zpracování obrazu MeSH
- ROC křivka MeSH
- strojové učení MeSH
- Publikační typ
- časopisecké články MeSH
- práce podpořená grantem MeSH
Wireless capsule endoscopy (WCE) is one of the most efficient methods for the examination of gastrointestinal tracts. Computer-aided intelligent diagnostic tools alleviate the challenges faced during manual inspection of long WCE videos. Several approaches have been proposed in the literature for the automatic detection and localization of anomalies in WCE images. Some of them focus on specific anomalies such as bleeding, polyp, lesion, etc. However, relatively fewer generic methods have been proposed to detect all those common anomalies simultaneously. In this paper, a deep convolutional neural network (CNN) based model 'WCENet' is proposed for anomaly detection and localization in WCE images. The model works in two phases. In the first phase, a simple and efficient attention-based CNN classifies an image into one of the four categories: polyp, vascular, inflammatory, or normal. If the image is classified in one of the abnormal categories, it is processed in the second phase for the anomaly localization. Fusion of Grad-CAM++ and a custom SegNet is used for anomalous region segmentation in the abnormal image. WCENet classifier attains accuracy and area under receiver operating characteristic of 98% and 99%. The WCENet segmentation model obtains a frequency weighted intersection over union of 81%, and an average dice score of 56% on the KID dataset. WCENet outperforms nine different state-of-the-art conventional machine learning and deep learning models on the KID dataset. The proposed model demonstrates potential for clinical applications.
Citace poskytuje Crossref.org