An X-ray image-based pruned dense convolution neural network for tuberculosis detection

Abstract

According to the Ministry of Health in Kenya, tuberculosis (TB) is the fifth greatest cause of death and the main infectious disease killer in Kenya and across the world. In Kenya and throughout Africa, TB continues to wreak havoc on many vulnerable populations, homes, and communities despite being preventable and treatable. Common TB diagnostics, like blood and skin tests, frequently fail to identify the precise kind of TB. As a result, the World Health Organization (WHO) advises expanding the use of X-rays, for screening. In TB-prevalent regions of Kenya, a shortage of radiologists hampers effective screening and diagnosis, highlighting the need for scalable solutions for accurate X-ray analysis.

Recent advancements in deep learning techniques have shown promise in the healthcare sector, particularly in radiology. However, many deep convolutional neural network (CNN) architectures are computationally intensive due to their size and resource requirements. This study designed and developed a Pruned CNN to address this issue by applying pruning techniques to baseline architectures. This approach significantly reduced model sizes while maintaining accuracy levels. Specifically, the pruned version of the DenseNet model achieved an impressive 99 % accuracy with a reduction rate of 65.8 %. These results highlight the potential of this pruned CNN as an effective and efficient tool for TB detection, particularly in resource-constrained environments. This study addresses the shortage of radiological expertise in many regions by providing a tool that can assist in the interpretation of X-ray images. This capability can help healthcare providers deliver timely and accurate diagnoses, thereby improving patient care.