Skip to main content
Advertisement
Browse Subject Areas
?

Click through the PLOS taxonomy to find articles in your field.

For more information about PLOS Subject Areas, click here.

  • Loading metrics

High-dimensional linear state space models for dynamic microbial interaction networks

  • Iris Chen ,

    Contributed equally to this work with: Iris Chen, Yogeshwar D. Kelkar

    Roles Conceptualization, Formal analysis, Investigation, Methodology, Writing – original draft

    Current address: Global Biostatistical Science, Amgen, Thousand Oaks, CA 91320, United States of America

    Affiliation Department of Biostatistics and Computational Biology, University of Rochester, Rochester, NY 14642, United States of America

  • Yogeshwar D. Kelkar ,

    Contributed equally to this work with: Iris Chen, Yogeshwar D. Kelkar

    Roles Investigation, Validation, Visualization, Writing – original draft, Writing – review & editing

    Affiliation Department of Biology, University of Rochester, Rochester, NY 14642, United States of America

  • Yu Gu,

    Roles Data curation, Formal analysis, Investigation, Methodology, Software, Writing – original draft

    Affiliation Department of Biostatistics and Computational Biology, University of Rochester, Rochester, NY 14642, United States of America

  • Jie Zhou,

    Roles Conceptualization, Investigation, Writing – original draft

    Affiliation Department of Statistics, Xidian University, Xian, Shanxi 71007, China

  • Xing Qiu ,

    Roles Conceptualization, Methodology, Project administration, Resources, Software, Supervision, Validation, Writing – original draft, Writing – review & editing

    xing_qiu@urmc.rochester.edu (XQ); Hulin.Wu@uth.tmc.edu (HW)

    Affiliation Department of Biostatistics and Computational Biology, University of Rochester, Rochester, NY 14642, United States of America

  • Hulin Wu

    Roles Conceptualization, Funding acquisition, Investigation, Methodology, Project administration, Resources, Supervision, Writing – original draft, Writing – review & editing

    xing_qiu@urmc.rochester.edu (XQ); Hulin.Wu@uth.tmc.edu (HW)

    Current address: Department of Biostatistics, School of Public Health, University of Texas Health Science Center at Houston, Houston, TX 77030, United States of America

    Affiliation Department of Biostatistics and Computational Biology, University of Rochester, Rochester, NY 14642, United States of America

Abstract

Medical researchers are increasingly interested in knowing how the complex community of micro-organisms living on human body impacts human health. Key to this is to understand how the microbes interact with each other. Time-course studies on human microbiome indicate that the composition of microbiome changes over short time periods, primarily as a consequence of synergistic and antagonistic interactions of the members of the microbiome with each other and with the environment. Knowledge of the abundance of bacteria—which are the predominant members of the human microbiome—in such time-course studies along with appropriate mathematical models will allow us to identify key dynamic interaction networks within the microbiome. However, the high-dimensional nature of these data poses significant challenges to the development of such mathematical models. We propose a high-dimensional linear State Space Model (SSM) with a new Expectation-Regularization-Maximization (ERM) algorithm to construct a dynamic Microbial Interaction Network (MIN). System noise and measurement noise can be separately specified through SSMs. In order to deal with the problem of high-dimensional parameter space in the SSMs, the proposed new ERM algorithm employs the idea of the adaptive LASSO-based variable selection method so that the sparsity property of MINs can be preserved. We performed simulation studies to evaluate the proposed ERM algorithm for variable selection. The proposed method is applied to identify the dynamic MIN from a time-course vaginal microbiome study of women. This method is amenable to future developments, which may include interactions between microbes and the environment.

Introduction

Human epithelial surfaces such as those of intestines, mouth and vagina provide rich environment for the growth of a variety of bacteria that together constitute most of the human microbiome. It is now widely understood that microbiome has direct relationship to human health. Consequently, it is important to understand how bacteria that constitute the microbiome interact with their hosts and with each other. These bacteria interact with each other in various forms of cooperative and antagonistic relationships, and this complex set of interactions can be depicted in the form of a Microbial Interaction Network (MIN) [1].

The degree of cooperative and antagonistic relationships between two types of bacteria can be gauged from the impact that one type has over the growth and abundance of the other. Interactions among bacteria have traditionally been inferred using microbiological assays involving co-culturing; however not all bacteria can be cultured, and laboratory inferred interactions may not occur in nature. In contrast, sequencing the variable regions of 16S ribosomal RNAs directly from biological samples gives estimates of abundance of a large variety of bacteria, which can provide a holistic and unbiased view of microbial interactions.

Few longitudinal studies of human microbiome have been undertaken, and the initial discoveries include discovery of the most abundant microbial taxa on various locations on human body [25]; large inter-personal and within-subject temporal variation in microbiome composition [3, 6, 7]; and the effect of external stimuli on microbiome [8, 9]. However, the construction of MIN operational in human microbiota remains a major challenge due to the high-dimensional and high-fluctuation nature of the data [1].

MIN can be constructed either by cross-sectional data [10, 11] or time-series data [1215]. Compared with MINs constructed from cross-sectional data, MINs constructed from time-course data can capture the dynamic relationship between different bacteria and/or external stimuli, which arguably provide more realistic representations of the interactions of microbiota as they operate in nature [1, 8]. In this paper, we will focus on reconstructing dynamic MINs based on time series data which also is technically more challenging compared to that based on the cross-sectional data.

Many models have been proposed for constructing dynamic MIN [1012, 1518], including those based on ordinary differential equation (ODE) models [13, 14]. An ODE model is formed by taking the derivative of bacterial abundance as a function of abundance of all other bacteria and/or external stimuli. This results in a directed network model, and the dynamic nature of MIN is automatically captured and quantified. However, it is computationally difficult to apply ODE models to more than a dozen or so variables to simultaneously estimate system dynamics and regulatory relationships. Furthermore, most ODE models ignore both system and measurement errors, which in many cases have critical impact on results. In this paper, we will explore the utility of state space model (SSM), which is an alternative to ODE models, to capture MIN dynamics from time-course data.

A state space model (SSM) is a special case of dynamic Bayesian networks (DBNs). For simplicity, we only consider linear SSM, also referred to as linear dynamic systems (LDS) [1921], for dynamic MINs in this study. SSMs have been extensively applied in the field of engineering, and recently, in systems biology, for noisy measurements over time, and to discover underlying true dynamics of the system [22]. In our study, we let represent a p-dimensional vector of microbial abundance of p bacterial operational taxonomic units (OTUs) observed at time t. Here, OTU is an operational definition of bacterial species, obtained using clustering of 16S ribosomal RNA sequences extracted from biological samples. In linear SSMs, yt is assumed to be generated from a k-dimensional real-valued hidden state variable vector , and the sequence of evolving xt follows a first-order Markov process, which can be written as [20] (1)

Here A is the k × k-dimensional state dynamic (autoregressive) matrix; C is the p × k-dimensional observation matrix; and wtN(0, Q), vtN(0, R) are independent system and measurement noises, respectively. Both Q and R are assumed to be diagonal in many practical applications. The initial state vector is usually assumed to have distribution N(μ, Σ).

In order to capture the dynamic MIN using state space model, it is necessary to investigate the problem of parameter estimation and variable selection for high dimensional SSMs [23, 24]. For example, Rangel and colleagues [23] applied SSMs in which observations were divided into a set of input (or exogenous) variables and a set of output (or response) variables, and the dimension was determined by cross-validations; Kojima et al [24] proposed a vector autoregression (VAR) model for the dynamic gene network. Based on the state space representation of VAR, they investigated the problem of parameter estimation and variable selection by L1 regularization and EM algorithm. Although these publications have suggested several useful ideas about the statistical inference of high dimensional state space model, efficient algorithms for establishing MIN have not been well addressed from a computational perspective.

In this paper, we develop a practical dynamic MIN reconstruction pipeline based on SSM that not only incorporates many existing SSM parameter estimation and model selection techniques, but also is computationally efficient and applicable for “large p, small n” data such as 16S microbiome abundance data. First, we propose a novel Expectation-Regularization-Maximization (ERM) computational framework for the SSMs, and provide a feasible implementation strategy for initialization of the ERM algorithm, i.e., to initialize the ERM algorithm from the R step using nonparametrically estimated state variables instead of initializing the algorithm from the E step, which is not feasible in the high-dimensional SSM case. Second, we propose the vectorization of the matrices in the SSM and use a concept of “pseudo-regression” to justify the R step for L1-regularization based on which the standard LARS algorithm with minor modifications can be carried out. Third, a new row-based algorithm is proposed in order to reduce the memory footprint, which is a major computational cost for high-dimensional data analyses. In simulation studies we demonstrate that the proposed row-based algorithm performs equally well and could handle a higher dimensional model compared to a matrix-based algorithm. Lastly, we apply the proposed method to reconstruct dynamic MIN for normal bacterial communities in human vagina. The MINs that were constructed revealed some previously known as well as some novel microbial relationships.

Methods

Model selection and parameter estimation of high-dimensional linear state space models

We propose a linear SSM with an Expectation-Regularization-Maximization (ERM) algorithm to efficiently construct dynamic MINs. Since time course abundance data for all bacteria in the network can be obtained using next generation sequence technology, we can set the observation matrix C = Ip×p (identity matrix) in Model (1). That is, the linear SSM for dynamic MIN can be written as (2)

Thus, in this model, the dimension of state vector equals the dimension of observation vector (k = p). Other assumptions remain the same as in Model (1). For simplicity, we also assume that both Q and R are diagonal, i.e., and .

The above model allows us to use the time-course microbiome data to construct a direct dynamic MIN with distinguishable system noise and measurement noise. Each element aij (denoting the ith-row and jth-column element) in p × p system matrix A represents a directed edge in the network which is time-invariant, and reflects the interacting effect from bacterial species j to bacterial species i. However, when p is very large, it is infeasible to directly estimate A since we may encounter the problem of estimating a high-dimensional matrix with sparse data that requires inverting high-dimensional matrices, which not only is computationally intensive, but also can be numerically unstable. In addition, microbial interaction networks are usually sparse, i.e., each bacterial species may only be impacted by a limited number of other species. In other words, high-dimensional p × p matrix A is a sparse matrix with many elements being zero. It is advantageous to perform variable selections to determine the zero elements of A while we can estimate the non-zero elements at the same time.

It is known that, from the Markov property of the state space model, the joint likelihood for complete data for the SSM can be written as (3) where θ = (A, Q, R, μ, Σ). For the linear SSM (2), the joint log-likelihood of complete data can be expressed as (4)

In the following subsections, we propose the Expectation-Regularization-Maximization (ERM) algorithm to simultaneously determine the zero-elements and estimate the non-zero elements of A based on the maximum likelihood principle.

Expectation-Regularization-Maximization (ERM) algorithm

When the SSM parameters are known, the Kalman filter and smoother can be used to estimate the hidden states [25]. Assuming that model parameters are given, the Kalman filter is the optimal method to estimate the state variables at time t of a linear Gaussian SSM from a sequence of noisy observations {y1, …, yt, …, yT}. Shumway and Stoffer [26] introduced an EM algorithm to estimate unknown parameters for the linear dynamic systems when the observation matrix C is known, such as in Model (1). The EM algorithm has gradually become a standard estimation tool for SSMs and related models [27]. In the EM algorithm, the Kalman filter is employed to estimate state variables in the E step and the maximum likelihood method is used to estimate unknown parameters in the M step.

We propose a novel three-step procedure, the Expectation-Regularization-Maximization (ERM) algorithm, to estimate the high-dimensional sparse system matrix A, and other parameters, as well as the state variables for linear SSMs. The procedure is outlined as follows:

  • E Step: the conditional expectation of the likelihood (4) is calculated by (5) where θ(r−1) are the estimated parameters at the (r − 1)th iteration. In addition, the state variables (xt) and their sufficient statistics (functions of xt) required for estimating unknown parameters in the R and M step, are also estimated through the Kalman filter and smoother at this step.
  • R Step: the L1 regularization, or the adaptive LASSO method, is employed to obtain the estimate of the sparse system matrix A denoted by A(r).
  • M Step: the MLE of other model parameters θ* = (Q, R, μ, Σ), denoted by θ*(r), is obtained by maximizing the conditional expectation of likelihood (6)

For the standard SSM, the EM algorithm starts from the E Step for a given initial value of the system matrix A based on the prior knowledge of the dynamic system. However, for the high-dimensional linear SSM, it is not feasible to provide a good initial value for a high-dimensional sparse system matrix A. Thus, we recommend to start the proposed ERM algorithm from the R Step, which depends on an initial estimation of the state variable (xt) that can be obtained by a nonparametric local polynomial or spline smoother [28] instead of the Kalman filter. Thus, the proposed ERM algorithm should follow the order of R-M-E steps iteratively, or one R step, then E-R-M steps iteratively, until the log-likelihood estimates converge. The detailed implementation for each step is discussed in the following subsections.

E step: Kalman filtering and smoothing.

The following sufficient statistics required for unknown parameter estimation in the R and M steps can be computed via the Kalman filter and smoother, E(xt|y1, …, yT), and , which are denoted by , and ; Var(xt|y1, …, yτ) by and by , where τ is an arbitrary time point. The Kalman filter and smoother involve forward and backward recursions. In the forward recursions, estimation of the current states (filtering) and prediction of the next state are made based on the past measurements. For the backward recursions (smoothing), the past states are estimated given all the measurements up to the very last time point. Namely, the state variables are estimated by . The general Kalman filtering and smoothing algorithms are given as follows.

  • Forward recursions: Prediction and Filtering (7) (8) As we mentioned earlier, in the first iteration of the E step, A will be estimated by an initial R step. The initial state mean can be replaced by a small nonzero initial vector such as (0.1, 0.1, …, 0.1), and the variance matrix can be replaced by a small diagonal matrix, such as 10−5 × Ip×p. R, Q can be initialized by two identity matrices—these two parameters will be updated in the M step.
  • Backward recursions: Smoothing (9) where is initialized by .

Notice that the above algorithms require p × p matrix inverse calculations, which is computationally heavy and numerically unstable for a high dimensional system. To avoid such problem, Kojima et al [24] derived a recursive formula from the blockwise matrix inversion theorem. Matrices in Eq (8) and in Eq (9) can be expressed alternatively as,

Note that the inverse of the matrix in brackets above are of the same form (B′ΔB + D−1)−1, where D is a symmetric n × n matrix, B is an arbitrary n × n matrix, and Δ is a diagonal matrix with diagonal elements δ1, …, δn. Let bi denote the ith row vector of B, i = 1, …, n and D0 = D, we can use the following recursive formula, to calculate the inverse matrix (B′ΔB + D−1)−1, which is given by Dn. Thus, and Jt−1 in Eqs (8) and (9) can be calculated by the recursive formula without the inverse matrix calculations.

R step: L1 regularization using the adaptive LASSO.

The implementation of the R Step in the proposed ERM algorithm is critical. The adaptive LASSO estimates can be obtained efficiently by using the computationally efficient LARS algorithm [29], which needs to be customized to the ERM algorithm. The extended BIC (eBIC) for large model selection proposed by [30] is employed to select the tuning parameters in the adaptive LASSO method. This model selection criterion is shown to be consistent under some mild conditions and also meets the needs of variable selection for larger model spaces [30].

LASSO [31] is a popular L1 regularization technique for performing estimation and variable selection simultaneously, which is consistent only under relatively restrictive mathematical assumptions. The adaptive LASSO proposed by [32], where adaptive weights are used for penalizing different coefficients in the L1 penalty, enjoys the desired oracle property under much weaker assumptions, namely, it performs as well as if the true underlying model were given in advance and produces asymptotically unbiased estimators for the nonzero parameters in linear regression models.

The adaptive LASSO estimates can be implemented by using the LARS algorithm [29]. In order to modify and apply the LARS algorithm for the R Step in the proposed ERM algorithm for linear SSMs, we need to use the vectorized matrix notations [33, 34], i.e., denote (10) (11) (12) (13) where vec is the stack operator and ⊗ is the Kronecker product; ai is the ith row vector of A; α is the vectorized A which is a (p2 × 1) vector; e is a (p(T − 1) × 1) vector that represents measurement errors.

The matrix-based ERM algorithm

In the standard EM algorithm for linear SSMs, the estimate of A, the system matrix, can be obtained by maximizing the conditional expectation of the likelihood function (5), which is equivalent to minimizing in vectorized notations. This produces the MLE estimator of A as (14)

This estimator for the high-dimensional sparse matrix A is over parameterized and requires that is invertible. In this study, we propose to use an L1-regularized estimator of A that minimizes (15) where λ is a tuning parameter. This is equivalent to applying the LASSO method to the restructured state equation in (2) in a matrix pseudo-regression form (16)

We call this a pseudo-regression model since X and Z are state variables estimated from SSMs, instead of measured response variables and covariates in a standard regression model. Note that the elements of e are independent due to the assumption that Q is diagonal. If the state variables were directly observable without measurement error, model (16) is a standard first-order VAR model and one could simply apply the LASSO method to this VAR model [33, 34]. However, for the linear SSM, we need to use the sufficient statistics, , , and obtained from the E step to evaluate E(ZZ) and E(ZX) and the corresponding LARS algorithm needs to be modified accordingly. If T − 1 > p, the maximum likelihood estimator of A defined in Eq (14) is root-n-consistent and can be employed to determine the adaptive weights in the adaptive LASSO procedure as follows (17)

For “large p, small n” problems (T − 1 ⩽ p), we adopt a method developed for sparse high-dimensional regression in [35] by using the following marginal estimator of A instead of (18)

Here is a diagonal matrix of which the diagonal elements matches those of . It was shown in [35] that under mild assumptions, is a zero-consistent estimator of A; using as weight for adaptive LASSO can achieve oracle efficiency in variable selection.

Once α is estimated by adaptive LASSO, the system matrix , can be easily reconstructed by “reshaping” the (p2 × 1) vector α into a (p × p) matrix. The elements in are shrunk toward zero as the L1 penalty parameter λ increases. Some elements are shrunk to exact zeros when λ is sufficiently large. Thus, it is important to determine the tuning parameter λ appropriately. We find that the use of standard AIC, BIC, cross-validation and other classical methods for determining λ tends to select a larger model for a high-dimensional model with sparse data, which may diminish the parsimonious property that we aim to preserve. The extended BIC [30] is recommended since it contains an extra penalty term with the consideration of different prior distributions over the model space.

M step: Maximization.

The estimation of the remaining parameters is straightforward. The estimates of (Q, R, μ, Σ) can be obtained by maximizing the expected conditional likelihood (6) with given sufficient statistics from the E step and from the R step, i.e., (19) (20)

The row-based ERM algorithm

Note that in the above R Step implementation, α is a (p2 × 1) vector and Z is a (p2 × p2) matrix. Thus, it involves p2-dimensional matrix manipulations and computations. When p is large, it requires a massive amount of computer memory to carry out these computations, which may cause out-of-memory crashes of the proposed ERM algorithm. In this subsection, we propose a row-based approach for the R Step in the proposed ERM algorithm. Denote Xi as the ith row of X*, ai as the ith row of A, and ei as the ith row of e*. The adaptive LASSO estimate of A can be obtained by equivalently applying the adaptive LASSO to a linear pseudo-regression model, (21) where is a (T − 1) × p matrix. Thus, the adaptive LASSO estimate of ai (i = 1, …, p) is (22) which only needs (p × p)-dimensional (instead of (p2 × p2)-dimensional) matrix manipulations. We use the same adaptive weights as in the matrix-based algorithm. Similarly, the sufficient statistics , and from the E step are used in this step. Although we need to repeat the above LASSO procedure for each row i = 1, 2, …, p, the row-based ERM algorithm takes less time and less memory compared to the matrix-based algorithm. Below we describe our simulation studies done for comparisons of the performance between the row-based and matrix-based algorithms.

Results

Simulation studies

The proposed ERM algorithm for high-dimensional SSMs to construct dyamic MINs invovles estimation and regularization of a large number of parameters. We designed simulation studies to evaluate the methodology and the implementation procedure. We compared the row-based ERM algorithm to the matrix-based ERM algorithm, and we also evaluated the performance of the row-based ERM algorithm in more detail.

The row-based algorithm was proposed to overcome the computational limitation of the matrix-based algorithm as discussed in the previous section. We design the first simulation study for different number of dimensions p = 8, 20, 50, and 80. Thus, the total number of elements in the system matrix A is 64, 400, 2500, and 6400, respectively. The number of nonzero elements in A is assumed as 15, 35, 84, and 150 for the four cases, respectively. The nonzero elements were randomly generated from ±(0.4, 0.5, 0.6, 0.7, 0.8, 0.9). We also assume the variance parameters as Q = I and R = 0.1 × I for all the cases and the number of time points T = 60. As suggested, we applied and started the ERM algorithm from the R step to the 100 simulated data sets (M = 100).

To evaluate the performance of the proposed ERM algorithm for variable selection, we calculated the false positive rate (FP) and false negative rate (FN) of by (23) where P is the number of nonzero elements and N is the number of zero elements in A, and is an indicator function. In this simulation experiment, we fixed Q and R as their true values in order to have a fair comparison for the two algorithms. We report the average FP and FN over M = 100 simulation runs in Table 1.

thumbnail
Table 1. Simulation results: Comparisons of variable selection performance between the row-based and matrix-based ERM algorithms.

https://doi.org/10.1371/journal.pone.0187822.t001

Table 1 shows that both row-based and matrix-based algorithms produce reasonable results. The matrix-based algorithm tends to yield a smaller false positive rate, but a larger false negative rate compared to that of the row-based algorithm. The false positive rate of the row-based algorithm is controlled very well although it is slightly larger than that of the matrix-based algorithm. The false negative rates for both algorithms are always higher (much higher for some cases) than the false positive rates. More importantly, a regular desktop machine running MATLAB ran out of memory estimating the 80×80 A matrix when the matrix-based algorithm is used, whereas the row-based algorithm can still perform reasonably well. Thus, we suggest using the row-base ERM algorithm for practical applications due to its efficiency and capability to handle high-dimensional matrices.

In the third simulation experiment, we evaluate the performance of the proposed row-based ERM algorithm for different sample sizes and the effect of system noise Q. Since the system noise Q and measurement noise R cannot be identified simultaneously based on the single sequence data without replication, we decided to fix R in this simulation to avoid the identifiability problem. The true system matrix for the SSM in this simulation experiment is a 41×41 system matrix A. We generated equally-spaced temporal data with the number of time points T = 20, 50 and 100 for each bacterial OTU. We expect to see the improved estimation with increased sample sizes. For the variance of the system noise Q, we compared the two cases: fixed as the true value or estimated from the data. Total of M = 100 data sets were simulated for each scenario. The simulation results are reported in Table 2.

thumbnail
Table 2. Evaluation of the row-based ERM algorithm for variable selection with respect to number of time points T and Q estimation. p = 41.

https://doi.org/10.1371/journal.pone.0187822.t002

Table 2 shows that, as the number of time points T increased from 20 to 100, the false negative rate significantly decreased from 0.31 or 0.35 to 0.092 or 0.098 for fixed or estimated Q, while the false positive rate roughly was stabilized at 0.015 to 0.019. Overall, for all three choices of T, fixing the system noise Q as the true value rather than estimating Q only reduced the false negative rate slightly, and it did not have much effect on the false positive rate. The adaptive LASSO procedure equipped with the eBIC method for tuning parameter selection seems to have enough power to identify important state variables in the SSMs. With more data, it becomes less conservative and selects more variables with a higher accuracy as expected. In summary, the proposed row-based ERM algorithm produces promising results for SSM variable selection and is computationally efficient. The false positive rate is not affected by the system noise or measurement noise, but the false negative rate can be improved if the true system noise or measurement noise can be accurately determined.

Applications to microbiota data

In one recent longitudinal microbiome study, mid-vaginal swabs from 32 nonpregnant, reproductive-age women were obtained twice weekly, over a period of 16 weeks [4] during which each subject’s sexual and menstrual activity was also tracked. For each of the samples, variable regions of 16S ribosomal RNA gene were sequenced, yielding abundance estimates of bacterial species/genera (OTUs). Although sexual activity and menses showed impact on bacterial diversity, no clear relationships between environment and abundance of specific types or species of bacteria were identified, suggesting a likely causal role of the inter-microbial interactions. Several studies have characterized the normal bacterial communities in human vagina [4, 3638]. Normal vaginal flora can be clustered into five to six groups based on their composition, most of which are dominated by lactic acid bacteria, and remaining few by anaerobic bacteria. Many members of the vaginal flora are highly specialized for the vaginal eco-niche indicating that the vaginal microbiome generates meaningful MINs [39]. We employ the proposed models to investigate the dynamic interactions among bacteria in this study, and infer the MINs for each subject.

Based on our simulation studies, we decided to use the extended BIC [30] to preserve the parsimonious property, and employ the row-based ERM algorithm for adaptive LASSO. As a matter of fact, we also tried the matrix-based ERM algorithm with extended BIC for this particular data set. Whilee the matrix-based algorithm yielded reasonable results in simulation studies, it shrank almost all the coefficients in the system matrix to zero for both subjects used in our study, which resulted in very poor fits. In contrast, row-based algorithm produced reasonable results in terms of fitting, and can be used effectively to infer MINs.

The magnitude of abundance varies widely for different bacteria. For example, for Subject 6, relative abundance of Atopobium (averaged over 27 days) is 1,542 times greater than that of L. crispatus, which is a known important beneficial species in vagina [40, 41]. Without proper standardization, a uniform L1 penalty is much more likely to set the edges related to less abundant OTUs to zero, and results in a simplistic network dominated by a few most abundant OTUs In order to make the results comparable, the measurements are standardized before the row-based ERM method is applied, i.e., we define , where is the jth raw measurement for the ith subject, is the mean of , and is the standard deviation of .

Ideally, time course microbiome data with technical or biological replicates at each time point for each subject would allow investigators to use statistical methods to provide more reliable MIN structure identification and estimation. Microbiome studies often lack such replicates. Our method is still useful in this scenario, although the variance of the system noise and the measurement noise may not be estimable and identified simultaneously for such datasets.

Inference of microbial interaction networks (MINs)

We applied the ERM method to infer MINs from the data published by [4]. Although we examined the available data for all the 32 subjects from this vaginal microbiome study, due to space limitation, here we report the results from two subjects: subject 15 and subject 6, because these two subjects cover very different vaginal microbiome profiles. For any given subject, the abundance measurements for most bacteria are zero. For consistency and simplicity, we selected only those bacteria for which at least 30% of the abundance measurements across the 16 weeks of study are nonzero. Based on this criterion, Subject 6 had 34 and Subject 15 had 12 bacterial OTUs which were identified for modeling.

Figs 1 and 2 present the one-step-ahead prediction for these two subjects, based on the estimated SSM model. We also calculated the coefficients of determination (R2) for all the bacteria involved in the MINs for each subject (in lower right corner of each plot in Figs 1 and 2). The predictions look reasonable and there is no apparent evidence of overfitting, which is a common pitfall in high-dimensional data analysis.

thumbnail
Fig 1. One-step-ahead prediction for subject 15.

Each cell represents prediction for a different OTU. Solid lines depict the predicted values whereas circles indicate standardized temporal abundances of OTUs. Abbreviations for Operational Taxonomic Units: LIN Lactobacillus iners; LCR Lactobacillus crispatus; SNE Sneathia sp.; LJE Lactobacillus jensenii; GAR Gardnerella sp.; ANA Anaerococcus sp.; STR Streptococcus sp.; COR Corynebacterium sp.; FIN Finegoldia sp.; LOT5 Lactobacillus otu5; URE Ureaplasma sp.; LVA Lactobacillus vaginalis.

https://doi.org/10.1371/journal.pone.0187822.g001

thumbnail
Fig 2. One-step-ahead prediction for subject 6.

Each cell represents prediction for a different OTU. Solid lines depict the predicted values whereas circles indicate standardized temporal abundances of OTUs. Abbreviations for Operational Taxonomic Units: LIN Lactobacillus iners; LCR Lactobacillus crispatus; ATO Atopobium sp.; PRE Prevotella sp.; PAR Parvimonas sp.; SNE Sneathia sp.; GAR Gardnerella sp.; PEPN Peptoniphilus sp.; ANA Anaerococcus sp.; STR Streptococcus sp.; COR Corynebacterium sp.; RUM3 Ruminococcaceae 3; MOB Mobiluncus sp.; AER Aerococcus sp.; FIN Finegoldia sp.; MEG Megasphaera sp.; GEM Gemella sp.; PEPS Peptostreptococcus sp.; LAC2 Lachnospiraceae 10; LAC11 Lachnospiraceae 11; MOR Moryella sp.; COR3 Coriobacteriaceae 3; DIA Dialister sp.; SHU Shuttleworthia sp.; PEPT Peptococcus sp.; ALL Allisonella sp.; LAC10 Lachnospiraceae 2; POR Porphyromonas sp.; BUL Bulleidia sp.; ACT Actinomyces sp.; COR1 Coriobacteriaceae 1; CLO12 Clostridiales 12; SEG Segniliparus sp.; BAC2 Bacteroidales 2.

https://doi.org/10.1371/journal.pone.0187822.g002

The inferred MIN for the two subjects, are reported in Tables 3 and 4, as well as in Figs 3 and 4.

thumbnail
Table 3. Interactions among bacteria for subject 15.

See legend for Fig 1 for OTU abbreviations.

https://doi.org/10.1371/journal.pone.0187822.t003

thumbnail
Table 4. Interactions among bacteria for subject 6.

See legend for Fig 2 for OTU abbreviations.

https://doi.org/10.1371/journal.pone.0187822.t004

thumbnail
Fig 3. Microbial interaction network (MIN) for subject 15.

Blue and red arrows indicate directed positive and negative effects respectively. Arrow width indicates effect magnitude. Circles highlight bacterial species that impact multiple other species in the MIN and whose critical role in the MIN has either experimental support in literature (L. iners) or has never been recognized before (Finegoldia sp.).

https://doi.org/10.1371/journal.pone.0187822.g003

thumbnail
Fig 4. Microbial interaction network (MIN) for subject 6.

Blue and red arrows indicate directed positive and negative effects respectively. Arrow width indicates effect magnitude. Circles highlight bacterial species that impact multiple other species in the MIN and whose critical role in the MIN has either experimental support in literature (L. iners) or has never been recognized before (Finegoldia sp. and L. crispatus).

https://doi.org/10.1371/journal.pone.0187822.g004

The number of edges from each bacterial OTU ranges from one to seven. There are a total of 38 (25.7%) nonzero elements in the reconstructed system matrix A for subject 15, and 37 (3.3%) nonzero elements in the system matrix A for subject 6, both of which are sparse matrices and consistent with the sparsity property of MINs. As reported in [4], we note that the composition of vaginal flora of these two subjects differs significantly: subject 15 microbiome is dominated by Lactobacillus iners, and that of subject 6 is dominated by many anaerobic bacteria. Gajer et al. [4] classified vaginal microbiota of their subjects into five community types based on their bacterial compositions. Community types I to III were found to be dominated by L. crispatus, Lactobacillus gasseri and L. iners, respectively. Community type IV-A were dominated by moderate proportions of either of the many Lactobacillus species typically found in vagina. Community type IV-B tends to be dominated by a collection of diverse bacteria, all of which are present in large abundance. According to this classification, the vaginal microbiome of Subject 15 is dominated by community state type III, and that Subject 6 is dominated by community type IVB. Parts of the MINs that we obtained from these two subjects are in agreement with some previous experimental investigations into interactions among vaginal bacteria, and some inferred relationships are novel and unexpected.

Subject 15.

We find that in the vaginal flora of Subject 15, L. iners is not just the predominant member, but also actively inhibits the proliferation of other lactobacilli, most prominently, that of Lactobacllus jensenii (Fig 3). L. iners reportedly shares reciprocal interference with a different lactobacillus (L. gasseri), but our results indicate that this interference may be phylogenetically more widespread [42]. We also find that L. jensenii actively aids the proliferation of Gardnerella Sp., which has been implicated in bacterial vaginosis.

Subject 6.

First we infer that Finegoldia sp., an anaerobic bacterium belonging to class Clostridia directs the growth of multiple anaerobic bacteria that are more abundant than itself. The synergistic interactions of Finegoldia sp. with Sneathia sp. and with Anarococcus sp. were also identified in MIN of Subject 15. In fact Finegoldia sp. seems to occupy an influential position in both the MINs, in spite of its low abundance in both the subjects. This is one of the classical hallmarks of a “keystone” species in an ecosystem [43], and Finegoldia spp. has never been noted as one before. Second, Gajer et al. [4] identified that communities dominated by L. iners often appear to shift to one dominated by Atopobium, Prevotella, Parvimonas, Sneathia, Gardnerella, or Mobiluncus. We find that even in a microbial community in which L. iners is a minor component, it promotes the growth of some of these bacteria directly as well as indirectly, as in the inferred MIN of Subject 6. Third, L. crispatus, an aerobic, hydrogen peroxide producing bacterium that is normally associated with establishing normal and aerobic microflora, is inferred to be promoting growth of a number of facultatively anaerobic Lachnospiraceae species, which is a novel and unexpected outcome of the inferred MIN, and may warrant future experimental validations.

Discussion

Dimensionality has always been a difficulty in identifying a complex microbial interaction network (MIN) due to the large number of bacteria observed in human microbiome. In this article, we proposed a new Expectation-Regularization-Maximization (ERM) algorithm for a high-dimensional linear state space model (SSM) to construct dynamic MINs from time course microbiome data. The Kalman filters in conjunction with the adaptive LASSO variable selection technique enable promising parameter estimation for the sparse high-dimensional microbiome interaction matrix. The implementation of the adaptive LASSO for the SSMs in the R step is shown to be straightforward and clean. To overcome the difficulty for large matrix manipulations for the high-dimensional SSMs, a row-based ERM algorithm was proposed and evaluated against the standard matrix-based algorithm. Our simulation and real data analysis results show that the row-based algorithm performs quite well. The proposed method can be easily adapted to accommodate data with longitudinal replicates. We successfully applied the method to time course vaginal microbiome data to construct dynamic MINs.

Similar VAR models with L1 regularization have been proposed for dynamic networks [33, 34]. However, the VAR model does not take measurement error into consideration. The SSMs equipped with the proposed ERM algorithm allow us to consider the system error and measurement error separately while constructing the dynamic network. Through the ERM algorithm, the variance of both errors can be estimated if the longitudinal data or biological replicates are available. A similar SSM representation with L1 regularization for dynamic gene regulatory network construction was proposed in [24]. However, an ad hoc method instead of the LARS algorithm was used in LASSO estimation which may reduce its performance. Furthermore, the program, or its implementation details, were not provided. In this paper, we fill the gap to clarify the methodological issues and provide a complete and simple implementation procedure. We also discussed how to initiate the proposed ERM algorithm from a practical perspective. From the computational perspective, we validated that the row-based ERM algorithm performs well for data analysis and recommended for practical use. Based on the proposed SSM and ERM algorithm, we establish the MIN for vaginal microbiome of women and some encouraging findings have been revealed.

We believe that our work is just the first step to reconstruct MIN using SSM model. Below we list a few weaknesses of our current method and possible directions for future work.

The state-space model considers time as discrete steps instead of a continuous variable as used in many alternative network models. As such, it is more resilient to temporal discontinuities and can be used to fit granular data and data with sharp jumps (discontinuities), which is demonstrated in Figs 1 and 2. That being said, we must point out that both system and measurement noises (wt and vt in Eq (1)) are modeled as Gaussian distributions in our current model, therefore we strongly suggest that our method should only be applied to modeling the interactions between key OTUs with relatively low sparsity. It will be very interesting to incorporate a proper discrete distribution such as the negative binomial distribution into the SSM model in the future. Secondly, we have only considered estimating MINs for each individual subjects. It is more meaningful to construct the common MIN for a population. This calls for more advanced SSM methods and data with more replications and better quality. Although convergence was not a big issue for the proposed algorithm in both simulations and real data analyses, we acknowledge that the addition of the R step may change the theoretical properties of the EM algorithm, which warrants further investigations. Exogenous variables such as gender, age, race et al can also be included in a more complex model. Nonlinear SSM models have been widely studied in recent years. The extension to nonlinear SSM models in conjunction with variable selection techniques also deserve further investigation. It would be interesting to compare the proposed model and method to alternative models and methods for dynamic network construction. Finally, we must point out that due to the large number of unknown parameters (k2 edges) in a high-dimensional network, inevitably there will be a certain number of false positives despite the best practice in model selection. It is therefore critical to conduct subsequent confirmatory experiments to validate the predicted interactions. The power of high-dimensional network models, such as the one proposed in this study, is that they help experimentalists generate high-quality hypotheses and select the most promising experiments to perform.

Acknowledgments

This research was supported in part by NIH 5 P30 AI078498-08, NIH AI087135, NIH 5 T32AI083206-05, NIH/NCATS UL1 TR002001, and NIH/NIAID HHSN272201200005C.

This work is part of the Ph. D. thesis of I.C. We would like to thank the following individuals for their expert contributions to the studies described in this manuscript: Hongmei Yang, Jeanne Holden-Wiltse, Sanjukta Bandyopadhyay, Lu Wang, Hongjun Li, and James Java.

References

  1. 1. Faust K, Raes J. Microbial interactions: from networks to models. Nature Reviews Microbiology. 2012;10(8):538–550. pmid:22796884
  2. 2. Grice EA, Segre JA. The skin microbiome. Nature Reviews Microbiology. 2011;9(4):244–253. pmid:21407241
  3. 3. Faith JJ, Guruge JL, Charbonneau M, Subramanian S, Seedorf H, Goodman AL, et al. The long-term stability of the human gut microbiota. Science. 2013;341(6141):1237439. pmid:23828941
  4. 4. Gajer P, Brotman RM, Bai G, Sakamoto J, Schütte UM, Zhong X, et al. Temporal dynamics of the human vaginal microbiota. Science translational medicine. 2012;4(132):132ra52–132ra52. pmid:22553250
  5. 5. Consortium HMP, et al. Structure, function and diversity of the healthy human microbiome. Nature. 2012;486(7402):207–214.
  6. 6. Grice EA, Kong HH, Conlan S, Deming CB, Davis J, Young AC, et al. Topographical and temporal diversity of the human skin microbiome. science. 2009;324(5931):1190–1192. pmid:19478181
  7. 7. Arumugam M, Raes J, Pelletier E, Le Paslier D, Yamada T, Mende DR, et al. Enterotypes of the human gut microbiome. nature. 2011;473(7346):174–180. pmid:21508958
  8. 8. Clemente JC, Ursell LK, Parfrey LW, Knight R. The impact of the gut microbiota on human health: an integrative view. Cell. 2012;148(6):1258–1270. pmid:22424233
  9. 9. David LA, Maurice CF, Carmody RN, Gootenberg DB, Button JE, Wolfe BE, et al. Diet rapidly and reproducibly alters the human gut microbiome. Nature. 2014;505(7484):559–563. pmid:24336217
  10. 10. Biswas S, McDonald M, Lundberg DS, Dangl JL, Jojic V. Learning microbial interaction networks from metagenomic count data. In: Research in Computational Molecular Biology. Springer; 2015. p. 32–43.
  11. 11. Barberán A, Bates ST, Casamayor EO, Fierer N. Using network analysis to explore co-occurrence patterns in soil microbial communities. The ISME journal. 2012;6(2):343–351. pmid:21900968
  12. 12. Berry D, Widder S. Deciphering microbial interactions and detecting keystone species with co-occurrence networks. Frontiers in microbiology. 2014;5. pmid:24904535
  13. 13. Marino S, Baxter NT, Huffnagle GB, Petrosino JF, Schloss PD. Mathematical modeling of primary succession of murine intestinal microbiota. Proceedings of the National Academy of Sciences. 2014;111(1):439–444.
  14. 14. Stein RR, Bucci V, Toussaint NC, Buffie CG, Rätsch G, Pamer EG, et al. Ecological modeling from time-series inference: insight into dynamics and stability of intestinal microbiota. PLoS Comput Biol. 2013;9(12):e1003388. pmid:24348232
  15. 15. Steinway SN, Biggs MB, Loughran TP Jr, Papin JA, Albert R. Inference of Network Dynamics and Metabolic Interactions in the Gut Microbiome. PLOS Comput Biol. 2015;11(6):e1004338. pmid:26102287
  16. 16. Faust K, Sathirapongsasuti JF, Izard J, Segata N, Gevers D, Raes J, et al. Microbial co-occurrence relationships in the human microbiome. PLoS Comput Biol. 2012;8(7):e1002606–e1002606. pmid:22807668
  17. 17. Chaffron S, Rehrauer H, Pernthaler J, von Mering C. A global network of coexisting microbes from environmental and whole-genome sequence data. Genome research. 2010;20(7):947–959. pmid:20458099
  18. 18. Fisher CK, Mehta P. Identifying keystone species in the human gut microbiome from metagenomic timeseries using sparse linear regression. PLoS ONE. 2014;9(7):e102451. pmid:25054627
  19. 19. Harvey AC. Forecasting, structural time series models and the Kalman filter. Cambridge university press; 1990.
  20. 20. Durbin J, Koopman SJ. Time series analysis by state space methods. 38. Oxford University Press; 2012.
  21. 21. West M, Harrison J. Bayesian Forecasting and Dynamic Models. In: Griliches Z, MD’ I, editors. Handbook of Econometrics. North-Holland, Amsterdam: Springer; 1997.
  22. 22. Quach M, Brunel N, d’Alché Buc F. Estimating parameters and hidden variables in non-linear state-space models based on ODEs for biological networks inference. Bioinformatics. 2007;23(23):3209–3216. pmid:18042557
  23. 23. Rangel C, Angus J, Ghahramani Z, Lioumi M, Sotheran E, Gaiba A, et al. Modeling T-cell activation using gene expression profiling and state-space models. Bioinformatics. 2004;20(9):1361–1372. pmid:14962938
  24. 24. Kojima K, Yamaguchi R, Imoto S, Yamauchi M, Nagasaki M, Yoshida R, et al. A state space representation of VAR models with sparse learning for dynamic gene networks. Genome Informatics. 2009;22:56–68.
  25. 25. Kalman RE. A new approach to linear filtering and prediction problems. Journal of Fluids Engineering. 1960;82(1):35–45.
  26. 26. Shumway RH, Stoffer DS. An approach to time series smoothing and forecasting using the EM algorithm. Journal of time series analysis. 1982;3(4):253–264.
  27. 27. Ghahramani Z, Hinton GE. Parameter estimation for linear dynamical systems. Technical Report CRG-TR-96-2, University of Totronto, Dept. of Computer Science; 1996.
  28. 28. Lu T, Liang H, Li H, Wu H. High-dimensional ODEs coupled with mixed-effects modeling techniques for dynamic gene regulatory network identification. Journal of the American Statistical Association. 2011;106(496). pmid:23204614
  29. 29. Efron B, Hastie T, Johnstone I, Tibshirani R, et al. Least angle regression. The Annals of statistics. 2004;32(2):407–499.
  30. 30. Chen J, Chen Z. Extended Bayesian information criteria for model selection with large model spaces. Biometrika. 2008;95(3):759–771.
  31. 31. Tibshirani R. Regression shrinkage and selection via the lasso. Journal of the Royal Statistical Society Series B (Methodological). 1996;p. 267–288.
  32. 32. Zou H. The adaptive lasso and its oracle properties. Journal of the American statistical association. 2006;101(476):1418–1429.
  33. 33. Hsu NJ, Hung HL, Chang YM. Subset selection for vector autoregressive processes using lasso. Computational Statistics and Data Analysis. 2008;52(7):3645–3657.
  34. 34. Ren Y, Zhang X. Subset selection for vector autoregressive processes via adaptive Lasso. Statistics & probability letters. 2010;80(23):1705–1712.
  35. 35. Huang J, Ma S, Zhang CH. Adaptive Lasso for sparse high-dimensional regression models. Statistica Sinica. 2008;p. 1603–1618.
  36. 36. Ravel J, Gajer P, Abdo Z, Schneider GM, Koenig SS, McCulle SL, et al. Vaginal microbiome of reproductive-age women. Proceedings of the National Academy of Sciences. 2011;108(Supplement 1):4680–4687.
  37. 37. Romero R, Hassan SS, Gajer P, Tarca AL, Fadrosh DW, Bieda J, et al. The vaginal microbiota of pregnant women who subsequently have spontaneous preterm labor and delivery and those with a normal delivery at term. Microbiome. 2014;2(1):18. pmid:24987521
  38. 38. Ravel J, Brotman RM, Gajer P, Ma B, Nandy M, Fadrosh DW, et al. Daily temporal dynamics of vaginal microbiota before, during and after episodes of bacterial vaginosis. studies. 2013;19:20.
  39. 39. Danielsson D, Teigen PK, Moi H. The genital econiche: focus on microbiota and bacterial vaginosis. Annals of the New York Academy of Sciences. 2011;1230(1):48–58. pmid:21824165
  40. 40. Antonio MA, Meyn LA, Murray PJ, Busse B, Hillier SL. Vaginal colonization by probiotic Lactobacillus crispatus CTV-05 is decreased by sexual activity and endogenous Lactobacilli. The Journal of infectious diseases. 2009;199(10):1506–1513. pmid:19331578
  41. 41. Stapleton AE, Au-Yeung M, Hooton TM, Fredricks DN, Roberts PL, Czaja CA, et al. Randomized, placebo-controlled phase 2 trial of a Lactobacillus crispatus probiotic given intravaginally for prevention of recurrent urinary tract infection. Clinical infectious diseases. 2011;52(10):1212–1217. pmid:21498386
  42. 42. De Backer E, Verhelst R, Verstraelen H, Alqumber MA, Edelman S, Burton JP et al. Quantitative determination by real-time PCR of four vaginal Lactobacillus species, Gardnerella vaginalis and Atopobium vaginae indicates an inverse relationship between L. gasseri and L. iners BMC microbiology. 2007;7(1):1:.
  43. 43. Paine RT. A conversation on refining the concept of keystone species. JSTOR; 1995.