{"status":"ok","message-type":"work","message-version":"1.0.0","message":{"indexed":{"date-parts":[[2026,6,1]],"date-time":"2026-06-01T23:25:20Z","timestamp":1780356320140,"version":"3.54.1"},"reference-count":41,"publisher":"Frontiers Media SA","license":[{"start":{"date-parts":[[2024,9,18]],"date-time":"2024-09-18T00:00:00Z","timestamp":1726617600000},"content-version":"vor","delay-in-days":0,"URL":"https:\/\/creativecommons.org\/licenses\/by\/4.0\/"}],"content-domain":{"domain":["frontiersin.org"],"crossmark-restriction":true},"short-container-title":["Front. Artif. Intell."],"abstract":"<jats:p>This paper investigates uncertainty quantification (UQ) techniques in multi-class classification of chest X-ray images (COVID-19, Pneumonia, and Normal). We evaluate Bayesian Neural Networks (BNN) and the Deep Neural Network with UQ (DNN with UQ) techniques, including Monte Carlo dropout, Ensemble Bayesian Neural Network (EBNN), Ensemble Monte Carlo (EMC) dropout, across different evaluation metrics. Our analysis reveals that DNN with UQ, especially EBNN and EMC dropout, consistently outperform BNNs. For example, in Class 0 vs. All, EBNN achieved a <jats:italic>U<\/jats:italic>Acc of 92.6%, <jats:italic>U<\/jats:italic>AUC-ROC of 95.0%, and a Brier Score of 0.157, significantly surpassing BNN's performance. Similarly, EMC Dropout excelled in Class 1 vs. All with a <jats:italic>U<\/jats:italic>Acc of 83.5%, <jats:italic>U<\/jats:italic>AUC-ROC of 95.8%, and a Brier Score of 0.165. These advanced models demonstrated higher accuracy, better discriaminative capability, and more accurate probabilistic predictions. Our findings highlight the efficacy of DNN with UQ in enhancing model reliability and interpretability, making them highly suitable for critical healthcare applications like chest X-ray imageQ6 classification.<\/jats:p>","DOI":"10.3389\/frai.2024.1410841","type":"journal-article","created":{"date-parts":[[2024,9,18]],"date-time":"2024-09-18T05:10:29Z","timestamp":1726636229000},"update-policy":"https:\/\/doi.org\/10.3389\/crossmark-policy","source":"Crossref","is-referenced-by-count":11,"title":["Uncertainty quantification in multi-class image classification using chest X-ray images of COVID-19 and pneumonia"],"prefix":"10.3389","volume":"7","author":[{"given":"Albert","family":"Whata","sequence":"first","affiliation":[],"role":[{"vocabulary":"crossref","role":"author"}]},{"given":"Katlego","family":"Dibeco","sequence":"additional","affiliation":[],"role":[{"vocabulary":"crossref","role":"author"}]},{"given":"Kudakwashe","family":"Madzima","sequence":"additional","affiliation":[],"role":[{"vocabulary":"crossref","role":"author"}]},{"given":"Ibidun","family":"Obagbuwa","sequence":"additional","affiliation":[],"role":[{"vocabulary":"crossref","role":"author"}]}],"member":"1965","published-online":{"date-parts":[[2024,9,18]]},"reference":[{"key":"B1","unstructured":"Abadi\n              M.\n            \n            \n              Barham\n              P.\n            \n            \n              Chen\n              J.\n            \n            \n              Chen\n              Z.\n            \n            \n              Davis\n              A.\n            \n            \n              Dean\n              J.\n            \n          \n          Tensorflow: Large-scale machine learning on heterogeneous systems\n          \n          2015"},{"key":"B2","doi-asserted-by":"publisher","first-page":"243","DOI":"10.1016\/j.inffus.2021.05.008","article-title":"A review of uncertainty quantification in deep learning: techniques, applications and challenges","volume":"76","author":"Abdar","year":"","journal-title":"Inf. Fusion"},{"key":"B3","doi-asserted-by":"publisher","first-page":"364","DOI":"10.1016\/j.inffus.2022.09.023","article-title":"Uncertaintyfusenet: robust uncertainty-aware hierarchical feature fusion model with ensemble monte carlo dropout for COVID-19 detection","volume":"90","author":"Abdar","year":"2023","journal-title":"Inf. Fusion"},{"key":"B4","doi-asserted-by":"publisher","first-page":"104418","DOI":"10.1016\/j.compbiomed.2021.104418","article-title":"Uncertainty quantification in skin cancer classification using three-way decision-based Bayesian deep learning","volume":"135","author":"Abdar","year":"","journal-title":"Comput. Biol. Med"},{"key":"B5","doi-asserted-by":"publisher","first-page":"1519","DOI":"10.1007\/s11042-022-13269-1","article-title":"Uncertainty estimation based adversarial attack in multi-class classification","volume":"82","author":"Alarab","year":"2023","journal-title":"Multimed. Tools Appl"},{"key":"B6","doi-asserted-by":"publisher","first-page":"406","DOI":"10.30855\/gmbd.0705001","article-title":"Clahe based enhancement to transfer learning in COVID-19 detection","volume":"8","author":"Altan","year":"2022","journal-title":"Gazi m\u00fchendis. Bilim. Derg"},{"key":"B7","doi-asserted-by":"publisher","first-page":"815","DOI":"10.1038\/s41598-022-05052-x","article-title":"Objective evaluation of deep uncertainty predictions for COVID-19 detection","volume":"12","author":"Asgharnezhad","year":"2022","journal-title":"Sci. Rep"},{"key":"B8","doi-asserted-by":"crossref","first-page":"69","DOI":"10.1109\/EDiS57230.2022.9996478","article-title":"\u201cBayesian convolutional neural networks for image classification with uncertainty estimation,\u201d","volume-title":"2022 3rd International Conference on Embedded Distributed Systems (EDiS)","author":"Bessai-Mechmache","year":"2022"},{"key":"B9","doi-asserted-by":"publisher","first-page":"e0177678","DOI":"10.1371\/journal.pone.0177678","article-title":"Optimal classifier for imbalanced data using matthews correlation coefficient metric","volume":"12","author":"Boughorbel","year":"2017","journal-title":"PLoS ONE"},{"key":"B10","doi-asserted-by":"publisher","first-page":"1","DOI":"10.1175\/1520-0493(1950)078&lt;0001:VOFEIT&gt;2.0.CO;2","article-title":"Verification of forecasts expressed in terms of probability","volume":"78","author":"Brier","year":"1950","journal-title":"Mon. Weather Rev"},{"key":"B11","doi-asserted-by":"publisher","first-page":"10604","DOI":"10.48550\/arXiv.1711:10604","article-title":"Tensorflow distributions","volume":"1711","author":"Dillon","year":"2017","journal-title":"arXiv"},{"key":"B12","first-page":"1050","article-title":"\u201cDropout as a Bayesian approximation: representing model uncertainty in deep learning,\u201d","volume-title":"International conference on machine learning","author":"Gal","year":"2016"},{"key":"B13","doi-asserted-by":"publisher","DOI":"10.48550\/arXiv.2003.10769","article-title":"Estimating uncertainty and interpretability in deep learning for coronavirus (COVID-19) detection","author":"Ghoshal","year":"2020","journal-title":"arXiv"},{"key":"B14","first-page":"627","article-title":"Receiver operating characteristic (ROC) curve analysis for medical diagnostic test evaluation","volume":"4","author":"Hajian-Tilaki","year":"2013","journal-title":"Caspian J. Intern. Med"},{"key":"B15","doi-asserted-by":"crossref","first-page":"87","DOI":"10.1109\/ICRA40945.2020.9196544","article-title":"\u201cBayesod: a Bayesian approach for uncertainty estimation in deep object detectors,\u201d","volume-title":"2020 IEEE International Conference on Robotics and Automation (ICRA)","author":"Harakeh","year":"2020"},{"key":"B16","doi-asserted-by":"publisher","first-page":"632901","DOI":"10.3389\/fgene.2020.632901","article-title":"Integrating somatic mutations for breast cancer survival prediction using machine learning methods","volume":"11","author":"He","year":"2021","journal-title":"Front. Genet"},{"key":"B17","doi-asserted-by":"publisher","first-page":"1711","DOI":"10.1007\/s12559-021-09946-2","article-title":"Deep learning approach for early detection of Alzheimer's disease","volume":"14","author":"Helaly","year":"2022","journal-title":"Cogn. Comput"},{"key":"B18","doi-asserted-by":"publisher","first-page":"106597","DOI":"10.1016\/j.asoc.2020.106597","article-title":"Uncertainty quantification for plant disease detection using Bayesian deep learning","volume":"96","author":"Hern\u00e1ndez","year":"2020","journal-title":"Appl. Soft Comput"},{"key":"B19","doi-asserted-by":"publisher","first-page":"489","DOI":"10.5555\/3295222.3295309","article-title":"What uncertainties do we need in Bayesian deep learning for computer vision?","volume":"30","author":"Kendall","year":"2017","journal-title":"Adv. Neural Inf. Process. Syst"},{"key":"B20","author":"Khairnar","year":"2020"},{"key":"B21","doi-asserted-by":"publisher","DOI":"10.48550\/arXiv.1412.6980","article-title":"Adam: a method for stochastic optimization","author":"Kingma","year":"2014","journal-title":"arXiv"},{"key":"B22","doi-asserted-by":"publisher","first-page":"1","DOI":"10.1145\/3065386","article-title":"Imagenet classification with deep convolutional neural networks","volume":"25","author":"Krizhevsky","year":"2012","journal-title":"Adv. Neural Inf. Process. Syst"},{"key":"B23","doi-asserted-by":"publisher","first-page":"106816","DOI":"10.1016\/j.csda.2019.106816","article-title":"Uncertainty quantification using Bayesian neural networks in classification: application to biomedical image segmentation","volume":"142","author":"Kwon","year":"2020","journal-title":"Comput. Stat. Data Anal"},{"key":"B24","doi-asserted-by":"publisher","first-page":"bbad319","DOI":"10.1093\/bib\/bbad319","article-title":"Sequence-based prediction model of protein crystallization propensity using machine learning and two-level feature selection","volume":"24","author":"Le","year":"2023","journal-title":"Brief. Bioinform"},{"key":"B25","doi-asserted-by":"publisher","first-page":"39420","DOI":"10.1021\/acsomega.3c05074","article-title":"Optimizing hyperparameter tuning in machine learning to improve the predictive performance of cross-species n6-methyladenosine sites","volume":"8","author":"Le","year":"2023","journal-title":"ACS Omega"},{"key":"B26","doi-asserted-by":"publisher","first-page":"174","DOI":"10.1038\/s41746-022-00709-3","article-title":"Improving the repeatability of deep learning models with monte carlo dropout","volume":"5","author":"Lemay","year":"2022","journal-title":"NPJ Digit. Med"},{"key":"B27","doi-asserted-by":"publisher","DOI":"10.48550\/arXiv.1612.02295","article-title":"Large-margin softmax loss for convolutional neural networks","author":"Liu","year":"2016","journal-title":"arXiv"},{"key":"B28","doi-asserted-by":"publisher","first-page":"e2553","DOI":"10.1002\/env.2553","article-title":"Deep echo state networks with uncertainty quantification for spatio-temporal forecasting","volume":"30","author":"McDermott","year":"2019","journal-title":"Environmetrics"},{"key":"B29","doi-asserted-by":"publisher","first-page":"1","DOI":"10.1038\/s41598-021-84854-x","article-title":"Dropconnect is effective in modeling uncertainty of Bayesian deep networks","volume":"11","author":"Mobiny","year":"2021","journal-title":"Sci. Rep"},{"key":"B30","unstructured":"Impact of local histogram equalization on deep learning architectures for diagnosis of COVID-19 on chest X-rays\n          \n          \n            \n              Narl\u0131\n              S. S.\n            \n          \n          Manchester J. Artif. Intell. Appl. Sci\n          \n          2021"},{"key":"B31","volume-title":"Bayesian Learning for Neural Networks, Volume 118","author":"Neal","year":"2012"},{"key":"B32","doi-asserted-by":"crossref","unstructured":"Evaluation of classification models in machine learning\n          \n          39\n          46\n          \n            \n              Novakovi\u0107\n              J. D.\n            \n            \n              Veljovi\u0107\n              A.\n            \n            \n              Ili\u0107\n              S. S.\n            \n            \n              Papi\u0107\n              \u017d.\n            \n            \n              Milica\n              T.\n            \n          \n          Theory Appl. Math. Comput. Sci\n          7\n          2017","DOI":"10.18642\/ijamml_7100121858"},{"key":"B33","doi-asserted-by":"publisher","first-page":"297","DOI":"10.31661\/jbpe.v0i0.2109-1403","article-title":"Prediction of breast cancer using machine learning approaches","volume":"12","author":"Rabiei","year":"2022","journal-title":"J. Biomed. Phys. Eng."},{"key":"B34","doi-asserted-by":"publisher","first-page":"1","DOI":"10.1109\/TPAMI.2016.2577031","article-title":"Faster R-CNN: towards real-time object detection with region proposal networks","volume":"28","author":"Ren","year":"2015","journal-title":"Adv. Neural Inf. Process. Syst"},{"key":"B35","first-page":"815","article-title":"\u201cFacenet: a unified embedding for face recognition and clustering,\u201d","volume-title":"Proceedings of the IEEE conference on computer vision and pattern recognition","author":"Schroff","year":"2015"},{"key":"B36","doi-asserted-by":"publisher","first-page":"1455","DOI":"10.1109\/TBME.2015.2496264","article-title":"A dataset for breast cancer histopathological image classification","volume":"63","author":"Spanhol","year":"2015","journal-title":"IEEE Trans. Biomed. Eng"},{"key":"B37","unstructured":"Dropout: a simple way to prevent neural networks from overfitting\n          \n          1929\n          1958\n          \n            \n              Srivastava\n              N.\n            \n            \n              Hinton\n              G.\n            \n            \n              Krizhevsky\n              A.\n            \n            \n              Sutskever\n              I.\n            \n            \n              Salakhutdinov\n              R.\n            \n          \n          33259321\n          J. Mach. Learn. Res\n          15\n          2014"},{"key":"B38","doi-asserted-by":"publisher","first-page":"272","DOI":"10.3390\/a17070272","article-title":"An improved Adam's algorithm for stomach image classification","volume":"17","author":"Sun","year":"2024","journal-title":"Algorithms"},{"key":"B39","doi-asserted-by":"publisher","first-page":"815","DOI":"10.1109\/TMI.2021.3123300","article-title":"Explanation and use of uncertainty quantified by Bayesian neural network classifiers for breast histopathology images","volume":"41","author":"Thiagarajan","year":"2021","journal-title":"IEEE Trans. Med. Imaging"},{"key":"B40","doi-asserted-by":"crossref","first-page":"671","DOI":"10.1007\/978-3-030-86365-4_54","article-title":"\u201cUncertainty quantification and estimation in medical image classification,\u201d","volume-title":"Artificial Neural Networks and Machine Learning-ICANN 2021: 30th International Conference on Artificial Neural Networks, Bratislava, Slovakia, September 14-17, 2021, Proceedings, Part III 30","author":"Yang","year":"2021"},{"key":"B41","doi-asserted-by":"publisher","first-page":"841","DOI":"10.1007\/s13042-019-00990-x","article-title":"Deepsite: bidirectional lstm and cnn models for predicting DNA-protein binding","volume":"11","author":"Zhang","year":"2020","journal-title":"Int. J. Mach. Learn. Cybern"}],"container-title":["Frontiers in Artificial Intelligence"],"original-title":[],"link":[{"URL":"https:\/\/www.frontiersin.org\/articles\/10.3389\/frai.2024.1410841\/full","content-type":"unspecified","content-version":"vor","intended-application":"similarity-checking"}],"deposited":{"date-parts":[[2024,9,18]],"date-time":"2024-09-18T05:10:35Z","timestamp":1726636235000},"score":1,"resource":{"primary":{"URL":"https:\/\/www.frontiersin.org\/articles\/10.3389\/frai.2024.1410841\/full"}},"subtitle":[],"short-title":[],"issued":{"date-parts":[[2024,9,18]]},"references-count":41,"alternative-id":["10.3389\/frai.2024.1410841"],"URL":"https:\/\/doi.org\/10.3389\/frai.2024.1410841","relation":{},"ISSN":["2624-8212"],"issn-type":[{"value":"2624-8212","type":"electronic"}],"subject":[],"published":{"date-parts":[[2024,9,18]]},"article-number":"1410841"}}