Back

Early prediction of Mycobacterium tuberculosis transmission clusters using supervised learning models

Gharamaleki, O. G.; Colijn, C.; Sekirov, I.; Johnston, J. C.; Sobkowiak, B.

2024-04-16 infectious diseases
10.1101/2024.04.16.24305900 medRxiv
Show abstract

Identifying individuals with tuberculosis with a high risk of onward transmission can guide disease prevention and public health strategies. Here, we train classification models to predict the first sampled isolates in Mycobacterium tuberculosis transmission clusters from demographic and disease data. We find that supervised learning models, in particular balanced random forests, can be used to develop predictive models that discriminate between individuals with TB that are more likely to form transmission clusters and individuals that are likely not to transmit further, with good model performance and AUCs of [≥] 0.75. We also identified the most important patient and disease characteristics in the best performing classification model, including patient demographics, site of infection, TB lineage, and age at diagnosis. This framework can be used to develop predictive tools for the early assessment of a patients transmission risk to prioritise individuals for enhanced follow-up with the aim of reducing further transmission.

Matching journals

The top 9 journals account for 50% of the predicted probability mass.

50% of probability mass above

"Similar papers" are the closest papers from that journal in the model's embedding space. They show what the match is built on, but the ranking comes mostly from a classifier over the whole training set, not from these examples alone.