Development and validation of an interpretable machine learning model for risk stratification among individuals with latent tuberculosis infection: a multicenter case-control study

Abstract Background Latent tuberculosis infection (LTBI) affects a quarter of the global population. Conventional LTBI risk screening tools have limited sensitivity, and existing machine learning prediction models mostly rely on costly laboratory biomarkers, restricting their application in grassroots healthcare and public health. This study aimed to develop an interpretable machine learning model to distinguish pulmonary tuberculosis (PTB) from LTBI and stratify individuals with LTBI to prioritize further evaluation and preventive interventions. Methods This multicenter case-control study enrolled 782 participants from 21 designated tuberculosis institutions in Henan Province, China, including 401 individuals with LTBI and 381 patients with newly diagnosed, treatment-naive PTB. Epidemiological predictors were collected via standardized face-to-face questionnaires. The dataset was randomly split into a training set and a test set at a 7:3 ratio. Eleven predictors were screened via ten-fold cross-validated LASSO regression. Seven machine learning algorithms were constructed and evaluated using multiple metrics including AUC, F1 score, Brier score, calibration curves, and decision curve analysis. A three-tier risk framework was defined based on a percentile strategy. SHapley Additive exPlanations (SHAP) was applied to interpret the optimal model, which was further deployed as a free web calculator. Results The eXtreme Gradient Boosting (XGB) model achieved the highest test-set performance, with an AUC of 0.818 (95% CI 0.763–0.873), an F1 score of 0.764 (95% CI 0.698–0.822), and a Brier score of 0.176 (95% CI 0.148–0.204), with favorable calibration (Hosmer-Lemeshow P > 0.05). DCA confirmed a stable positive net benefit. Using cutoffs of > 0.763 (high-risk) and < 0.328 (low-risk), the model stratified participants into three tiers, with the proportion of PTB cases increasing from 20.00% in the low-risk tier to 84.29% in the high-risk tier. SHAP analysis identified male sex, low body mass index, advanced age, unintentional weight loss within six months, and low annual household income as the most influential features. Conclusions The interpretable XGB model based on low-cost epidemiological indicators can distinguish prevalent PTB from LTBI and may help prioritize individuals with LTBI for further evaluation. The web calculator is a research prototype; its clinical utility requires external validation and prospective evaluation before it can guide preventive-treatment decisions.

Authors

Publication Details

Journal
BMC Public Health
Published
2026-10-22
DOI
https://doi.org/10.1186/s12889-026-30066-9
Primary Topic
Tuberculosis Research and Epidemiology
Type
article
Field-Weighted Citation Impact
0.00
Controls
|||
ALL TIME
JAN
FEB
MAR
APR
MAY
JUN
JUL
AUG
SEP
OCT
article

Development and validation of an interpretable machine learning model for risk stratification among individuals with latent tuberculosis infection: a multicenter case-control study

Chunyu Xie, Wenjing Chang, Linqi Diao, Dingyong Sun et al.
BMC Public Health
Tuberculosis Research and Epidemiology
article

Development and validation of an interpretable machine learning model for risk stratification among individuals with latent tuberculosis infection: a multicenter case-control study

Chunyu Xie, Wenjing Chang, Linqi Diao, Dingyong Sun, Xiaowan Xu, Jianhui Zhu, Yafei Xu, Mengya He, Danwei Zheng, Yunfeng Qi, Yanqiu Zhang, Weidong Wang
article en

Abstract

Abstract Background Latent tuberculosis infection (LTBI) affects a quarter of the global population. Conventional LTBI risk screening tools have limited sensitivity, and existing machine learning prediction models mostly rely on costly laboratory biomarkers, restricting their application in grassroots healthcare and public health. This study aimed to develop an interpretable machine learning model to distinguish pulmonary tuberculosis (PTB) from LTBI and stratify individuals with LTBI to prioritize further evaluation and preventive interventions. Methods This multicenter case-control study enrolled 782 participants from 21 designated tuberculosis institutions in Henan Province, China, including 401 individuals with LTBI and 381 patients with newly diagnosed, treatment-naive PTB. Epidemiological predictors were collected via standardized face-to-face questionnaires. The dataset was randomly split into a training set and a test set at a 7:3 ratio. Eleven predictors were screened via ten-fold cross-validated LASSO regression. Seven machine learning algorithms were constructed and evaluated using multiple metrics including AUC, F1 score, Brier score, calibration curves, and decision curve analysis. A three-tier risk framework was defined based on a percentile strategy. SHapley Additive exPlanations (SHAP) was applied to interpret the optimal model, which was further deployed as a free web calculator. Results The eXtreme Gradient Boosting (XGB) model achieved the highest test-set performance, with an AUC of 0.818 (95% CI 0.763–0.873), an F1 score of 0.764 (95% CI 0.698–0.822), and a Brier score of 0.176 (95% CI 0.148–0.204), with favorable calibration (Hosmer-Lemeshow P > 0.05). DCA confirmed a stable positive net benefit. Using cutoffs of > 0.763 (high-risk) and < 0.328 (low-risk), the model stratified participants into three tiers, with the proportion of PTB cases increasing from 20.00% in the low-risk tier to 84.29% in the high-risk tier. SHAP analysis identified male sex, low body mass index, advanced age, unintentional weight loss within six months, and low annual household income as the most influential features. Conclusions The interpretable XGB model based on low-cost epidemiological indicators can distinguish prevalent PTB from LTBI and may help prioritize individuals with LTBI for further evaluation. The web calculator is a research prototype; its clinical utility requires external validation and prospective evaluation before it can guide preventive-treatment decisions.

BMC Public Health
Openalex Percentile: Top 12%
Tuberculosis Research and Epidemiology
AI Navigator

Ask Laika to Summarize, Analyze, and Connect papers live on the map.

Summarize Papers & Methodologies

Extract key findings, datasets, and comparative methods across publications.

Benchmark Rankings & Visual Analytics

Rank top research institutions, authors, funders, topics, and journals by Field-Weighted Citation Impact (FWCI) and paper volume with instant charts.

Connect Distant Disciplines

Bridge topological clusters on the map to find hidden collaborative intersections.

Laika access coming soon