banner

Deep learning-based uncertainty estimation for accurate lung ultrasound image analysis

Anjelin Genifer Edward Thomas, J. Shiny Duela

Abstract


Lung ultrasound imaging has become an important diagnostic tool for various respiratory conditions. Deep learning models have shown impressive results in classifying abnormalities in lung ultrasound images. However, these models typically provide deterministic predictions, disregarding the inherent uncertainty inherent in medical image analysis. This research paper introduces a novel approach to quantify uncertainty in deep learning models for accurate lung ultrasound image analysis. The proposed framework leverages a unique combination of Monte Carlo Dropout and Bayesian neural networks to provide reliable uncertainty estimates. By integrating these techniques, the model gains the ability to capture and represent the inherent uncertainty associated with medical image analysis. Extensive experiments conducted on a diverse dataset demonstrate the effectiveness and novelty of this approach. The inclusion of uncertainty estimation enhances classification accuracy and decision-making processes in lung ultrasound-based diagnosis, setting a new standard for the application of deep learning in medical image analysis. The novel methodology presented in this study has the potential to foster greater trust in AI-based diagnostic tools, promoting their integration into clinical practice and ultimately improving patient care and outcomes.


Keywords


lung ultrasound; deep learning; uncertainty estimation; Bayesian neural networks; Monte Carlo Dropout; medical image analysis; classification accuracy; image classification; uncertainty quantification; artificial intelligence

Full Text:

PDF

References


1. Gonzalez RC, Woods RE. Digital Image Processing. Prentice Hall, 2nd ed. 2002.

2. Born J, Beymer D, Rajan D, et al. On the role of artificial intelligence in medical imaging of COVID-19. Patterns. 2021; 2(8): 100330. doi: 10.1016/j.patter.2021.100330

3. van Amersfoort J, Smith L, Teh YW, Gal Y. Uncertainty Estimation Using a Single Deep Deterministic Neural Network. 2020.

4. Lakshminarayanan B, Pritzel A, Blundell C. Simple and Scalable Predictive Uncertainty Estimation using Deep Ensembles. ArXiv, abs/1612.01474. 2016.

5. Oberdiek P, Fink GA, Rottmann M. UQ-GAN: A Unified Model for Uncertainty Quantification of Deep Classifiers trained via Conditional GANs. ArXiv, abs/2201.13279. 2022.

6. Sarma R, Gupta YK. A comparative study of new and existing segmentation techniques. IOP Conference Series: Materials Science and Engineering. 2021; 1022(1): 012027. doi: 10.1088/1757-899x/1022/1/012027

7. Abdar M, Fahami MA, Rundo L, et al. Hercules: Deep Hierarchical Attentive Multilevel Fusion Model with Uncertainty Quantification for Medical Image Classification. IEEE Transactions on Industrial Informatics. 2023; 19(1): 274-285. doi: 10.1109/tii.2022.3168887

8. Selva Mary G, Blesswin AJ, Kumar SM. Self-authentication Model to Prevent Cheating Issues in Grayscale Visual Secret Sharing Schemes. Wireless Personal Communications. 2022; 125(2): 1695-1714. doi: 10.1007/s11277-022-09628-8

9. Dolezal JM, Srisuwananukorn A, Karpeyev D, et al. Uncertainty-informed deep learning models enable high-confidence predictions for digital histopathology. Nature Communications. 2022; 13(1). doi: 10.1038/s41467-022-34025-x

10. de Bruijne M, Cattin PC, Cotin S, et al. (editors). Medical Image Computing and Computer Assisted Intervention – MICCAI 2021. Springer International Publishing; 2021. doi: 10.1007/978-3-030-87199-4

11. Blesswin AJ, Mary GS, Kumar SMM. Multiple Secret Image Communication Using Visual Cryptography. Wireless Personal Communications. 2022; 122: 3085–3103. doi: 10.1007/s11277-021-09041-7.

12. Ghosal S, Xie A, Shah P. Uncertainty Quantified Deep Learning for Predicting Dice Coefficient of Digital Histopathology Image Segmentation. ArXiv, abs/2109.00115. 2021.

13. Ghoshal B, Tucker A, Sanghera B, et al. Estimating uncertainty in deep learning for reporting confidence to clinicians in medical image segmentation and diseases detection. Computational Intelligence. 2020; 37(2): 701-734. doi: 10.1111/coin.12411

14. Wang Q, Blesswin A J, Manoranjitham T, et al. Securing image-based document transmission in logistics and supply chain management through cheating-resistant visual cryptographic protocols. Mathematical Biosciences and Engineering. 2023; 20(11): 19983-20001. doi: 10.3934/mbe.2023885

15. Blesswin AG, Mary GS, Kumar SM. Secured Communication Method using Visual Secret Sharing Scheme for Color Images”, Journal of Internet Technology. 2021; 22(4): 803-810.

16. Laves MH, Ihler S, Fast JF, et al. Recalibration of Aleatoric and EpistemicRegression Uncertainty in Medical Imaging. Machine Learning for Biomedical Imaging. 2021; 1(MIDL 2020): 1-26. doi: 10.59275/j.melba.2021-a6fd

17. Kuang Z, Yan Z, Yu L, et al. Uncertainty-Aware Deep Learning with Cross-Task Supervision for PHE Segmentation on CT Images. IEEE Journal of Biomedical and Health Informatics. 2022; 26(6): 2615-2626. doi: 10.1109/jbhi.2021.3137603

18. Bhat I, Kuijf HJ, Cheplygina V, et al. Using Uncertainty Estimation to Reduce False Positives in Liver Lesion Detection. 2021 IEEE 18th International Symposium on Biomedical Imaging (ISBI). Published online April 13, 2021. doi: 10.1109/isbi48211.2021.9434119

19. Ghesu FC, Georgescu B, Mansoor A, et al. Quantifying and leveraging predictive uncertainty for medical image assessment. Medical Image Analysis. 2021; 68: 101855. doi: 10.1016/j.media.2020.101855

20. Selva Mary G, Manoj Kumar S. Secure grayscale image communication using significant visual cryptography scheme in real time applications. Multimedia Tools and Applications. 2019; 79(15-16): 10363-10382. doi: 10.1007/s11042-019-7202-7

21. Blesswin AJ, Visalakshi P. A Novel Visual Image Confirmation (VIC) Protocol Using Visual Cryptography for Securing Ubiquitous Bluetooth Mobile Communications. Research Journal of Applied Sciences. 2024; 9(8): 503-510. doi: 10.36478/rjasci.2014.503.510

22. Selva Mary G, Manoj Kumar S. A self-verifiable computational visual cryptographic protocol for secure two-dimensional image communication. Measurement Science and Technology. 2019; 30(12): 125404. doi: 10.1088/1361-6501/ab2faa

23. Gawlikowski J, Tassi CRN, Ali M, et al. A survey of uncertainty in deep neural networks. Artificial Intelligence Review. 2023; 56(S1): 1513-1589. doi: 10.1007/s10462-023-10562-9

24. Hosseiny B, Mahdianpari M, Hemati M, et al. Beyond Supervised Learning in Remote Sensing: A Systematic Review of Deep Learning Approaches. IEEE Journal of Selected Topics in Applied Earth Observations and Remote Sensing. 2024; 17: 1035-1052. doi: 10.1109/jstars.2023.3316733

25. Blesswin AJ, Visalakshi P. Optimal Visual Secret Sharing on Electrocardiography Images for Medical Secret Communications. International Journal of Control Theory and Applications. 2016; 9(2): 1055-1062.

26. Blesswin AJ, Mary GS. Optimal Grayscale Visual Cryptography using Error Diffusion to Secure Image Communication. International Journal of Control Theory and Applications. 2015; 8(4): 1511-1519.

27. Zhou ZH. A brief introduction to weakly supervised learning. National Science Review. 2017; 5(1): 44-53. doi: 10.1093/nsr/nwx106

28. Selva M, John B, Mithra V, et al. Enhancing conversational sentimental analysis for psychological depression prediction with Bi-LSTM. Journal of Autonomous Intelligence. 7(1).




DOI: https://doi.org/10.32629/jai.v7i5.1562

Refbacks

  • There are currently no refbacks.


Copyright (c) 2024 Anjelin Genifer Edward Thomas, J. Shiny Duela

License URL: https://creativecommons.org/licenses/by-nc/4.0/