Abstract:Tuberculosis (TB) remains a global health problem, and is the leading cause of death from an infectious disease. A crucial step in the treatment of tuberculosis is screening high risk populations and the early detection of the disease, with chest x-ray (CXR) imaging being the most widely-used imaging modality. As such, there has been significant recent interest in artificial intelligence-based TB screening solutions for use in resource-limited scenarios where there is a lack of trained healthcare workers with expertise in CXR interpretation. Motivated by this pressing need and the recent recommendation by the World Health Organization (WHO) for the use of computer-aided diagnosis of TB in place of a human reader, we introduce TB-Net, a self-attention deep convolutional neural network tailored for TB case screening. We used CXR data from a multi-national patient cohort to train and test our models. A machine-driven design exploration approach leveraging generative synthesis was used to build a highly customized deep neural network architecture with attention condensers. We conducted an explainability-driven performance validation process to validate TB-Net's decision-making behavior. Experiments on CXR data from a multi-national patient cohort showed that the proposed TB-Net is able to achieve accuracy/sensitivity/specificity of 99.86/100.0/99.71%. Radiologist validation was conducted on select cases by two board-certified radiologists with over 10 and 19 years of experience, respectively, and showed consistency between radiologist interpretation and critical factors leveraged by TB-Net for TB case detection for the case where radiologists identified anomalies. The proposed TB-Net not only achieves high tuberculosis case detection performance in terms of sensitivity and specificity, but also leverages clinically relevant critical factors in its decision making process. While not a production-ready solution, we hope that the open-source release of TB-Net as part of the COVID-Net initiative will support researchers, clinicians, and citizen data scientists in advancing this field in the fight against this global public health crisis.

Deep Feature Learning from a Hospital-Scale Chest X-ray Dataset with Application to TB Detection on a Small-Scale Dataset

A Lightweight, Rapid and Efficient Deep Convolutional Network for Chest X-Ray Tuberculosis Detection

Reliable Tuberculosis Detection using Chest X-ray with Deep Learning, Segmentation and Visualization

Exploiting Cascaded Ensemble of Features for the Detection of Tuberculosis Using Chest Radiographs

Deep Learning for Automated Screening of Tuberculosis from Indian Chest X-rays: Analysis and Update

A novel dense-net deep neural network with enhanced feature selection method for classification of different stages of tuberculosis using chest X-ray images

Deep learning-based automatic detection of tuberculosis disease in chest X-ray images

Self-Trained Convolutional Neural Network (CNN) for Tuberculosis Diagnosis in Medical Imaging

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

ChestX-ray8: Hospital-scale Chest X-ray Database and Benchmarks on Weakly-Supervised Classification and Localization of Common Thorax Diseases

TX-CNN: Detecting Tuberculosis in Chest X-ray Images Using Convolutional Neural Network.

AI-Assisted Tuberculosis Detection and Classification from Chest X-Rays Using a Deep Learning Normalization-Free Network Model

Comparison of Deep Learning Approaches for Multi-Label Chest X-Ray Classification

Deep Learning Methods for Screening Pulmonary Tuberculosis Using Chest X-rays

TB-Net: A Tailored, Self-Attention Deep Convolutional Neural Network Design for Detection of Tuberculosis Cases From Chest X-Ray Images

Abnormality Detection and Localization in Chest X-Rays using Deep Convolutional Neural Networks

Anatomy-XNet: An Anatomy Aware Convolutional Neural Network for Thoracic Disease Classification in Chest X-rays

Multi-Label Classification of Chest X-ray Abnormalities Using Transfer Learning Techniques

Automated Identification of Thoracic Pathology from Chest Radiographs with Enhanced Training Pipeline

Learning to Read Chest X-Ray Images from 16000+ Examples Using CNN

TB-CXRNet: Tuberculosis and Drug-Resistant Tuberculosis Detection Technique Using Chest X-ray Images