Detect feature edges for diagnosis of bacterial vaginosis

PeerJ. 2023 Jan 17:11:e14667. doi: 10.7717/peerj.14667. eCollection 2023.

Abstract

One of the most common diseases among women of reproductive age is bacterial vaginosis (BV). However, the etiology of BV remains unknown. In this study, we modeled the temporal sample of the vaginal microbiome as a network and investigated the relationship between the network edges and BV. Furthermore, we used feature selection algorithms including decision tree (DT) and ReliefF (RF) to select the network feature edges associated with BV and subsequently validated these feature edges through logistic regression (LR) and support vector machine (SVM). The results show that: machine learning can distinguish vaginal community states (BV, ABV, SBV, and HEA) based on a few feature edges; selecting the top five feature edges of importance can achieve the best accuracy for the feature selection and classification model; the feature edges selected by DT outperform those selected by RF in terms of classification algorithm LR and SVM, and LR with DT feature edges is more suitable for diagnosing BV; two feature selection algorithms exhibit differences in the importance of ranking of edges; the feature edges selected by DT and RF cannot construct sub-network associated with BV. In short, the feature edges selected by our method can serve as indicators for personalized diagnosis of BV and aid in the clarification of a more mechanistic interpretation of its etiology.

Keywords: Bacterial vaginosis; Classification; Feature edges; Machine learning; Network.

Publication types

  • Research Support, Non-U.S. Gov't

MeSH terms

  • Algorithms
  • Female
  • Humans
  • Machine Learning
  • Support Vector Machine
  • Vagina
  • Vaginosis, Bacterial* / diagnosis

Grants and funding

This study received funding from the scientific research fund of Yunnan Provincial Department of Education (No: 2021J0960). The funders had no role in study design, data collection and analysis, decision to publish, or preparation of the manuscript.