 Methodology article
 Open access
 Published:
Dimension reduction with gene expression data using targeted variable importance measurement
BMC Bioinformatics volume 12, Article number: 312 (2011)
Abstract
Background
When a large number of candidate variables are present, a dimension reduction procedure is usually conducted to reduce the variable space before the subsequent analysis is carried out. The goal of dimension reduction is to find a list of candidate genes with a more operable length ideally including all the relevant genes. Leaving many uninformative genes in the analysis can lead to biased estimates and reduced power. Therefore, dimension reduction is often considered a necessary predecessor of the analysis because it can not only reduce the cost of handling numerous variables, but also has the potential to improve the performance of the downstream analysis algorithms.
Results
We propose a TMLEVIM dimension reduction procedure based on the variable importance measurement (VIM) in the frame work of targeted maximum likelihood estimation (TMLE). TMLE is an extension of maximum likelihood estimation targeting the parameter of interest. TMLEVIM is a twostage procedure. The first stage resorts to a machine learning algorithm, and the second step improves the first stage estimation with respect to the parameter of interest.
Conclusions
We demonstrate with simulations and data analyses that our approach not only enjoys the prediction power of machine learning algorithms, but also accounts for the correlation structures among variables and therefore produces better variable rankings. When utilized in dimension reduction, TMLEVIM can help to obtain the shortest possible list with the most truly associated variables.
Background
Gene expression microarray data are typically characterized by large quantities of variables with unknown correlation structures [1, 2]. This high dimensionality has presented us challenges in analyzing the data, especially when correlations among variables are complex. Including many variables in standard statistical analyses can easily cause problems such as singularity and overfitting, and sometimes is not even doable. To manage this problem, the dimensionality of the data will often be reduced in the first step. There are multiple ways to achieve this goal. One is to select a subset of genes based on certain criteria such that this subset of genes is believed to best predict the outcome. This gene selection strategy is typically based on some univariate measurement related to the outcome, such as ttest and rank test [3, 4]. Another strategy is to use a weighted combination of genes of lower dimension to represent the total variation of the data. Representative approaches are principle component analysis (PCA) [5] and partial least squares (SLR) [6–9]. Machine learning algorithms such as LASSO [10, 11] and Random Forest [12] have embedded capacity to select variables while simultaneously making predictions, and can be used to accommodate high dimensional microarray data.
As always, there is no onesizefitsall solution to this problem, and one often needs to resort to a mixandmatch strategy. The univariatemeasurement based gene selection is a very popular approach in the field. It is fast and scales easily to the dimension of the data. The output is usually stable and easy to understand, and fulfills the objectives of the biologists to directly pursue interesting findings. However, it often relies on oversimplified models. For instance, the univariate analysis evaluates every gene in isolation of others, with the unrealistic assumption of independence among genes. As a result, it carries a lot of noise and the selected genes are often highly correlated, which themselves create problems in subsequent analysis. Also, due to the practical limit of the size of the gene subset, real informative genes with weaker signals will be left out. In contrast, PCA/PLS constructs a few gene components as linear combinations of all genes in a dataset. This "Super Gene" approach assumes that the majority of the variation in the dataset can be explained by a small number of underlying variables. One then uses these gene components to predict the outcome. These approaches can better handle the dependent structure of genes and their performances are quite acceptable [13]. But it is harder to interpret gene components biologically, and to assess the effect of individual genes one needs to look at the weight coefficients of the linear combination. Machine learning algorithms are very attractive variable selection tools to deal with large quantities of genes. They are prediction algorithms with embedded abilities to select gene subsets. However, whether or not a gene is chosen by a learning algorithm may not be the best measurement of its importance. Machine learning algorithms are constructed to achieve an optimal prediction accuracy, which often overlooks the importance of each variable. Consequently, small changes in data or tuning parameters may result in big changes in variable rankings and the the selected gene subsets are instable. For example, Random Forest, a treebased nonparametric method, has a variable importance measurement that greatly contributes to its popularity. This measurement is sensitive to the parameter choices of trees in the presence of high correlations among variables, because different sets of variables can produce nearly unchanged prediction accuracy [14, 15]. Another example is LASSO  one of the most popular regularization algorithms. Assuming a sparse signal, LASSO handles the high dimensionality problem by shrinking the coefficients of most variables towards zero [16]. A recent implementation of LASSO is in the GLMNET R package [17]. The package uses a coordinate descent algorithm and can finish an analysis of 20,000 variables within a few seconds. To us, its result is somewhat sensitive to the choice of the penalizing parameter λ. Different λ s may result in gene subsets with little overlapping. In the mean time, variable importance measurements are not readily available in LASSO. One can simply rank genes by their coefficients, but this can be quite subtle. Although permutation tests may be used to derive pvalues, how to perform the permutation is a tricky matter due to selection of tuning parameters. For small pvalues, it is still computationally infeasible. In this paper, inspired by concepts of counterfactual effects from the causal inference literature, we propose a targeted variable importance measurement [18, 19] to rank genes and reduce the dimensionality of the dataset. Counterfactuals are usually defined in the context of treatment to disease. It is the outcome a patient would have had a treatment been assigned differently, with everything else held the same. Hence counterfactuals are "counter"fact and apparently impossible to be observed. But it can be estimated statistically. Suppose that we have an outcome Y, a binary treatment A, and the confounding variables W of A, and we have worked out correctly an estimate of the conditional expectation of Y given A and W. A common way to estimate the counterfactual effect of A is to compute the difference between the and the for every observation and then average over all observations, referred to as the Gcomputation method [20]. Although counterfactuals may not be completely relevant to gene microarray data, thinking about the data in this way is very helpful for us to assess the importance of a gene. Our VIM definition uses the concepts of counterfactuals and the estimation framework is built on the methodology of targeted maximum likelihood estimation (TMLE) [21]. By tailoring this recently developed technique specifically to gene expression data, we hope to introduce to the community an alternative strategy to carry out gene selection in addition to current methods. Our approach takes the advantage of prediction power of learning algorithms while targeting at the individual importance of each variable. Its mathematical property has been studied in [22], and we will focus on its application. In brief, our approach consists of twostages. In the first stage, we predict the outcome given all genes. In the second stage, we improve the first stage by modeling the mechanism between an individual gene and its confounding variables. Both stages can be very flexible ranging from using univariate analysis to refined learning algorithms. When machine learning algorithms are used, we have the flexibility to determine how to make predictions without restricting ourselves to explicit models and distributions. In the meanwhile, as in the case of the univariate analysis, we return to a simple and well interpretable measure of the importance of each gene. This importance measurement is derived in the presence of the confounding variables of a gene, and hence can help to exploit the redundant information among correlated genes. It is generally also more stable than the variable importance produced by machine learning algorithms. In addition, our approach provides a simple way for statistical inference based on asymptotic theories, and is well suited for the exploratory analysis of microarray data.
Methods
Suppose the observed data are i.i.d. O_{ i } = (Y_{ i }, A_{ i }, W_{ i }), where Y is a continuous outcome, A is the gene of interest, W is a set of confounding variables of A, and i = 1, ⋯, n indexes the observation. Let Ψ(a) represents the variable importance measurement (VIM) of A. One can define the VIM of A as the marginal effect of A on the outcome Y at value A = a relative to A = 0 adjusted for W, and then averaged over the distribution of W [18]:
Consider the semiparametric regression model:
where f(W) is a function of W. With this parameterization, we have Ψ(a) = βa. We can then view β as an index of the VIM of A. In the above model, the only assumption we make is the linearity of A. The definition of the VIM of A is closely related to the definition of the counterfactual effect in causal inference [23]. Although β can not be directly interpreted as an causal effect without proper assumptions [19], it serves well as a surrogate of the magnitude of the causal relationship between the outcome and a gene. The motivation of this parameterization is that by selecting more causally related genes, the resulting prediction function will be better generalized to new experiments with the same causal relation between the outcome Y and A, but a different joint distribution of W. If in a next experiment, the technology or the sampling population is somewhat different, but the causal mechanism is still the same, then a prediction function that uses the correlates of the true causal variables will perform poorly while a prediction function using the true causal variables will still perform nicely. This idea will be illustrated in our simulations.
Our goal is to estimate β. In [22], this estimation problem was addressed in the framework of targeted maximum likelihood estimation (TMLE). TMLE is an estimating equation and efficient estimation theory based methodology [24], and is particularly useful when it comes to semiparametric models. Estimators from the traditional method such as MLE perform well for parametric models, however, they are generally biased relative to their variances especially when the model space is large. This is because the MLE focuses on doing a good job on the estimation of the whole density rather than on the parameter itself. TMLE is designed to achieve an optimal tradeoff between the bias and the variance of the estimator. It uses an MLE framework, but instead of estimating the overall density, TMLE targets on the parameter of interest and produces estimators minimally affected by changes of the nuisance parameters in a model. In Additional File 1 we provide a brief overview of this methodology with a demo simulation example. The formal mathematical formulation of TMLE can be found in the original paper by van der Laan and Rubin [21]. The implementation of TMLE to estimate β is fairly simple and consists of two stages. First, we estimate E(Y A, W) without any parametric restriction. We then regress the residual of Y and E(Y A = 0, W) onto βA to conform with our semiparametric regression model. This will yield an initial estimator of β and fitted values of E(Y A, W), denoted by and the . In the second stage, we update these initial estimates in a direction targeted at β. This involves regressing the residuals of Y and the fitted on the clever covariate A  E(AW). The E(AW) evaluates the confounding of A with W, and we name it the "gene confounding mechanism". It needs to be estimated if unknown. Let us denote the coefficient before the clever covariate as ε. The updated TMLE estimate of β is , where ε_{ n } is the estimated value of ε. The variance estimate of β can be computed from its efficient influence curve. Below is a stepbystep implementation of our algorithm, and we refer to it as the TMLEVIM procedure.

1.
Obtain the initial estimator and . Use your favorite algorithm here, for example, linear regression, LASSO, Random Forest, etc.

2.
Obtain the g_{ n }(W) estimate for the gene confounding mechanism E(AW ). As in the case of Q_{ n }(0), a broad spectrum of algorithms can be used. In this paper, we use LASSO (in the GLMNET R package) for its optimal speed.

3.
Compute the "clever covariate":

4.
Fit regression .

5.
Update the initial estimate with
and update the initial fitted values with

6.
Compute the variance estimate for according to its efficient influence curve:
where i indexes the ith observation.

7.
Construct the test statistic:
T(A) follows the standard Gaussian distribution under the null hypothesis β = 0 when the sample size n goes to infinity.
The TMLE estimator is a consistent estimator of β when either the or the g_{ n }(W ) is consistent. When the is consistent, it is also asymptotically efficient. The derivations of the clever covariate, the efficient influence curve, the TMLE estimate and its mathematical properties can be found in [22] and [18]. Upon the construction of the test statistic, a pvalue can be calculated for the adjusted marginal effect of A and used as an index of the variable importance.
In the application to dimension reduction, for each variable in the dataset, we compute a TMLEVIM pvalue. We then reduce our variable space based on these pvalues. There are two notions. First, in principle, a separate initial estimator should be fitted for every gene A by forcing A as a term in the algorithm used. This can become quite time consuming. To solve the problem, instead of estimating E(Y A, W) for each A, we obtain a grand estimate G_{ n }(V) for E(Y V ). Here V represents all variables in the dataset. Then for every A in V, we carry out the regression Y ~ βA with the offset G_{ n }(A = 0) to get β_{ n }(0) and . Second, when obtaining the g_{ n }(W), we want to be attentive to how closely W is correlated with A. The independence between W and A results in zero adjustment to the initial estimator, while a complete association causes β to be unidentifiable. A simple option is to use all variables less than a predefined correlation with A as W. In [22], they authors suggest 0.7 as a conservative threshold. Instead of applying a universal cutoff, we can also set individualized correlation threshold for each A. Below we provide a data adaptive procedure to do it. One first defines a sequence of correlation cutoffs δ. For each choice of δ, one computes the corresponding TMLE pvalue for A. One then sets a pvalue threshold λ, and chooses the maximum δ that has produced a pvalue less than λ. The degree of the protection is determined with the value of λ. In general, the smaller the λ is, the more the protection. The value of λ can be either fixed a priori or chosen by cross validation. We refer to it as the TMLEVIM(λ) procedure. It allows us to adjust for the confounding in the dataset adaptively and flexibly, and protect the algorithm against the harm from high correlations among variables. It works best when many variables are closely correlated in a complex way. However, it does require more time to run, especially when λ needs to be chosen by cross validation. In many cases, a universal cutoff of 0.7 will work fine. In Additional File 1 we provide the mathematical formulation of the TMLEVIM(λ) procedure. Once we have all the variables ranked by their pvalues. the candidate list can be truncated by either applying a pvalue threshold or taking the top k ranked variables. Both of them are sound practices. In our simulations and data analysis, we usually truncate the list at a pvalue threshold 0.05.
Results and Discussion
Simulation studies
We performed two sets of simulations. The first set of simulations investigates how TMLEVIM responds to changes in the number of confounding variables, the correlation level among variables, and the noise levels. The second set studies the TMLEVIM with more complex correlation structures and model misspecification. The performance of the dimension reduction procedure was primarily evaluated by the achieved prediction accuracy using a prediction algorithm on the reduced sets of variables, illustrated in the following analysis flow:
Two prediction algorithms, LASSO and D/S/A (Deletion/Substitution/Addition) [25], were used. D/S/A searches through the variable space and selects the best subset of covariates by minimizing the cross validated residual sum of squares. In our simulations, LASSO and D/S/A predictions are often similar. We used D/S/A in simulation I as it provides convenience to count what variables are included in the prediction model. LASSO was used in simulation II for its faster speed. We also used multivariate linear regression (MVR) as a comparison to machine learning algorithms when applicable.
Part I
In simulation I, we varied the number of noncausal variables (W), the correlation coefficient ρ among variables, and the noise level to see how TMLEVIM responds to them. For each simulated observation O_{ i } = (Y_{ i }, A_{ i }, W_{ i }), where i indexes the ith sample, the outcome Y_{ i } was generated from a main effect model of 25 A s:
where j indexes the jth A, and e_{ i } is a normal error with mean 0 and variance . Each A_{ j } was correlated with m W s, and hence the total number of W s is m_{ w } = 25 m. A_{ j } and its associated W s were jointly sampled from a multivariate Gaussian distribution with mean 0 and variancecovariance matrix S, where S is a correlation matrix with an exchangeable correlation coefficient ρ. This simulation scheme resulted in 25 independent clusters of covariates. Within each cluster, the covariates are correlated at level ρ.
Simulations were run for combinations of:

m = (10, 20) corresponding to m_{ w }= (250, 500);

σ_{ e }= (1, 5, 10);

and ρ = (0.1, 0.3, 0.5, 0.7, 0.9).
For each combination, we simulated a training set of 500 data points and a testing set of 5000 data points. The training set was used to obtain the prediction model while the testing set was used to calculate the L2 risk. We also calculated a crossexamined L2 risk using a testing set with a ρ other than that of the training set. This is to demonstrate that by identifying more causally related variables, TMLEVIM is robust to the change of the joint distribution among the covariates A s and W s. In specific, for each prediction model obtained from a training set, we calculated the L 2 risk on the testing set generated with ρ = 0.1 regardless of what ρ was used to generate the training set. As a benchmark, we also used univariate regression in parallel with TMLEVIM to reduce the dimensionality of the dataset, denoted with URVIM. Once the variable importance was calculated, we cut short the variable list using a pvalue threshold 0.05. Each combination was replicated 10 times and results took the average.
TMLEVIM used LASSO to obtain both the initial estimator and the gene confounding mechanism estimator g_{ n }(W ). In the g_{ n }(W ), W was all the variables excluding A. TMLEVIM has demonstrated a consistent advantage over URVIM with respect to the final prediction error over a range of simulation settings. This is particularly the case when the joint distribution of the covariates changes and when predictions were made by MVR that lacks internal capacity of model selection. Smaller σ_{ e }, larger m_{ w }, and larger ρ tend to magnify this advantage. Also, TMLEVIM risks have smaller standard errors than the URVIM risks. In Table 1, we present our simulation results for five different ρ values and two different m_{ w } values, with fixed at 5. The following summary quantities are reported:

R_{ r }= (URVIM risk  TMLEVIM risk)/ UR risk: the proportion of the risk reduction of TMLEVIM relative to the URVIM risk. It measures by how much TMLEVIM outperforms URVIM. The bigger the number, the more the advantage.

R_{ A }= TMLEVIM N_{ A }/ URVIM N_{ A }: the ratio of the number of A s (N_{ A }) in the TMLEVIM list to the number of A s in the URVIM list.

R_{ W }= TMLEVIM N_{ W }/ URVIM N_{ W }: the ratio of the number of W_{s} (N_{ W }) in the TMLEVIM list to the number of W s in the URVIM list.

RR_{DSA} = TMLEVIM P_{ A }/ URVIM P_{ A }: the ratio of the proportion of A s (P_{ A }) in the final D/S/A prediction model resulted from the TMLEVIM procedure to that from the URVIM. It measures the relative chance of arriving at a truly associated variable in the final model through the path of TMLEVIM, referenced to the URVIM.
The R_{ r } was calculated on two different testing sets. One is the testing set generated with the same ρ as the corresponding training set, and we refer it to "testing set (a)"; the other is the testing set generated with ρ = 0.1, and we refer it to "testing set (b)". Testing set (a) shares the same correlation structure as the training set, while in testing set (b) all the variables are essentially independent of each other. Testing set (b) is a simple representation of the scenario that when a new experiment is conducted the overall joint distribution of the covariates changes while the causal mechanism remains the same. In Table 1, the bold R_{ r } was calculated on testing set (a), and the Italic R_{ r } was on testing set (b). We make a few points here about Table 1:

The proportion of the risk reduction (R_{ r }) of the TMLEVIM relative to the URVIM is typically more than 20% for the MVR prediction and 10% for the D/S/A prediction. In some cases, the risk reduction of the MVR can be very significant. For example, when m_{ w }= 500 and ρ = 0.7, the TMLEVIM risk is close to only half of the URVIM risk. TMLEVIM tends to deliver more advantages when m_{ w }= 500 than when m_{ w }= 250. When the correlation coefficient ρ increases, the TMLEVIM performs increasingly better than the URVIM for the MVR prediction. For the D/S/A prediction, small or large ρ s seem to benefit most from the TMLEVIM. For intermediate ρ, the benefit is still there but reduced. We believe that how much the risk can be reduced by the TMLEVIM is a combination of factors such as the number of A s and W s in the reduced candidate list, the correlation structures among covariates and the internal optimization procedures of D/S/A. The advantage of the TMLEVIM over the URVIM does seem to be more significant on the testing set (b) than the testing set (a), in support of our hypothesis that by identifying more causally related variables the TMLEVIM results generalize better to new experiments.

Most R_{ A }values are slightly higher than 1 while the R_{ W }values are much smaller. This indicates that on average, in the TMLEVIM list, the number of correctly identified A s is slightly higher than that in the URVIM list, while the number of falsely associated W_{s} is much less. It is especially the case when the correlations are high among variables. The low counts of false positives is a major contributing factor that the prediction made on the TMLEVIM candidate list is better than that on the URVIM.

As to the number of A s that are finally made into the D/S/A prediction model, the TMLEVIM in most cases displays a slight advantage over the URVIM. A closer look reveals that the variables included in the D/S/A model only differs by one or two between the TMLEVIM and the URVIM. But the prediction risk has a measurable difference. This probably implies that every single variable counts in making good predictions in these simulations.

When ρ = 0.9, the situation seems to be losing its track. The TMLEVIM did worse than the URVIM in terms of correctly identified variables as well as the prediction risk of the testing set (a). Considering the high correlations among variables, this could possibly be attributed to the overfitting in the g_{ n }(W ). Indeed, in [22], the authors showed that TMLE deteriorates when adjusting for variables with correlation coefficients beyond 0.7. However, the RR_{ DSA }indicates that the chance of including a correct variable in the final D/S/A model based on the TMLEVIM list is higher than that on the URVIM. Further looking into the data, we found out that the number of A s that made into the D/S/A model from the TMLEVIM list is actually greater than that from the URVIM, while the number of W s is much less. Henceforth, the D/S/A model built on the TMLEVIM list is closer to truth, but somehow its prediction is worse than the model built on the URVIM list. This seems to suggest that when provided with the URVIM list, the D/S/A has offset its model for the missed A s from highly correlated W s, while for the TMLEVIM, this can not be done since there are not many W s in the list. It is the same reason that the URVIM underperforms the TMLEVIM on the testing set (b) when those surrogates of A s were lost. For the MVR, although the TMLEVIM shows a dominant advantage over the URVIM with respect to the prediction accuracy, the TMLEVIM only identified 77% (when m_{ w }= 250) and 57% (when m_{ w }= 500) of the A s identified by the URVIM. The better prediction is merely due to the fact that the MVR breaks down when too many variables entered the model. This is particularly the case when m_{ w }= 500.
Figure 1 presents a graphical representation of a typical example in simulation I with (σ_{ e } = 5, m_{ w } = 250, ρ = 0.7). Besides the advantage displayed by the TMLEVIM relative to the URVIM, we also see much smaller differences between the TMLEVIM risks of the testing set (a) and (b) compared to the URVIM because TMLEVIM was able to detect more A s. In summary, when confounder are properly adjusted in g_{ n }(W ), TMLEVIM improves not only the performance of relatively simple algorithms such as the MVR, but also the more complex learning algorithms with builtin capacities of variable selection. Interested readers can find all the original prediction risks and counts of A s and W s and their standard errors in Additional File 2 for this simulation.
Part II
Simulation II examines the TMLEVIM on largerscale datasets with much more complex correlation structures. The simulation consists of 500 samples and 1000 variables. We used a correlation matrix derived from the top 800 genes in a real dataset published in [26]. For these genes, the median absolute correlation coefficient was centered at 0.26, the 1st/3rd quartile being 0.16/0.37, and the maximum as high as 0.9977. Hence, simulation II tried to mimic the correlation structure in this real data set. The outcome Y was generated from two different models using 20 A s. One is a linear model, and the other is polynomial.
Details of this simulation is provided in the Additional File 1. A test dataset of 5000 points were simulated to assess the L2 prediction risk. We repeated the simulation for 10 times and results took the average. In TMLEVIM, we tried two different initial estimators. One is the univariate regression as simple as Y ~ A, and the other is the LASSO estimator. LASSO was also used to get the g_{ n }(W ) and to make the final predictions. we adjusted universally in the g_{ n }(W ) for the variables that are correlated with A with an arbitrary correlation coefficient less than 0.7. All the and g_{ n }(W ) models were mainterm linear. Hence, with the polynomial outcome, we could examine how TMLEVIM performs when misspecified models were provided. To summarize the result, we computed a R^{2} quantity, representing the proportion of explained variance relative to an intercept model. It is defined as 1  mean risk/MST, where and is the mean of Y. Table 2 lists the R^{2} and the number of true positives (T.P.) and false positives (F.P.) in the reduced list of candidate variables, for the URVIM, the , and the . Compared to URVIM, TMLEVIM improved the prediction risk by providing LASSO a candidate list with more truly and less falsely associated variables for both the linear and polynomial simulations. It is worth noting that even with an initial estimator as simple as the univariate regression , TMLEVIM still achieves a significant increase in R^{2} by modeling the g_{ n }(W ).
The numbers in Table 2 were based on candidate lists that were cut short with a pvalue threshold of 0.05. In Table 3, we provide the results based on the top 100 ranked genes. The numbers of URVIM and the are less satisfying than those in Table 2, while the achieved comparable results. This suggests that the pvalues of A s are among the smallest ones, and shortening the length of the list does not affect the final result. Regardless of the weakened results, The still displays a nonignorable advantage over the URVIM with respect to the prediction accuracy, while the number of correctly identified A s is slightly smaller than that of the URVIM. We then looked at the correlation matrix among the top 100 selected genes, and it occurs that the correlation among them is the least for the , the most for the URVIM, and the lies in between. This could explain why the does a better job in prediction regardless of less A s.
We also carried out the TMLEVIM(λ) procedure with LASSO as the initial estimator, allowing the data select the correlation cutoff for variables to be adjusted in the g_{ n }(W ). Results are also reported in Table 2 and Table 3. TMLEVIM(λ) identified more A s but also more W s, and the prediction accuracy is only slightly improved. On the other hand, the correlations among the selected top 100 variables are quite small. It seems by data adaptively adjusting for the correlation levels in the g_{ n }(W), TMLEVIM(λ) returns a more independent set of genes. The actual risks and standard errors are contained in Additional File 2.
Data Analysis
Breast cancer patients are often put on chemotherapy after the surgical removal of the tumor. However not all patients will respond to chemotherapy, and proper guidance for selecting the optimal regimen is needed. Gene expression data have the potential for such predictions, as studied in [26]. The dataset from [26] contains the gene expression profiling on 22283 genes for 133 breast cancer patients. The outcome is the pathological complete response (pCR). This is a binary response associated with longterm cancer free survival. There are also 13 clinical variables collected in the dataset including the ER (estrogen receptor) status, which is a very significant clinical indicator for chemotherapy response.
The goal of the study is to select a set of genes that best predict the clinical response pCR. The first step is to reduce the number of genes worth of consideration, and we applied both URVIM and TMLEVIM (with Q^{(0)} = UR and Q^{(0)} = LASSO) for this purpose. For the TMLEVIM(Q^{(0)} = LASSO), the was estimated by LASSO using the top 5000 ranked genes. We then took all the genes with the FDRadjusted pvalues less than 0.005 [27], as suggested in the original paper, and upon them we built a predictor using the Random Forest (tuning parameters mtry = number of variables/3, ntree = 3000 and nodesize = 1). The clinical covariates were treated in the same way as genes. To prevent the algorithm from breaking down, we only adjusted for the confounder with correlation coefficients less than 0.7 with A in the g_{ n }(W ). We carried out a 10fold honest cross validation. We divided the dataset into 10 subsets. Each subset was regarded as a validation set and the rest as the training set. We reperformed the entire analysis, i.e. VIM calculation → dimension reduction → Random Forest classifier, on all 10 training sets and predicted the outcome of the validation set using the classifier built on the training samples. We can then use these cross validated predictions to assess the true classification accuracy of our algorithm.
Analysis results are tabulated in Table 4. The URVIM produced a candidate list of 326 genes and one clinical variable the "ER status", while the list of the TMLEVIM(Q^{(0)} = UR) consists of 660 genes and TMLEVIM(Q^{(0)} = LASSO) 818 genes. The TMLEVIM identified many more genes than the URVIM. Among all the identified genes, 429 overlap between the and , 15 overlap between the URVIM and TMLEVIM(Q^{(0)} = UR), 10 overlap between the URVIM and TMLEVIM(Q^{(0)} = LASSO), and only 4 genes are shared among all three (please see Figure 2). The TMLEVIM appeared to have selected almost a different set of genes than the URVIM.
The TMLEVIM(Q^{(0)} = UR) and the TMLEVIM(Q^{(0)} = LASSO) results are quite similar to each other regardless of the adequate difference between the initial estimators. It seems the modeling of the g_{ n }(W ) had played a significant role and steered away the initial univariate estimates. Further investigation found out that genes in the URVIM list are highly correlated with the clinical indicator ER status, while the TMLEVIM genes are not. Consequently, the TMLEVIM genes are less correlated to each other than the URVIM genes. Looking at the first 100 ranked genes, the absolute median of the correlation coefficients for the URVIM is 0.43, while for the TMLEVIM, it is about half of that number. Although the input variables to the Random Forest are different, The cross validated (CV) classification accuracy are quite similar among these three methods. We also passed all 22,000 genes to Random Forest and looked at its variable importance measurement. The Random Forest VIM (RFVIM) is more similar to the URVIM: about 50% of them overlap but only a few overlap with the TMLEVIM. The RFVIM genes are also highly correlated with the ER status, albeit the less severity than the URVIM. Its OOB classification accuracy (0.7669) is comparable with all three other methods.
In summary, the URVIM and RFVIM seemed to have identified genes that are strong predictors of the clinical variable ER status. The ER status is a strong indicator of the outcome pCR. Hence, the final prediction accuracy still seems quite good. The TMLEVIM has identified a list of genes of which a small proportion is strong predictors of ER status and others are not associated with the ER status. Its prediction accuracy is slightly better than that of the URVIM and RFVIM.
Conclusions
We have shown in this paper with extensive simulations that the TMLE based variable importance measurement can be incorporated into a dimension reduction procedure to improve the quality of the list of the candidate variables. It requires an initial estimator and a gene confounding mechanism estimate g_{ n }(W). A consistent ensures the consistency and the efficiency of the TMLE estimate. When is not consistent, a correct specification of g_{ n }(W) can still produce consistent estimates while that estimate will not be efficient any more. We generally recommend to do as good a job as we can on obtaining the , as a better means both a smaller bias and a smaller variance. Nevertheless, algorithms as simple as univariate regression are also valid choices, and in this case, we will rely solely on the goodness of g_{ n }(W ). The computation of g_{ n }(W ) directly affects the speed of the TMLEVIM, as it has to be redone for every variable. Hence, one may want to choose an approach that is reasonably fast. In our study, we chose the GLMNET R Package as our primary tool to get g_{ n }(W ), and it worked very well. In practice, one needs to balance the resources used for the initial estimator and the gene confounding mechanism. With a proper design of the two estimating stages, TMLEVIM is a fairly fast procedure. It is also worth mentioning that the TMLEVIM can sometimes be sensitive to the overfitting in the , and hence, caution needs to be exercised when choosing an aggressive algorithm.
A popular dimension reduction approach is the principle component analysis (PCA). The PCA computation does not involve the outcome, and so it could be less powerful when prediction is the primary goal. Its output is a linear combination of all the genes. Though not a gene selection approach, we still carried it out on our simulation I data as an interesting comparison to our approach. PCA demonstrates an intermediate performance with respect to the URVIM and the TMLEVIM on small pvalue cutoffs. This means a few top components carry all the prediction power. When the pvalue cutoff is increased, and more components enter the candidate list, its results became quite unsatisfying. When the correlation structure changes among the genes, PCA has done a poor predicting job. The PCA results are contained in Additional File 3.
Usually, the reduced set of variables will serve as the input of a prediction algorithm to build a model. Such algorithms used in this article include MVR, LASSO, and D/S/A. We have noticed that in most of our simulations, the MVR prediction often achieves a similar risk as LASSO and D/S/A on the TMLEVIM reduced set of variables. It suggests that further variable selection may not be necessary for the TMLEVIM candidate list, and we can use simpler algorithms to get a good prediction. In fact, the TMLEVIM can go beyond the scope of dimension reduction. It can be iteratively applied to the data until it converges to a list of several variables that are most likely to be causal to the outcome. In this case, one may want to use the Super Learner [28] as the prediction algorithm, which works more effectively with the TMLEVIM. The Super Learner is an ensemble learner that combines predictions from multiple candidate learners with optimal weights. It has been shown in [29] that the Super Learner performs asymptotically equal to or better than any of its candidate learners. The Super Learner allows the data to objectively blend results from different algorithms rather than relying on a single algorithm chosen subjectively by an analyst. Hence it enjoys a greater flexibility to explore the model space and usually produces reasonable predictions consistently across a wide variety of datasets, and serves as a very good prediction algorithm for the TMLEVIM. On the other hand, it is also more computationally demanding.
TMLEVIM is a quite general approach. Besides gene expression data, TMLEVIM can also be applied to genetic mapping problems. The genomewide association studies (GWAS) can involve more than a million of genetic markers. In this case, only the univariate analysis seems to be feasible of ranking every marker. With the TMLEVIM procedure, we can run more complex algorithms on a subset of top ranked markers, taking it as the initial estimator, and then evaluate every single marker. The variable importance of each marker is thus obtained through a multimarker approach and being adjusted for its confounder. However, the GWAS in human beings is usually casecontrol data, and the current TMLEVIM needs to be extended to accommodate such outcomes.
References
West M, Blanchette C, Dressman H, Huang E, Ishida S, Spang R, Zuzan H, Olson JA, Marks J, J N: Predicting the Clinical Status of Human Breast Cancer using Gene Expression Profiles. Proc Natl Acad Sci USA 2001, 98: 11462–11467. 10.1073/pnas.201162998
Dudoit S, Fridlyand J, Speed T: Comparison of discrimination methods for the classification of tumors using gene expression data. JASA 2002, 97: 77–87.
Hedenfalk I, Duggan D, Chen Y, Radmacher M, Bittner M, Simon R, Meltzer P, Gusterson B, Esteller M, Rafield M, Yakhini Z, A BD, Dougherty E, Kononen J, Bubendorf L, Fehrle W, Pittaluga S, Gruvberger D, Loman N, Johannsson O, Olsson H, Wilfond B, Sauter G, Kallioniemi O, Borg A, Trent J: Gene expression profiles in hereditary breast cancer. New England Journal of Medicine 2001, 244: 539–548.
Dettling M, Buhlmann P: Boosting for tumor classification with gene expression data. Bioinformatics 2003, 19: 1061–1069. 10.1093/bioinformatics/btf867
Ghosh D: Singular value decomposition regression modeling for classification of tumors from microarray experiments. Proceedings of the Pacific Symposium on Biocomputing 2002, 18–29.
Nguyen DV, Rocke DM: Tumor classification by partial least squares using microarray gene expression data. Bioinformatics 2002, 18: 39–50. 10.1093/bioinformatics/18.1.39
Nguyen DV, Rocke DM: Multiclass cancer classification via partial least squares with gene expression profiles. Bioinformatics 2002, 18: 1216–1226. 10.1093/bioinformatics/18.9.1216
Huang X, W P: Linear regression and twoclass classification with gene expression data. Bioinformatics 2003, 19: 2072–2978. 10.1093/bioinformatics/btg283
Boulesteix A: PLS Dimension reduction for classification with microarray data. Statistical applications in genetics and molecular biology 2004, 3: 1–33.
R T: Regression shrinkage and selection via the lasso. J Royal Statist Soc B 1996, 58: 267–288.
Efron B, Hastie T, Johnstone I, R T: Least angle regression. Annals of Statistics 2004, 32: 407–499. 10.1214/009053604000000067
L B: Random forests. Machine Learning 2001, 45: 5–32. 10.1023/A:1010933404324
Dai JJ, Lieu L, Rocke D: Dimension Reduction for Classification with Gene Expression Microarray Data. Statistical Applications in Genetics and Molecular Biology 2006, 5: Article 6.
Strobl C, Boulesteix AL, Zeileis A, Hothorn : Bias in random forest variable importance measures: illustrations, sources and a solution. BMC Bioinformatics 2007, 8: 25. 10.1186/14712105825
Strobl C, Boulesteix AL, Kneib T, Augustin T, Zeileis A: Conditional variable importance for random forests. BMC Bioinformatics 2008, 9: 307. 10.1186/147121059307
Rosset S, Zhu J, Hastie T: Boosting as a regularized path to a maximum margin classifier. Journal of Machine Learning Research 2004, 5: 941–973.
Friedman J, Hastie T, Tibshirani R: Regularization paths for generalized linear models via coordinate descent. Journal of Statistical Software 2010., 33:
Yu Z, van der Laan MJ: Measuring treatment effects using semiparametric models. U.C. Berkeley Division of Biostatistics Working Paper Series 2003. [http://www.bepress.com/ucbbiostat/paper136]
van der Laan MJ: Statistical inference for variable importance. Int J Biostat 2006, 2: Article 2.
Robins JM: A new approach to causal inference in mortality studies with sustained exposure periods  application to control of the healthy worker survivor effect. Mathematical Modelling 1986, 7: 1393–1512. 10.1016/02700255(86)900886
van der Laan MJ, Rubin DB: Targeted maximum likelihood learning. Int J Biostat 2006, 2: Article 11.
Tuglus C, van der Laan MJ: Targeted methods for biomarker discovery, the search for a standard. U.C. Berkeley Division of Biostatistics Working Paper Series 2008. [http://www.bepress.com/ucbbiostat/paper233]
Rubin DB: Estimating Causal Effects of Treatments in Randomized and Nonrandomized Studies. Journal of Educational Psychology 1974, 66: 688–701.
Bickel PJ, Klaassen CAJ, Ritove Y, Wellner JA: Efficient and adaptive estimation for semiparametric models. Baltimore: The Johns Hopkins University Press; 1993.
Sinisi SE, van der Laan MJ: Deletion/Substitution/Addition algorithm in learning with applications in genomics. Statistical Applications in Genetics and Molecular Biology 2004, 3: Article 18.
Hess KR, Anderson K, Symmans WF, Valero V, Ibrahim N, Mejia JA, Booser D, Theriault RL, Buzdar AU, Dempsey PJ, Rouzier R, Sneige N, Ross JS, Vidaurre T, Gomez HL, Hortobagyi GN, Pusztai L: predictor of sensitivity to preoperative chemotherapy with paclitaxel and fluorouracil, doxorubicin, and cyclophosphamide in breast cancer. J Clin Oncol 2006, 24: 4236–4244. 10.1200/JCO.2006.05.6861
Benjamini Y, Hochberg Y: Controlling the false discovery rate: a practical and powerful approach to multiple testing. J Royal Statist Soc B 1995, 57: 289–300.
van der Laan MJ, Polley EC, Hubbard AE: Super Learner. Statistical Applications in Genetics and Molecular Biology 2007, 6: Article 25.
van der Laan MJ, Dudoit S, van der Vaart AW: The crossvalidated adaptive epsilonnet estimator. Statistics and Decisions 2006, 24: 373–395. 10.1524/stnd.2006.24.3.373
Acknowledgements
The authors want to thank Cathy Tugulus for sharing her codes and her helpful comments on this work. The authors also thank the reviewers for their precious appraisal of the earlier version of this manuscript. This work was by NIH R01 AI074345. The authors declare no conflicts of interest.
Author information
Authors and Affiliations
Corresponding author
Additional information
Competing interests
The authors declare that they have no competing interests.
Authors' contributions
MvdL conceived the project and designed the algorithm. HW implemented the algorithm, designed the simulation studies, and collected and analyzed the data. All authors participated in drafting the manuscript.
Authors’ original submitted files for images
Below are the links to the authors’ original submitted files for images.
Rights and permissions
This article is published under license to BioMed Central Ltd. This is an Open Access article distributed under the terms of the Creative Commons Attribution License (http://creativecommons.org/licenses/by/2.0), which permits unrestricted use, distribution, and reproduction in any medium, provided the original work is properly cited.
About this article
Cite this article
Wang, H., van der Laan, M.J. Dimension reduction with gene expression data using targeted variable importance measurement. BMC Bioinformatics 12, 312 (2011). https://doi.org/10.1186/1471210512312
Received:
Accepted:
Published:
DOI: https://doi.org/10.1186/1471210512312