Bayesian Deep Learning Algorithms for Classifying COVID-19 Images

Authors: I. Oloyede

Abstract:

The study investigates the accuracy and loss of deep learning algorithms with the set of coronavirus (COVID-19) images dataset by comparing Bayesian convolutional neural network and traditional convolutional neural network in low dimensional dataset. 50 sets of X-ray images out of which 25 were COVID-19 and the remaining 20 were normal, twenty images were set as training while five were set as validation that were used to ascertained the accuracy of the model. The study found out that Bayesian convolution neural network outperformed conventional neural network at low dimensional dataset that could have exhibited under fitting. The study therefore recommended Bayesian Convolutional neural network (BCNN) for android apps in computer vision for image detection.

Keywords: BCNN, CNN, Images, COVID-19, Deep Learning.

References:

[1] A. Martín, A. Ashish, B. Paul, B. Eugene, C. Chifeng, C. Craig, S.C. Greg, D. Andy, D. Jeffrey, D. Matthieu, G. Sanjay, G. Lan, H. Andrew, L. Geoffrey, L. Michael, J. Rafal, J. Yangqing, K. Lukasz, K. Manjunath, L. Josh, M. Dan, S. Mike, M. Rajat, M. Sherry, M. Derek, O. Chris, S. Jonathon, S. Benoit, S. Ilya, T. Kunal, T. Paul, V. Vincent, V. Vijay, V. Fernanda, V. Oriol, W. Pete, W. Martin, Y. Yuan, and Z. Xiaoqiang, ‘TensorFlow: Large-scale machine learning on heterogeneous systems’. Software available from tensorflow.org, 2015.
[2] G. Yarin, I. Riashat and G. Zoubin, ‘Deep Bayesian Active Learning with Image Data’, Workshop on Bayesian Deep Learning, Neural Information Processing Systems, Barcelona, Spain, 2016.
[3] J. Schmidhuber, ‘Deep learning in neural networks: “An overview’, Neural Networks, Vol 61, pp 85-117, 2015.
[4] Y. Li. and Y. Liang, ‘Learning over parameterized neural networks via stochastic gradient descent on structured data,’ inadvances in Neural Information Processing Systems (NeurIPS), 2018.
[5] A. Kendall and Y. Gal, ‘What uncertainties Do We need in Bayesian deep learning for computer vision?,’ in Advances in Neural Information Processing Systems (NIPS), 2017.
[6] H. Jonathan and K. Nal, ‘Bayesian Inference for Large Scale Image Classification’, arXiv:1908.03491v1 (cs.LG), 2019.
[7] D. Giacomo, B. Christopher and Z. Xian, ‘Bayesian Neural Networks for Cellular Image Classification and Uncertainty Analysis’, bioRxiv preprint doi: https://doi.org/10.1101/824862, 2020.
[8] T. Toan, P. Trung, C. Gustavo, P. Lyle and R. Lan, ‘A Bayesian Data Augmentation Approach for Learning Deep Models’, 31st Conference on Neural Information Processing Systems, Long Beach, CA, USA, 2017.
[9] F.K. Chen, R. D. Viljoen, and D.M. Bukowska,‘Classification of image artefacts in optical coherence tomography angiography of the choroid in macular diseases’. Clinical & Experimental Ophthalmology, 44(5), 388–399. doi:10.1111/ceo.12683 2015.
[10] M. Krasser, ‘Variational inference in Bayesian neural networks’. http://krasserm.github.io/2019/03/14/bayesian-neural-networks/ unpublished, 2019.
[11] M. Takashi, ‘Bayesian deep learning: A model-based interpretable approach nonlinear Theory and Its Applications’, IEICE, vol. 11, no. 1, pp. 16–35 c_ IEICE 2020 DOI: 10.1587/nolta.11.16
[12] R. G. Van and F. L. Drake Jr. Python reference manual. Centrum voor Wiskunde en Informatica Amsterdam. 1995
[13] Anaconda Software Distribution. Anaconda Documentation. Anaconda Inc. Retrieved from https://docs.anaconda.com/2020.