Abstract
This work presents a smart healthcare system for the detection of various abnormalities present in the gastrointestinal (GI) region with the help of time–frequency analysis and convolutional neural network. In this regard, the KVASIR V2 dataset comprising of eight classes of GI-tract images such as Normal cecum, Normal pylorus, Normal Z-line, Esophagitis, Polyps, Ulcerative Colitis, Dyed and lifted polyp, and Dyed resection margins are used for training and validation. The initial phase of the work involves an image pre-processing step, followed by the extraction of approximate discrete wavelet transform coefficients. Each class of decomposed images is later given as input to a couple of considered convolutional neural network (CNN) models for training and testing in two different classification levels to recognize its predicted value. Afterward, the classification performance is measured through the following measuring indices: accuracy, precision, recall, specificity, and F1 score. The experimental result shows 97.25% and 93.75% of accuracy in the first level and second level of classification, respectively. Lastly, a comparative performance analysis is carried out with several other previously published works on a similar dataset where the proposed approach performs better than its contemporary methods.
Graphic Abstract
Similar content being viewed by others
Abbreviations
- GI:
-
Gastrointestinal
- CNN:
-
Convolutional neural network
- AI:
-
Artificial intelligence
- WT:
-
Wavelet transform
- 2D-DWT:
-
Two-dimensional discrete wavelet transform
- TF:
-
Time–frequency
- CWT:
-
Continuous wavelet transform
- α :
-
Scaling parameter
- t :
-
Translation parameter
- \(A\left( n \right)\) :
-
Approximate coefficients
- \(D\left( n \right)\) :
-
Detail coefficients
- \(g(n)\) :
-
Low-pass filter
- \(h(n)\) :
-
High-pass filter
- MRA:
-
Multi-resolution analysis
- L:
-
Low pass
- H:
-
High pass
- ReLU:
-
Rectified linear unit
- L(α):
-
Loss function
- A:
-
Accuracy
- R:
-
Recall
- P:
-
Precision
- F1:
-
F1 Score
- S:
-
Specificity
- TD:
-
Truly detected
- FD:
-
Falsely detected
- IPP:
-
Image pre-processing
- BF:
-
Baseline features
- ANN:
-
Artificial neural network
- SVM:
-
Support vector machine
- RF:
-
Random forest
- BMFA:
-
Bidirectional marginal Fisher analysis
References
Tandon R (2007) Progress of gastroenterology in India. Indian J Gastroenterol 26:S31–S34
Cogan T, Cogan M, Tamil L (2019) MAPGI: accurate identification of anatomical landmarks and diseased tissue in gastrointestinal tract using deep learning. Comput Biol Med 111:103351
Kasban H, El-Bendary MAM, Salama DH (2015) A comparative study of medical imaging techniques. Int J Inf Sci Intell Syst 4(2):37–58
https://www.databridgemarketresearch.com/news/global-surgical-endoscopes-market, Accessed 25 Aug 2010
Watanabe K, Nagata N, Shimbo T, Nakashima R, Furuhata E, Sakurai T, Akazawa N, Yokoi C, Kobayakawa M, Akiyama J, Mizokami M, Uemura N (2013) Accuracy of endoscopic diagnosis of Helicobacter pylori infection according to level of endoscopic experience and the effect of training. BMC Gastroenterol 13(128):1–7
Hoogenboom SA, Bagci U, Wallace MB (2019) AI in gastroenterology. The current state of play and the potential. How will it affect our practice and when? Tech Gastrointest Endosc 22:42–47
Mohapatra S, Swarnkar T, Das J (2020) Deep convolutional neural network in medical image processing. In: Balas VE, Mishra BK, Kumar R (eds) Handbook of Deep Learning in Biomedical Engineering. Academic Press, pp 25–60. https://doi.org/10.1016/B978-0-12-823014-5.00006-5
S. Mohapatra, and T. Swarnkar, "Artificial Intelligence for Smart Healthcare Management: Brief Study," In Intelligent and Cloud Computing, pp. 365–373, Springer, Singapore, 2019.
Alagappan M, Brown JRG, Mori Y, Berzin TM (2018) Artificial intelligence in gastrointestinal endoscopy: The future is almost here. World J Gastrointest Endoscopy 10(10):239–249
Itoh T, Kawahira H, Nakashima H, Yata N (2018) Deep learning analyzes Helicobacter pylori infection by upper gastrointestinal endoscopy images. Endosc Int open 6(2):E139–E144
Zhu Y, Wang QC, Xu MD, Zhang Z, Cheng J, Zhong YS, Zhang YQ, Chen WF, Yao LQ, Zhou PH, Li QL (2019) Application of convolutional neural network in the diagnosis of the invasion depth of gastric cancer based on conventional endoscopy. Gastrointest Endosc 89(4):806–815
Hirasawa T, Aoyama K, Tanimoto T, Ishihara S, Shichijo S, Ozawa T, Ohnishi T, Fujishiro M, Matsuo K, Fujisaki J, Tada T (2018) Application of artificial intelligence using a convolutional neural network for detecting gastric cancer in endoscopic images. Gastric Cancer 21(4):653–660
Shichijo S, Nomura S, Aoyama K, Nishikawa Y, Miura M, Shinagawa T, Takiyama H, Tanimoto T, Ishihara S, Matsuo K, Tada T (2017) Application of convolutional neural networks in the diagnosis of Helicobacter pylori infection based on endoscopic images. EBioMedicine 25:106–111
Pogorelov K, Ostroukhova O, Jeppsson M, Espeland H, Griwodz C, de Lange T, Johansen D, Riegler M, Halvorsen P (2018) Deep learning and hand-crafted feature based approaches for polyp detection in medical videos. In: 2018 IEEE 31st International Symposium on Computer-Based Medical Systems (CBMS), pp. 381–386, IEEE, June 2018.
Bernal J, Aymeric H (2020) Miccai endoscopic vision challenge polyp detection and segmentation. https://endovissub2017-giana.grand-challenge.org/home/. Accessed 11 Aug 2020
Bernal J, S´anchez FJ, Fern´andez-Esparrach G, Gil D, Rodr´ıguez C, Vilari˜no F (2015) Wm-dova maps for accurate polyp highlighting in colonoscopy: Validation vs. saliency maps from physicians. Comput Med Imaging Graph 43:99–111
Pogorelov K, Randel KR, Griwodz C, Eskeland SL, de Lange T, Johansen D, Spampinato C, Dang-Nguyen D-T, Lux M, Schmidt PT, Riegler M, Halvorsen P (2017) KVASIR: a multi-class image dataset for computer aided gastrointestinal disease detection. In: Proc. of MMSYS, pp 164–169, June 2017
Pogorelov K, Randel KR, de Lange T, Eskeland SL, Griwodz C, Johansen D, Spampinato C, Taschwer M, Lux M, Schmidt PT, Riegler M, Halvorsen P (2017) Nerthus: A BOWEL preparation quality video dataset. In: Proc. of MMSYS, pp. 170–174, June 2017
Asperti A, Mastronardo C (2017) The effectiveness of data augmentation for detection of gastrointestinal diseases from endoscopical images,” 1712.03689, Dec 2017.
Hicks S, Riegler M, Pogorelov K, Anonsen KV, de Lange T, Johansen D, Jeppsson M, Randel KR, Eskeland SL, Halvorsen P (2018) Dissecting deep neural networks for better medical image classification and classification understanding. In: 2018 IEEE 31st International Symposium on computer-based medical systems, pp 363–368, June 2018
Ghatwary N, Ye X, Zolgharni M (2019) Esophageal abnormality detection using densenet based faster R-CNN with gabor features. IEEE Access 7:84374–84385
Sub-Challenge Early Barrett's Cancer Detection, https://endovissub-barrett.grand-challenge.org. Accessed 11 Aug 2020
Owais M, Arsalan M, Choi J, Mahmood T, Park KR (2019) Artificial intelligence-based classification of multiple gastrointestinal diseases using endoscopy videos for clinical diagnosis. J Clin Med 8(7):986
Gastrolab—The Gastrointestinal Site, http://www.gastrolab.net/ni.htm. Accessed 24 Aug 2020
Majid A, Khan MA, Yasmin M, Rehman A, Yousafzai A, Tariq U (2020) Classification of stomach infections: a paradigm of convolutional neural network along with classical features fusion and selection. Microsc Res Tech 83(5):562–576
Öztürk S, Özkaya U (2020) Gastrointestinal tract classification using improved LSTM based CNN. Multimed Tools Appl 79(39):28825–28840
Gopi VP, Palanisamy P (2011) Endoscopic image compression based on Double Density Discrete Wavelet Transform and SPIHT coding. In: 2011 IEEE International Conference on Control System, Computing and Engineering, pp 466–471, Nov 2011
Bonnel J, Khademi A, Krishnan S, Ioana C (2019) Small bowel image classification using cross-co-occurrence matrices on wavelet domain. Biomed Signal Process Control 4(1):7–15
Barbosa DC, Roupar DB, Ramos JC, Tavares AC, Lima CS (2012) Automatic small bowel tumor diagnosis by using multi-scale wavelet-based analysis in wireless capsule endoscopy images. BiomedEng Online 11:3–20
Xue Y, Li N, Wei X, Wan RA, Wang C (2020) Deep learning-based earlier detection of esophageal cancer using improved empirical wavelet transform from endoscopic image. IEEE Access 8:123765–123772
Billah M, Waheed S, Rahman MM (2017) An automatic gastrointestinal polyp detection system in video endoscopy using fusion of color wavelet and convolutional neural network features. Int J Biomed Imaging 2017:1–9. https://doi.org/10.1155/2017/9545920
Liu X, Gu J, Xie Y, Xiong J, Qin W (2012) A new approach to detecting ulcer and bleeding in Wireless capsule endoscopy images. In: Proceedings of 2012 IEEE-EMBS International Conference on biomedical and health informatics, pp. 737–740, IEEE, Jan 2012
Borgli H, Thambawita V, Smedsrud PH, Hicks S, Jha D, Eskeland SL, Ranheim Randel K et al (2020) HyperKvasir, a comprehensive multi-class image and video dataset for gastrointestinal endoscopy. Sci Data 7(1):1–14
Vetterli M, Herley C (1992) Wavelets and filter banks: Theory and design. IEEE Trans Signal Process 40:2207–2232
Sinha SK, Tiwari LK (2018) Enhancement of image classification for forest encroachment mapping with destriped SWIR band in the wavelet domain. IEEE J Sel Top Appl Earth Observ Remote Sens 11(7):2276–2281
Nayak DR, Dash R, Majhi B (2016) BrFn MR image classification using two-dimensional discrete wavelet transform and AdaBoost with random forests. Neurocomputing 177:188–197
Si L, Xiong X, Wang Z, Tan C (2020) A deep convolutional neural network model for intelligent discrimination between coal and rocks in coal mining face. Math Probl Eng 2020:1–12. https://doi.org/10.1155/2020/2616510
Liu Y, Gu Z, Cheung WK (2017) HKBU at MediaEval 2017 medico: medical multimedia task. In: MediaEval, Sept 2017
Petscharnig S, Schöffmann K, Lux M (2017) An Inception-like CNN Architecture for GI Disease and Anatomical Landmark Classification. In: MediaEval, Sept 2017
Agrawal T, Gupta R, Sahu S, Espy-Wilson CY (2017) SCL-UMD at the medico task-MediaEval 2017: transfer learning based classification of medical images. In: MediaEval, Sept 2017
KahsayGebreslassie A, Hagos Mt (2019) Automated gastrointestinal disease recognition for endoscopic images. In: IEEE International Conference on Computing, Communication, and Intelligent Systems (ICCCIS), pp 312–316, Oct 2019
Author information
Authors and Affiliations
Corresponding author
Ethics declarations
Conflict of interest
On behalf of all authors, the corresponding author states that there is no conflict of interest.
Rights and permissions
About this article
Cite this article
Mohapatra, S., Nayak, J., Mishra, M. et al. Wavelet Transform and Deep Convolutional Neural Network-Based Smart Healthcare System for Gastrointestinal Disease Detection. Interdiscip Sci Comput Life Sci 13, 212–228 (2021). https://doi.org/10.1007/s12539-021-00417-8
Received:
Revised:
Accepted:
Published:
Issue Date:
DOI: https://doi.org/10.1007/s12539-021-00417-8