Machine Learning Maps the Microbial Tipping Points of Bacterial Vaginosis

Bacterial vaginosis is one of the most common vaginal syndromes affecting reproductive-age women worldwide, yet its diagnosis and progression remain surprisingly difficult to pin down. Clinicians have long known that the condition arises when the normally Lactobacillus-dominated vaginal microbiome shifts toward a community dominated by anaerobic bacteria. What has remained unclear is how gradual changes in that balance translate into a clinical diagnosis, and which bacterial species matter most along the way. A new study published in PLOS Computational Biology by Cameron G. Celeste, Carleigh C. Sokolik, Wambui Gachunga, and Ivana K. Parker tackles this question head-on, using machine learning to model the microbial transitions that separate a healthy vaginal ecosystem from a BV-positive one.

The stakes of getting this right are considerable. Bacterial vaginosis is associated with a range of adverse obstetric and gynecological outcomes, including an increased risk of acquiring sexually transmitted infections, heightened susceptibility to HIV, elevated rates of cervical cancer, and pre-term birth. Because the syndrome is defined by a shift in relative abundance between Lactobacilli and anaerobes rather than by the presence of a single pathogen, diagnosis has traditionally relied on broad criteria that may not capture the full complexity of the microbial community. Understanding precisely which species drive the transition, and at what abundances, could sharpen both diagnostics and research into how the condition develops.

To address this, the research team assembled 16S rRNA sequencing data from patients presenting with bacterial vaginosis and subjected it to a rigorous, systematic comparison of machine learning architectures and feature selection methods. Rather than committing to a single algorithm, the authors evaluated multiple classifiers to determine which combination of model and feature selection strategy could best predict BV status from the microbial profiles. This kind of head-to-head benchmarking is essential in microbiome studies, where results can vary dramatically depending on the analytical pipeline chosen, and where overfitting to small or heterogeneous datasets is a persistent concern.

The comparison produced a clear verdict. Support vector machine and random forest models, when paired with appropriate feature selection, predicted BV diagnosis with the most balanced accuracy of all the approaches tested. Balanced accuracy is a particularly informative metric in this context because it accounts for performance on both BV-positive and BV-negative cases, avoiding the trap of models that appear accurate simply because they favor the majority class. The finding aligns with a broader pattern in microbiome research, in which tree-based ensembles and margin-based classifiers often outperform more flexible deep learning architectures when the number of samples is modest relative to the number of microbial features.

With the best-performing models in hand, the researchers turned to explainable artificial intelligence methods to interrogate why the classifiers made the predictions they did. Explainable AI techniques allow researchers to move beyond a single accuracy score and identify which input features — in this case, which bacterial taxa — carry the most weight in the model’s decisions. Applied to the vaginal microbiome data, this analysis pinpointed ten species as the most important contributors to BV diagnosis: four species of Lactobacilli and six anaerobic bacteria. This shortlist represents a data-driven distillation of the microbial community down to the organisms that most reliably signal whether a patient’s ecosystem has tipped toward disease.

The identification of these ten key species has practical implications for both diagnostics and pathogenesis research. Several of the anaerobes implicated in BV have received varying degrees of scientific attention over the years, and the authors note that the determination of key bacteria can inform research into species that have previously eluded scientific focus. In other words, the machine learning pipeline did not merely confirm what was already known; it surfaced specific organisms whose role in the syndrome may have been underappreciated, providing a prioritized list of targets for future experimental and clinical investigation.

Perhaps the most visually and conceptually striking contribution of the study, however, lies in its use of Voronoi-based decision boundaries. A Voronoi diagram partitions a space into regions based on proximity to a set of reference points, and in this application the technique was adapted to visualize how a classifier divides the space of possible microbial abundance combinations into BV-positive and BV-negative territories. By plotting the relative abundances of pairs of key bacteria, the researchers produced maps in which every point corresponds to a possible microbial state, and the boundaries between regions mark the tipping points at which the model’s diagnosis flips from healthy to BV-positive.

These decision boundary plots offer something that standard classification metrics cannot: an intuitive, interpretable point of reference for how the relative abundances of key vaginal flora translate into diagnostic outcomes. A clinician or researcher looking at such a plot can immediately see, for example, how a decline in protective Lactobacilli combined with a rise in a particular anaerobe moves a sample across the boundary into BV territory. Because the boundaries are built from the highest-performing models rather than arbitrary thresholds, they reflect patterns learned directly from patient data, making them a candidate tool for guiding diagnostic interpretation and for generating hypotheses about the ecological dynamics of the syndrome.

The study also speaks to a larger methodological conversation in microbiome science. Population-specific microbial interactions — the ways in which bacterial species relate to one another within a given community — are notoriously difficult to quantify, and findings from one population do not always generalize to another. By combining careful model benchmarking, feature selection, and explainability techniques, the authors demonstrate a framework that could be applied beyond BV to other conditions defined by community-level shifts rather than single pathogens. The approach treats the microbiome as a system of interacting players whose collective balance, not any one organism alone, determines health or disease.

For patients and clinicians, the immediate promise of this work is a more precise, species-level understanding of what bacterial vaginosis actually is at the microbial level. For researchers, the ten key species and the Voronoi boundary maps provide concrete, testable targets: experiments can now be designed to probe how these organisms interact, how their abundance thresholds vary across populations, and whether manipulating them can prevent or reverse the transition to BV. As machine learning continues to move from the margins of microbiology into its methodological mainstream, studies like this one illustrate how predictive models, when paired with interpretability tools, can do more than classify — they can illuminate the hidden structure of the ecosystems within us.

Subject of Research: Machine learning analysis of 16S rRNA data to identify key bacterial interactions and diagnostic decision boundaries in bacterial vaginosis

Article Title: Machine learning and Voronoi-based decision boundaries for Bacterial vaginosis to determine population- specific microbial interactions

Article References: Celeste, C. G., Sokolik, C. C., Gachunga, W., & Parker, I. K. (2026). Machine learning and Voronoi-based decision boundaries for Bacterial vaginosis to determine population- specific microbial interactions. PLOS Computational Biology, 22(10), e1014767. https://doi.org/10.1371/journal.pcbi.1014767

Image Credits: AI Generated

DOI: 10.1371/journal.pcbi.1014767

Keywords: bacterial vaginosis, machine learning, microbiome, 16S rRNA, Lactobacillus, anaerobic bacteria, support vector machine, random forest, Voronoi diagram, explainable AI, feature selection, vaginal health