A Deep Learning-Based Framework for Uncertainty Quantification in Medical Imaging Using the DropWeak Technique: An Empirical Study with Baresnet

Lung cancer is a leading cause of cancer-related deaths globally. Early detection is crucial for improving patient survival rates. Deep learning (DL) has shown promise in the medical field, but its accuracy must be evaluated, particularly in the context of lung cancer classification. In this study, we conducted uncertainty analysis on various frequently used DL architectures, including Baresnet, to assess the uncertainties in the classification results. This study focuses on the use of deep learning for the classification of lung cancer, which is a critical aspect of improving patient survival rates. The study evaluates the accuracy of various deep learning architectures, including Baresnet, and incorporates uncertainty quantification to assess the level of uncertainty in the classification results. The study presents a novel automatic tumor classification system for lung cancer based on CT images, which achieves a classification accuracy of 97.19% with an uncertainty quantification. The results demonstrate the potential of deep learning in lung cancer classification and highlight the importance of uncertainty quantification in improving the accuracy of classification results. This study’s novelty lies in the incorporation of uncertainty quantification in deep learning for lung cancer classification, which can lead to more reliable and accurate diagnoses in clinical settings.

[1]  R. Dhuli,et al.  A Novel Lightweight CNN Architecture for the Diagnosis of Brain Tumors Using MR Images , 2023, Diagnostics.

[2]  G. Giraldi,et al.  Prediction of permeability of porous media using optimized convolutional neural networks , 2022, Computational Geosciences.

[3]  Yanghua Wang,et al.  Regularization of anisotropic full waveform inversion with multiple parameters by adversarial neural networks , 2022, Geophysics.

[4]  V. P. Magboo,et al.  Detection of Brain Tumors from MRI Images using Convolutional Neural Networks , 2022, 2022 5th International Conference of Computer and Informatics Engineering (IC2IE).

[5]  D. Bhattacharyya,et al.  A bi-directional deep learning architecture for lung nodule semantic segmentation , 2022, The Visual computer.

[6]  Karma M. Fathalla,et al.  DETECT-LC: A 3D Deep Learning and Textural Radiomics Computational Model for Lung Cancer Staging and Tumor Phenotyping Based on Computed Tomography Volumes , 2022, Applied Sciences.

[7]  Hui Li,et al.  Stochastic Gradient Langevin Dynamics for Massive MIMO Detection , 2022, IEEE Communications Letters.

[8]  Yen-Yu Lin,et al.  Computer-assisted three-dimensional quantitation of programmed death-ligand 1 in non-small cell lung cancer using tissue clearing technology , 2022, Journal of Translational Medicine.

[9]  S. Jayakumar,et al.  Cloud-Based Lung Tumor Detection and Stage Classification Using Deep Learning Techniques , 2022, BioMed research international.

[10]  Dillip Ranjan Nayak,et al.  Brain Tumour Classification Using Noble Deep Learning Approach with Parametric Optimization through Metaheuristics Approaches , 2022, Comput..

[11]  J. Pauly,et al.  NeRP: Implicit Neural Representation Learning With Prior Embedding for Sparsely Sampled Image Reconstruction , 2021, IEEE Transactions on Neural Networks and Learning Systems.

[12]  Thomas Pock,et al.  Bayesian Uncertainty Estimation of Learned Variational MRI Reconstruction , 2021, IEEE Transactions on Medical Imaging.

[13]  Multidisciplinary Computational Anatomy: Toward Integration of Artificial Intelligence with MCA-based Medicine , 2022 .

[14]  Yu Wang,et al.  Large-scale real-world radio signal recognition with deep learning , 2021, Chinese Journal of Aeronautics.

[15]  M. Reiher Molecule‐Specific Uncertainty Quantification in Quantum Chemical Studies , 2021, Israel Journal of Chemistry.

[16]  Yongkeun Park,et al.  Label-Free White Blood Cell Classification Using Refractive Index Tomography and Deep Learning , 2021, BME frontiers.

[17]  Pragya Chaturvedi,et al.  Prediction and Classification of Lung Cancer Using Machine Learning Techniques , 2021 .

[18]  Thomas M. Phelan,et al.  Applications of Markov Chain Approximation Methods to Optimal Control Problems in Economics , 2021, Working paper (Federal Reserve Bank of Cleveland).

[19]  Mehedi Masud,et al.  A Machine Learning Approach to Diagnosing Lung and Colon Cancer Using a Deep Learning-Based Classification Framework , 2021, Sensors.

[20]  Chen Zhang,et al.  Quantifying Model Uncertainty in Inverse Problems via Bayesian Deep Gradient Descent , 2020, 2020 25th International Conference on Pattern Recognition (ICPR).

[21]  Brian Nord,et al.  Deeply uncertain: comparing methods of uncertainty quantification in deep learning algorithms , 2020, Mach. Learn. Sci. Technol..

[22]  Dipanjan Moitra,et al.  Classification of non-small cell lung cancer using one-dimensional convolutional neural network , 2020, Expert Syst. Appl..

[23]  W. Gwenzi,et al.  Sources and Health Risks of Rare Earth Elements in Waters , 2020 .

[24]  S. Hernández,et al.  Uncertainty quantification for plant disease detection using Bayesian deep learning , 2020, Appl. Soft Comput..

[25]  David Wilson,et al.  2D CNN versus 3D CNN for false-positive reduction in lung cancer screening , 2020, Journal of medical imaging.

[26]  Jung-Hsien Chiang,et al.  Reproducible Machine Learning Methods for Lung Cancer Detection Using Computed Tomography Images: Algorithm Development and Validation , 2019, Journal of medical Internet research.

[27]  Jun Pan,et al.  A Deep Learning Network via Shunt-Wound Restricted Boltzmann Machines Using Raw Data for Fault Detection , 2020, IEEE Transactions on Instrumentation and Measurement.

[28]  Dustin Tran,et al.  Simple and Principled Uncertainty Estimation with Deterministic Deep Learning via Distance Awareness , 2020, NeurIPS.

[29]  M. Avanzo,et al.  Radiomics and deep learning in lung cancer , 2020, Strahlentherapie und Onkologie.

[30]  Moulay A. Akhloufi,et al.  Deep Learning for Lung Cancer Nodules Detection and Classification in CT Scans , 2020 .

[31]  Yahui Zhang,et al.  Deep Transfer Convolutional Neural Network and Extreme Learning Machine for Lung Nodule Diagnosis on CT images , 2020, Knowl. Based Syst..

[32]  Jorge Calvo-Zaragoza,et al.  Ensemble classification from deep predictions with test data augmentation , 2019, Soft Comput..

[33]  Yu Wang,et al.  Research progress of computer aided diagnosis system for pulmonary nodules in CT images. , 2019, Journal of X-ray science and technology.

[34]  Taghi M. Khoshgoftaar,et al.  A survey on Image Data Augmentation for Deep Learning , 2019, Journal of Big Data.

[35]  G. Corrado,et al.  End-to-end lung cancer screening with three-dimensional deep learning on low-dose chest computed tomography , 2019, Nature Medicine.

[36]  T. Coroller,et al.  Deep Learning Predicts Lung Cancer Treatment Response from Serial Medical Imaging , 2019, Clinical Cancer Research.

[37]  D. Cremers,et al.  Efficient Deep Network Architectures for Fast Chest X-Ray Tuberculosis Screening and Visualization , 2019, Scientific Reports.

[38]  Hüseyin Polat,et al.  Classification of Pulmonary CT Images by Using Hybrid 3D-Deep Convolutional Neural Network Architecture , 2019, Applied Sciences.

[39]  Lei Tian,et al.  Illumination coding meets uncertainty learning: toward reliable AI-augmented phase imaging , 2019, 1901.02038.

[40]  Hwee Kuan Lee,et al.  Gated-Dilated Networks for Lung Nodule Classification in CT Scans , 2019, IEEE Access.

[41]  George R. Thoma,et al.  A novel stacked generalization of models for improved TB detection in chest radiographs , 2018, 2018 40th Annual International Conference of the IEEE Engineering in Medicine and Biology Society (EMBC).

[42]  Sergey Ioffe,et al.  Batch Normalization: Accelerating Deep Network Training by Reducing Internal Covariate Shift , 2015, ICML.

[43]  Bradley M. Hemminger,et al.  Contrast Limited Adaptive Histogram Equalization image processing to improve the detection of simulated spiculations in dense mammograms , 1998, Journal of Digital Imaging.

[44]  Omar Mohd Rijal,et al.  Wavelet as features for tuberculosis (MTB) using standard X-ray film images , 2002, 6th International Conference on Signal Processing, 2002..

[45]  K. H. Hohne,et al.  Pictorial Information Systems in Medicine , 1986, NATO ASI Series.

[46]  S. Pizer Psychovisual issues in the display of medical images , 1986 .