Explainable deep-neural-network supported scheme for tuberculosis detection from chest radiographs.

Journal: BMC medical imaging
Published Date:

Abstract

Chest radiographs are examined in typical clinical settings by competent physicians for tuberculosis diagnosis. However, this procedure is time consuming and subjective. Due to the growing usage of machine learning techniques in applied sciences, researchers have begun applying comparable concepts to medical diagnostics, such as tuberculosis screening. In the period of extremely deep neural nets which comprised of hundreds of convolution layers for feature extraction, we create a shallow-CNN for screening of TB condition from Chest X-rays so that the model is able to offer appropriate interpretation for right diagnosis. The suggested model consists of four convolution-maxpooling layers with various hyperparameters that were optimized for optimal performance using a Bayesian optimization technique. The model was reported with a peak classification accuracy, F1-score, sensitivity and specificity of 0.95. In addition, the receiver operating characteristic (ROC) curve for the proposed shallow-CNN showed a peak area under the curve value of 0.976. Moreover, we have employed class activation maps (CAM) and Local Interpretable Model-agnostic Explanations (LIME), explainer systems for assessing the transparency and explainability of the model in comparison to a state-of-the-art pre-trained neural net such as the DenseNet.

Authors

  • B Uma Maheswari
    Computer Science and Engineering, St. Joseph's College of Engineering, OMR, Chennai, India.
  • Dahlia Sam
    Department of Information Technology, Dr. M.G.R Educational and Research Institute, Periyar E.V.R High Road, Vishwas Nagar, Maduravoyal, Chennai, Tamilnadu, 600095, India.
  • Nitin Mittal
    University Centre for Research and Development, Chandigarh University, Mohali, Punjab, 140413, India.
  • Abhishek Sharma
    Department of Biotechnology and Bioengineering, Institute of Advanced Research, Koba Institutional Area, Gandhinagar, India.
  • Sandeep Kaur
    Department of Computer Engineering & Technology, Guru Nanak Dev University, Amritsar, Punjab, 143005, India.
  • S S Askar
    Department of Statistics and Operations Research, College of Science, King Saud University, P.O. Box 2455, 11451, Riyadh, Saudi Arabia.
  • Mohamed Abouhawwash
    Department of Computational Mathematics, Science and Engineering (CMSE), College of Engineering, Michigan State University, East Lansing, MI, 48824, USA. abouhaww@msu.edu.