X-ray Classification of Tuberculosis Based on Convolutional Networks
Kai Cao, Jingyi Zhang, Mengge Huang, Tao Deng
Abstract
Tuberculosis is a chronic infectious disease caused by Mycobacterium tuberculosis, which can invade many organs, and pulmonary tuberculosis is the most common infection. It is the key to treat tuberculosis to detect and diagnose the disease in the early stage. The existing computer-aided detection system has made preliminary progress in the diagnosis of pulmonary tuberculosis based on chest X-ray, but there is still a lack of further research on the classification of image signs of tuberculosis. In recent years, with the in-depth research and development in the field of deep learning, convolutional networks have emerged. Convolutional networks have achieved the best current results in image recognition, image classification, image segmentation, and other fields. Therefore, this paper applies the convolutional network to tuberculosis CT images and uses different convolutional network models to study the classification of tuberculosis CT images. Experiments show that the DenseNet121 model has higher performance than VGGNet16, VGGNet19, and ResNet152 models. As a result of classification, the accuracy rate is over 90%.
MeSH terms
- Tuberculosis
- Artificial intelligence
- Contextual image classification
- Computer science
- Mycobacterium tuberculosis
- Convolutional neural network
- Deep learning
- Pattern recognition (psychology)
- Segmentation
- Image segmentation
- Pulmonary tuberculosis
- Disease
- Image (mathematics)