Active and Inactive Tuberculosis Classification Using Convolutional Neural Networks with MLP-Mixer

Early detection of tuberculosis plays a critical role in effective treatment management. Like active tuberculosis, early identification of inactive forms such as latent or healed tuberculosis is essential to prevent future reactivation. In this study, we developed a deep-learning-based binary classification model to distinguish between active and inactive tuberculosis cases. Our model architecture incorporated an EfficientNet backbone with an MLP-Mixer classification head and was fine-tuned on a dataset annotated by Cheonan Soonchunhyang Hospital. To enhance predictive performance, we applied transfer learning using weights pre-trained on the JFT-300M dataset via the Noisy Student training method. Unlike conventional models, our approach achieved competitive results, with an accuracy of 96.3%, a sensitivity of 95.9%, and a specificity of 96.6% on the test set. These promising outcomes suggest that our model could serve as a valuable asset to support clinical decision-making and streamline early screening workflows for latent tuberculosis.

Перевод пока недоступен