Detect feature edges for diagnosis of bacterial vaginosis

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.