Deep learning based CT images automatic analysis model for active/non-active pulmonary tuberculosis differential diagnosis

Abstract

Active pulmonary tuberculosis (ATB), which is more infectious and has a higher mortality rate compared with non-active pulmonary tuberculosis (non-ATB), needs to be diagnosed accurately and timely to prevent the tuberculosis from spreading and causing deaths. However, traditional differential diagnosis methods of active pulmonary tuberculosis involve bacteriological testing, sputum culturing and radiological images reading, which is time consuming and labour intensive. Therefore, an artificial intelligence model for ATB differential diagnosis would offer great assistance in clinical practice. In this study, computer tomography (CT) scans images and corresponding clinical information of 1160 ATB patients and 1131 patients with non-ATB were collected and divided into training, validation, and testing sets. A 3-dimension (3D) Nested UNet model was utilized to delineate lung field regions in the CT images, and three different pre-trained deep learning models including 3D VGG-16, 3D EfficientNet and 3D ResNet-50 were used for classification and differential diagnosis task. We also collected an external testing set with 100 ATB cases and 100 Non-ATB cases for further validation of the model. In the internal and external testing set, the 3D ResNet-50 model outperformed other models, reaching an AUC of 0.961 and 0.946, respectively. The 3D ResNet-50 model reached even higher levels of diagnostic accuracy than experienced radiologists, while the CT images reading and diagnosing speed was 10 times faster than human experts. The model was also capable of visualizing clinician interpretable lung lesion regions important for differential diagnosis, making it a powerful tool assisting ATB diagnosis. In conclusion, we developed an auxiliary tool to differentiate active and non-active pulmonary tuberculosis, which would have broad prospects in the bedside.