Abstract
Exploring features associated with the clinical outcome of interest is a rapidly advancing area of research. However, with contemporary sequencing technologies capable of identifying over thousands of genes per sample, there is a challenge in constructing efficient prediction models that balance accuracy and resource utilization. To address this challenge, researchers have developed feature selection methods to enhance performance, reduce overfitting, and ensure resource efficiency. However, applying feature selection models to survival analysis, particularly in clinical datasets characterized by substantial censoring and limited sample sizes, introduces unique challenges. We propose a robust ensemble feature selection approach integrated with group Lasso to identify compelling features and evaluate its performance in predicting survival outcomes. Our approach consistently outperforms established models across various criteria through extensive simulations, demonstrating low false discovery rates, high sensitivity, and high stability. Furthermore, we applied the approach to a colorectal cancer dataset from The Cancer Genome Atlas, showcasing its effectiveness by generating a composite score based on the selected genes to correctly distinguish different subtypes of the patients. In summary, our proposed approach excels in selecting impactful features from high-dimensional data, yielding better outcomes compared to contemporary state-of-the-art models.
Introduction
Next-generation sequencing (NGS) techniques () can provide us with information on the expression of more than 30,000 genes, which helps researchers understand gene regulations and interactions to find treatments for diseases. However, the number of genes associated with a particular disease is small (). Therefore, we need to develop powerful tools to select genes that work as a group and are associated with clinical outcomes. Feature selection approaches were developed to choose the most relevant and informative features for research questions from the original raw set of features; therefore, they can help avoid overfitting, reduce training time, handle the challenge of dimensionality, and simplify data representations.
Survival analysis () is a statistical model studying time-to-event data in which the event may not be observed (censored) during the study because of loss to follow-up or early end of the study. Due to the presence of censoring, which is a unique characteristic in survival analysis, there is a need to develop novel techniques to work with feature selections for survival data, especially for high-throughput gene expression data in which most of the potential predictors are unimportant, with nearly no effect on the outcome (). The Cox proportional hazards model is the most commonly used technique for analyzing survival data. However, it was not designed for high-dimensional datasets with a large number of predictors. Lasso (Least Absolute Shrinkage and Selection Operator) introduces a penalty term to the Cox model’s likelihood function, which penalizes the absolute values of the regression coefficients. By forcing some coefficients to be exactly zero, Lasso effectively performing variable selection. In addition, there are models tailored to effectively handle situations where the number of features outweighs the number of observations (). Machine learning techniques that inherently handle high-dimensional data have been adapted to handle censored data, offering more flexible alternatives for analyzing high-dimensional, right-censored, heterogeneous data. However, unlike statistical models based on a mathematical framework, machine learning approaches do not impose a specified relationship on the predictors and outcomes and rely mainly on–data-driven algorithms, which makes it hard to interpret results. Furthermore, a lot of feature selection methods for survival analysis use a scoring model () to measure variations of features to select important features. Since the scoring algorithm was developed specifically to take care of the data censors and tie events of survival data, the results are biased () which may lead to selecting nonimportant features and provide a less accurate prediction.
We introduce a robust and effective “Pseudo-variables Assisted Group Lasso” method built on the ensemble idea, i.e., “more heads are better than one”, where features obtained from different selectors are aggregated to enhance the final selection. Moreover, we incorporated pseudo-variables which we know are irrelevant to the outcome and the permutation technique to assist the selection. The ensemble and pseudo-variables are nicely embedded into the Group Lasso model to yield the final output. Among aggregated features, only the features that consistently show stronger signals than the pseudo-variables (known noises) across permutations will be selected. We used colorectal cancer data from The Cancer Genome Atlas (TCGA) for illustration of our proposed approach. In addition, we performed simulation studies based on two different settings, where the first one mimicked the colorectal cancer data, and the second considered more complicated situations under various scenarios. For each simulation, we first simulated gene expression data for hundreds of genes and then generated survival outcomes based on some causal genes. The proposed feature selection ensemble method was applied to “uncover” the causal genes and compared to the existing methods.
Materials and methods
Colorectal cancer data set from TCGA database
Raw gene expression counts were downloaded from colon cancer () datasets using The Cancer Genomics Cloud (); additional clinical metadata was downloaded from cBioportal (). The mRNA-Seq data from TCGA was produced using the Illumina HiSeq 2000 platform and processed by the RNAseqV2 pipeline, which used MapSplice for alignment and RSEM for quantification.
A robust feature selection ensemble
The proposed pseudo-variable-assisted feature ensemble procedure has two major steps: 1) aggregating the feature selection results from multiple feature selectors (Figure 1A) and 2) fitting a group Lasso model on the identified feature set with a new permutation-assisted tuning strategy (Figure 1B). In the second step, the group is defined based on the correlation structure, ensuring that features are highly correlated within each group.
FIGURE 1
Aggregating the results from different feature selection approaches is a critical step in ensemble learning. The outputs of the different approaches can be various, either the subsets of selected features, the rankings of all features, or both. We applied the same scheme as in () to obtain the ranked feature set depending on the types of outputs (Figure 1A), where the final rank is an aggregation from each ranking. We assume that the observations are , where is a G-dimensional vector in which each feature has its aggregated rank, and is a survival outcome. Without loss of generality, we assume the G features are quantitative variables (e.g., gene expressions). However, the proposed method can be applied to categorical or mixed-type variables. Similar to (), we can rewrite the G-dimensional vector as with of dimension , , based on their correlation structure such that within each block, the absolute value of pairwise correlation is all greater than a correlation threshold .
Next we consider a Group Lasso model () on the ranked feature set (Figure 1B) for survival outcomes. For commonly seen right censored survival data, is a survival outcome, where , , with and denote the event time of interest and the censoring time for the th subject, respectively. We model the relationship between the survival outcomes and features using the Cox proportional hazards model ()where is the intercept, and is the parameter vector for the bth block, is the (unknown) baseline hazard function at time t, and is the hazard function at time t for the ith subject with covariate vector . We aim to identify which gene groups amongst the groups associated with the survival outcomes.
Based on the partial likelihood function,, we can obtain the estimation of the complete parameter vector by minimizing the following objective function.
Recall that is the tuning parameter that controls the amount of shrinkage (larger shrinks more coefficients to zero), and is used to rescale the penalty to each group. To ensure the top-ranked features are more likely to be selected, we put a small penalty on top-ranked feature sets by proposing using the product of the minimum rank among each feature set and .
The objective of this study is more about selecting the important features than improving the prediction accuracy. Therefore, we propose to use the pseudo-variables assisted tuning strategy () to facilitate the group-lasso tuning parameter selection. This strategy is built on the idea of combining the original and permutated input features (e.g., expressed genes), where the permutations work as a control to determine the significance of each group. Hence, we can select significantly important genes (not by chance).
It is known that the in group-lasso-type regularization controls the amount of shrinkage. As increases, fewer groups are selected. A group can be considered more important one if it is selected when is large. Based on these observations, we can define an importance measure : the coefficient for bth group is nonzero}, for each of the 2B groups, including the B groups from original input features () and their B groups of permutated copies (). For each permutation, groups from original input features are selected if their is larger than , i.e., the strongest signal among permutated groups which we have known are irrelevant groups. After running K (e.g., K = 50) times of permutations, we selected the groups of features that have been selected more than a certain number of percentages (i.e., among K permutations.
Feature selection and machine learning algorithms
In our study, we evaluated nine different feature selection methods, including seven existing feature selection methods and two robust ensemble feature selectors we constructed. The nine selectors can be divided into four major groups: (I) feature selection algorithms based on mutual information optimization: mutual information maximization (MIM) (), minimum redundancy maximum relevance (MRMR) (); and (II) random forest-based approaches: a random forest minimal depth (RF Min Depth) (; ), a random forest variable importance (RF Var Imp) (), a random forest variable hunting (RF Var Hunt) (); and (III) Cox-based approaches: Cox hazard proportional (Cox) () and penalized Cox (Lasso) (); (IV) ensemble learners (). We created two feature ensembles, Ensemble 1 and Ensemble 2, where the first one is the ensemble of Lasso, Cox, and MIM, and the second is the ensemble of Lasso, Cox, MIM, and MRMR. Parameters used in the paper were included in Table 1.
TABLE 1
| Approach | R package | Parameter | Description | Value |
|---|---|---|---|---|
| MIM (select top k) | praznik | k | Select top k features | 25 |
| MRMR (select top k) | ||||
| RF Min Depth (select top k) | randomForestSRC | ntree | Number of trees | 1,000 |
| RF Var Imp (select top k) | mtry | Number of variables to possibly split at each node | default | |
| nodesize | Minimum size of terminal node | 15 | ||
| RF Var Hunt (select top k) | k | Select top k feature | 25 | |
| nsplit | Number of random splits for splitting a variable | 10 | ||
| Cox (select up to top k which have p-value less than ) | survival | k | Select top k feature | 25 |
| alpha | p-value threshold | 0.05 | ||
| LASSO | glmnet | lamba | Tuning parameter grid values | 10(−10,−9.9,…,0,…,9.9,10) |
| Ensemble1 | Minimum pairwise correlation within block | 0.75 | ||
| Ensemble2 | K | Total number of permutations | 50 | |
| Threshold of selection percentage | 0.5 |
Parameters used for feature selection methods.
To compare the results of our feature selection ensemble method with others, we tested the selected features on five well-known prediction models using machine learning and non-parametric techniques: (I) the Cox model with regularization (Lasso) (); (II) models based on boosted trees: xgboost (XGB) () (III) boosted gradient linear models: xgboost based on linear learner (XGB linear) () and (IV) random forest-based methods: random survival forest (RF) () and ranger (). All feature selection methods and machine learning algorithms assessed here can handle the time-to-event outcome.
Simulation
To mimic the correlation structure in real data, we conducted a simulation based on the colorectal cancer data. Considering in the real world, we usually do not often observe the causal variables directly, but rather the variables that are highly correlated with the causal variables, if any. Here we use a modified version of the simulation strategy as in (; ) to mimic this real-world situation. We first picked six correlated gene expression blocks from the colorectal cancer data, where each block included 6,7,8,7,8 and 9 highly correlated genes (correlation coefficient greater than 0.5) respectively (Supplementary Table S1). For each of the first three blocks, we randomly selected one of the genes as the unobserved causal variables (, , and ) which are in the boldface in Supplementary Table S1 and the rest of the genes in the first three blocks as observed causal variables , while considering the genes from the last three blocks as observed noncausal variables. For th block, the variables {} were generated using multivariate normal distribution with mean zero and the correlation matrix computed based on the real data. Then we generated survival outcomes using the three unobserved causal variables based on a Cox proportional hazards model using the reda R package () (simEvent function), with set as 1,
In addition, we generated independent predictor variables , which are uncorrelated with the base variables , are simulated based on a uniform distribution of (0,1). The input features consisted of and . We generated paired replicates (two matrixes) with the first used for feature selection evaluation and the prediction models training, and the second used for assessing stability of feature selection and evaluating the prediction performance, and we repeated the processes for 100 times. The details of this real-data-based simulation, including the coefficients, full list of the gene blocks, and names of the unobserved causal genes, are provided in Supplementary Table S1. For ease of presentation, we will refer this real-data-based simulation as Simulation A below.
To further evaluate the performance of the proposed method under more diverse scenarios, we performed additional simulations (referred as Simulation B below). Similar to Simulation A, we first generate unobserved causal variables (, , and ) and then the observed variables, where some are highly corrected with the causal variables (i.e., observed causal variables), and the rest are irrelevant (i.e., noise variables). The survival outcome is also simulated based on a Cox proportional hazards model using the reda R package () (simEvent function)where is set as 1. The three base variables , , and , the unobserved causal variables) and three additional independent base variables (, and , the unobserved non-causal variables) are independently sampled from a uniform distribution of (0,1). For each of the base variables , we generate a set of 10 correlated predictor variables denoting the th variable in group , for and , using the following formula:
The correlation between the base variable and decreased as increased. Please note that , are only used to simulate correlated variables , and are not included for feature selection and classification. independent predictor variables , which are uncorrelated with the base variables , are also simulated based on a uniform distribution of (0,1). Here we assume that the base variables are not observed. Hence, the input features consist of 30 observed causal variables and 30 correlated, non-causal variables and uncorrelated, non-causal variables , a total of variables.
We consider twelve different simulation scenarios (Table 2) including 1) different event rates ( which are mainly determined by the coefficients in the Cox proportional hazards model; 2) sparsity of causal genes (2.5%, 5%) with a different number of genes (G = 600 and 1,200); and 3) different sample sizes (n = 100 and 200). Similar as in Simulation A, for each of the scenarios, we generated 100 paired replicates, where each pair is consisted of two matrixes.
TABLE 2
| Scenarios | Label | Sample size | # Of genes | Event rate | Sparsity (# of causal genes/# of genes) | |||
|---|---|---|---|---|---|---|---|---|
| 1 | n100_G1200_er0.3 | 100 | 1,200 | 0.3 | 30/1,200 | −6 | 8 | −10 |
| 2 | n100_G1200_er0.5 | 100 | 1,200 | 0.5 | 30/1,200 | −2 | 3 | −4 |
| 3 | n100_G1200_er0.7 | 100 | 1,200 | 0.7 | 30/1,200 | −2 | −3 | 4 |
| 4 | n100_G600_er0.3 | 100 | 600 | 0.3 | 30/600 | −6 | 8 | −10 |
| 5 | n100_G600_er0.5 | 100 | 600 | 0.5 | 30/600 | −2 | 3 | −4 |
| 6 | n100_G600_er0.7 | 100 | 600 | 0.7 | 30/600 | −2 | −3 | 4 |
| 7 | n200_G1200_er0.3 | 200 | 1,200 | 0.3 | 30/1,200 | −6 | 8 | −10 |
| 8 | n200_G1200_er0.5 | 200 | 1,200 | 0.5 | 30/1,200 | −2 | 3 | −4 |
| 9 | n200_G1200_er0.7 | 200 | 1,200 | 0.7 | 30/1,200 | −2 | −3 | 4 |
| 10 | n200_G600_er0.3 | 200 | 600 | 0.3 | 30/600 | −6 | 8 | −10 |
| 11 | n200_G600_er0.5 | 200 | 600 | 0.5 | 30/600 | −2 | 3 | −4 |
| 12 | n200_G600_er0.7 | 200 | 600 | 0.7 | 30/600 | −2 | −3 | 4 |
Simulation scenario.
Model evaluation
In the real data studies, the causal variables are unknown. Moreover, due to different algorithm, we may have different lists of selected features across all methods. Therefore, to determine the important rank of features, we proposed using a weighted relative frequency (WRF) to measure the relative frequency that a feature is selected across five different folds as in (). The weight of each selection is reciprocal to the number of features selected, i.e., larger set of selection adds less weight to each selected feature. A higher WRF indicates this feature is more consistently and sparsely selected across different folds.
Since the causal variables are known in simulation studies, we can evaluate the feature selection performance by comparing the selection to the truth (the known causal variables). Specifically, we used the following four commonly used metrics: false discovery rate (FDR), sensitivity, stability, F-1 score and empirical powers. FDR is the proportion of false-positive features in the selected feature set. Sensitivity is calculated as the proportion of selected causal variables among all the causal variables. Stability is calculated using Jaccard’s index: the ratio of the length of the intersection and the length of the union of two sets, where the two sets are the selections from the paired replicates. F-1 score is calculated as , serving as a balanced metric (harmonic mean) between sensitivity and precision (1-FDR). Empirical power could be calculated for each of the causal variables. It is the ratio that this particular causal variable is selected among the simulation replicates. A power of 1 indicates this casual variable was identified in each replicate, and a power of 0 means it was never selected across replicates. For each feature selection method and each of the twelve scenarios, we reported the average FDR, sensitivity, stability, F-1 score, and empirical powers across the first replicate of each of the 100 simulations.
Furthermore, to check the effectiveness on the predictions of our selected features compared to other well-known models, we used the Integrated Brier score (; ) to assess the accuracy of predicted survival probabilities over a specified time period of events. Lower values of the Integrated Brier Score indicate better predictive accuracy, with 0 being the optimal score (perfect prediction) and 1 representing a model with no predictive ability. Harrell’s C-statistic, also known as the concordance index (C-index), was used to evaluate discrimination with a higher value indicating better discrimination, meaning the model is better at distinguishing between different outcomes. A C-index of 0.5 suggests that the model’s predictions are no better than random chance, while a C-index of 1.0 indicates perfect discrimination.
Results
Key features selected by the ensemble feature selection approach on a colorectal cancer (CRC) dataset
In the cohort of 253 colorectal cancer subjects, encompassing 19,947 genes, the median overall survival (OS) was 83.2 months, with a median follow-up time of 22.5 months. We identified 2,303 genes with p-value less than 0.05 based on the univariable Cox proportional hazard model to further evaluate different feature selection and prediction approaches. We then applied our proposed ensemble approach, where the groups were defined based on the correlation structure of the 2,303 genes, such that within each block, the absolute value of pairwise correlation is all greater than 0.75. The proposed ensemble approaches (Ensemble 1 and Ensemble 2) show the consistency of selected genes and their important rankings compared to all genes, while other methods can only recognize some of them based on WRF (Figure 2A). Notably, the gene SLC30A3, although selected by Lasso with the highest WRF, was not identified by other methods. However, it attained the top rank in our proposed ensemble approach, showing the strength of the ensemble approach. Conversely, several genes (MOS, C1ORF61, and MBL1P) that did not rank highest in Lasso achieved top positions in random forest approaches, contributing to higher WRF in the ensemble approaches. Within the top five genes based on WRF (Ensemble 1 and Ensemble 2), SLC30A3, MOS, C1ORF61, and MBL1P genes were found to have an association with CRC (; ; ; ; ), gene PAGE2, a gene from cancer-germline genes, was found to be upregulated in Caco-2 colorectal cancer cell line (). On the other hand, other methods identified some of the above genes that has connections with colorectal cancers. Using these five genes, we created a composite score by calculating a linear combination of the gene expressions multiplied by their respective coefficients in a multivariable Cox proportional hazards model. Figure 2B presents the Kaplan Meier curves for the subjects with a composite score above and below the median composite score (which is −0.40), with a median OS of 54.6 months and not reachable (log-rank test p-value <0.001), respectively. The DCA (Decision Curve Analysis) curves based on 2 years, 3 years and 5 years (Figures 2C–E) consistently show that the net benefit curve outperforms reference lines across various threshold probabilities, indicating clinical utility. As shown in C-index (Figure 2F) and Brier scores (Figure 2G), in general, the prediction approaches have the most impact on the prediction performance rather than the feature selectors. Lasso has a higher C index, and random forest, XGB, and XBG linear yield the lowest Brier scores, while Ranger demonstrates relatively poorer performance.
FIGURE 2
Improved performance by the ensemble feature selection approach based on simulation studies
Our ensemble approaches consistently demonstrated superior feature selection performance compared to other methods (Supplementary Figure S1A; Figure 3) with Ensemble 1 and Ensemble 2 exhibiting similar performance based on both Simulation A and Simulation B. Although the Lasso method also had low FDRs, it had the lowest sensitivity, reduced F-1 and lower stability. The random forest approaches overall showed poor performance. As expected, in general, a larger sample size (200 vs. 100) resulted in improved performance for all feature selection approaches. However, the impact of the gene sparsity of (2.5% vs. 5%) and event rates (0.3, 0.5, 0.7) on prediction performance was minimal, with slightly better performance observed at lower sparsity. In Supplementary Figure S1B; Figure 4, the empirical power of our ensemble approaches is consistently higher than or at least equivalent to that of other feature selectors across all thirteen scenarios (1 scenario for simulation A, and 12 for simulation B) for all 30 causal variables.
FIGURE 3
FIGURE 4
Similar to the real data analysis, the overall impact on prediction performance is predominantly driven by the choice of prediction approaches rather than the feature selectors due to models’ bias. This observation is expected, as feature selection does not guarantee an improvement in prediction performance. Nevertheless, feature selection proves valuable by reducing the dimensionality and complexity of predictive models, leading to quicker model training times and improved convergence. Predictably, across all prediction approaches, feature selection based on the univariate Cox proportional hazards model consistently exhibited the least favorable performance, while the various selector approaches appeared quite similar. Notably, a higher event rate corresponded to larger Brier scores (Figure 5) and smaller C-index (Figure 6), indicating poorer prediction performance. A larger sample size contributed to slightly improved prediction performance in terms of Brier score and C-index. Interestingly, gene sparsity did not exert a notable impact on prediction performance. While our feature selection models may not have surpassed others in terms of accuracy measurements, we observed that they provided a stable and consistent accuracy across all measurements (as shown in Figure 5, 6; Supplementary Figures S1C, S1D). This suggests that the features we selected are significant and exhibit less bias, contributing to the reliability of our selected features. We also performed Simulation C with smaller effect sizes (Supplementary Table S2) with the same setting as Simulation B. The results (Supplementary Figures S2–S4) were consistent with all the observations mentioned above.
FIGURE 5
FIGURE 6
Conclusion and discussion
This paper proposes a robust ensemble feature selection approach tailored explicitly for survival analysis. The ensemble feature selection approach is built on enhancing the feature selection process by combining different feature selection algorithms, ultimately improving the quality of feature selection and providing stabilized results. This is accomplished through a novel ranking algorithm integrated with a group lasso model, which is particularly advantageous when dealing with feature groups. Therefore, our proposed model is well-suited for applications in genetic data studies, where it is imperative to analyze genes as cohesive groups rather than individual entities. The proposed approach demonstrates a unique ability to select the most compelling features from top-tier models.
The key benefits of ensemble feature selections are 1) Robustness: by aggregating the results from diverse feature selection methods, the final ensemble is less likely to be influenced by the biases or limitations of a single feature selector; 2) Improved Generalization: the ensemble of multiple feature selectors, each built on a different algorithm, can lead to improved generalization and better performance on unseen data; 3) Model Agnosticism: feature selection ensembles are usually model-agnostic, meaning it is not tied to or dependent on a specific machine learning model. Instead, they can be applied across various feature selection models without favoring one over the other, making them widely applicable.
Though we only applied the proposed method to gene expression data, our method can be applied to a wide variety of data having very large number of features in genetics/genomics studies and medical research in general, such as genomic data, transcriptomic data, epigenomic data, proteomic data, clinical and phenotypical data and so on. Besides, the proposed method can smoothly take care of the correlated structure, and even utilize the natural set from certain biological knowledge such as pathway. Moreover, ensemble feature selection can be applied to different response variable, including quantitative, qualitative and time-to-event responses. Although the prediction gain is incremental, the benefits of feature selection are still significant. Firstly, it can enhance the interpretability particularly in the biomedical field and aims the discovery of meaning biological insights. Secondly, it can greatly improve the computational efficiency of downstream analysis, making it more feasible to handle large-scale data sets. Thirdly, it can help filter out irrelevant noise variable, avoid overfitting and enhance the reliability of the analyses.
Statements
Data availability statement
Publicly available datasets were analyzed in this study. This data can be found here: https://portal.gdc.cancer.gov/projects/TCGA-COAD.
Ethics statement
The studies involving humans were approved by the Institutional Review Boards or Ethics Committees of the participating centers. The studies were conducted in accordance with the local legislation and institutional requirements. The participants provided their written informed consent to participate in this study.
Author contributions
PL: Formal Analysis, Investigation, Software, Visualization, Writing–original draft, Writing–review and editing. XG: Formal Analysis, Investigation, Software, Visualization, Writing–review and editing. LU: Formal Analysis, Investigation, Software, Visualization, Writing–review and editing. HY: Data curation, Investigation, Writing–review and editing. BK: Data curation, Investigation, Writing–review and editing. LZ: Conceptualization, Data curation, Formal Analysis, Funding acquisition, Investigation, Methodology, Project administration, Resources, Software, Supervision, Visualization, Writing–original draft, Writing–review and editing. TH: Conceptualization, Formal Analysis, Funding acquisition, Investigation, Methodology, Project administration, Resources, Software, Supervision, Validation, Visualization, Writing–original draft, Writing–review and editing.
Funding
The author(s) declare that financial support was received for the research, authorship, and/or publication of this article. PL, XG, LU, YH, BK, LZ, and TH are partially supported by National Cancer Institute, National Institutes of Health, through grant number R21CA264381. PL, LU, YH, BK, LZ, and TH are partially supported by the National Library of Medicine, National Institutes of Health, through grant number R01LM013763-01A1. TH and LU are partially supported by National Science Foundation through grant number DMS-2137983. BK is supported by Helen Diller Family Comprehensive Cancer Center Cancer Physician Scholar Program in Clinical Oncology (PSPCO) Career Development Award (1K12CA260225-01).
Conflict of interest
The authors declare that the research was conducted in the absence of any commercial or financial relationships that could be construed as a potential conflict of interest.
The author(s) declared that they were an editorial board member of Frontiers, at the time of submission. This had no impact on the peer review process and the final decision.
Publisher’s note
All claims expressed in this article are solely those of the authors and do not necessarily represent those of their affiliated organizations, or those of the publisher, the editors and the reviewers. Any product that may be evaluated in this article, or claim that may be made by its manufacturer, is not guaranteed or endorsed by the publisher.
Supplementary material
The Supplementary Material for this article can be found online at: https://www.frontiersin.org/articles/10.3389/fsysb.2024.1355595/full#supplementary-material
References
1
ArcherK. J.KimesR. V. (2008). Empirical characterization of random forest variable importance measures. Comput. Statistics Data Analysis52 (4), 2249–2260. 10.1016/j.csda.2007.08.015
2
BinderH. (2015). R package “CoxBoost”. Available from: https://github.com/binderh/CoxBoost.
3
CeramiE.GaoJ.DogrusozU.GrossB. E.SumerS. O.AksoyB. A.et al (2012). The cBio cancer genomics portal: an open platform for exploring multidimensional cancer genomics data. Cancer Discov.2 (5), 401–404. 10.1158/2159-8290.CD-12-0095
4
ChenT.GuestrinC. (2016). “XGBoost: a scalable tree boosting system,” in Proceedings of the 22nd ACM SIGKDD International Conference on Knowledge Discovery and Data Mining, San Francisco, CA, USA, August 13-17, 2016 (San Francisco California USA: ACM), 785–794. 10.1145/2939672.2939785
5
ChenX.IshwaranH. (2013). Pathway hunting by random survival forests. Bioinformatics29 (1), 99–105. 10.1093/bioinformatics/bts643
6
CuiJ.GuoF.YuY.MaZ.HongY.SuJ.et al (2022). Development and validation of a prognostic 9-gene signature for colorectal cancer. Front. Oncol.12, 1009698. 10.3389/fonc.2022.1009698
7
DegenhardtF.SeifertS.SzymczakS. (2019). Evaluation of variable selection methods for random forests and omics data sets. Briefings Bioinforma.20 (2), 492–503. 10.1093/bib/bbx124
8
DeoS. V.DeoV.SundaramV. (2021). Survival analysis—part 2: cox proportional hazards model. Indian J. Thorac. Cardiovasc Surg.37 (2), 229–233. 10.1007/s12055-020-01108-7
9
FriedmanJ.HastieT.TibshiraniR. (2010). Regularization paths for generalized linear models via coordinate descent. J. Stat. Softw.33 (1), 1–22. 10.18637/jss.v033.i01
10
FuH.HanS.YanJ. (2022). Reda R package. Available from: https://github.com/wenjie2wang/reda.
11
GoemanJ. J. (2010). L1 penalized estimation in the cox proportional hazards model. Biom. J.52 (1), 70–84. 10.1002/bimj.200900028
12
HeT.BaikJ. M.KatoC.YangH.FanZ.ChamJ.et al (2022). Novel ensemble feature selection approach and application in repertoire sequencing data. Front. Genet.13, 821832. 821832. 10.3389/fgene.2022.821832
13
HuT.ChitnisN.MonosD.DinhA. (2021). Next-generation sequencing technologies: an overview. Hum. Immunol.82 (11), 801–811. 10.1016/j.humimm.2021.02.012
14
IshwaranH.KogalurU. B.BlackstoneE. H.LauerM. S. (2008). Random survival forests. Ann. Appl. Stat.2 (3). 10.1214/08-aoas169
15
IshwaranH.KogalurU. B.ChenX.MinnA. J. (2011). Random survival forests for high‐dimensional data. Stat. Anal.4 (1), 115–132. 10.1002/sam.10103
16
KleinJ. P.GoelP. K. (1992). “Survival analysis: state of the art,” in NATO ASI series. Series E, Applied sciences Editors KleinJ. P.GoelP. K. (Dordrecht ; Boston: Kluwer Academic Publishers), 451.
17
LauJ. W.LehnertE.SethiA.MalhotraR.KaushikG.OnderZ.et al (2017). The cancer genomics Cloud: collaborative, reproducible, and democratized—a new paradigm in large-scale computational research. Cancer Res.77 (21), e3–e6. 10.1158/0008-5472.CAN-17-0387
18
LiJ.ChengK.WangS.MorstatterF.TrevinoR. P.TangJ.et al (2018). Feature selection: a data perspective. ACM Comput. Surv.50 (6), 1–45. 10.1145/3136625
19
LinH. M.ChatterjeeA.LinY. H.AnjomshoaaA.FukuzawaR.McCallJ.et al (2007). Genome wide expression profiling identifies genes associated with colorectal liver metastasis. Oncol. Rep.17, 1541–1549. 10.3892/or.17.6.1541
20
MoradianH.LarocqueD.BellavanceF. (2017). L₁ splitting rules in survival forests. Lifetime Data Anal.23 (4), 671–691. 10.1007/s10985-016-9372-1
21
MunsonM. A.CaruanaR. (2009). “On feature selection, bias-variance, and bagging,” in Machine learning and knowledge discovery in databases Editors BuntineW.GrobelnikM.MladenićD.Shawe-TaylorJ. (Berlin, Heidelberg: Springer Berlin Heidelberg), 144–159. (Lecture Notes in Computer Science; vol. 5782). Available from: http://link.springer.com/10.1007/978-3-642-04174-7_10.
22
NeumsL.MeierR.KoestlerD. C.ThompsonJ. A. (2019). “Improving survival prediction using a novel feature selection and feature reduction framework based on the integration of clinical and molecular data,” in Biocomputing 2020 (Kohala Coast, Hawaii, USA: World Scientific), 415–426. 10.1142/9789811215636_0037
23
PengJ.PengJ.WangR.LiuC.WangZ. (2021). Expression of MOS gene and its correlations with clinicopathological features and prognosis of patients with colorectal cancer. Chin. General Pract.24 (24), 3077. 10.12114/j.issn.1007-9572.2021.00.434
24
RadovicM.GhalwashM.FilipovicN.ObradovicZ. (2017). Minimum redundancy maximum relevance feature selection approach for temporal gene expression data. BMC Bioinforma.18 (1), 9. 10.1186/s12859-016-1423-9
25
SegalM. (2004). Machine learning benchmarks and random forest regression. UCSF: Center for Bioinformatics and Molecular Biostatistics. Available at: https://escholarship.org/uc/item/35x3v9t4
26
The Cancer Genome Atlas Network (2012). Comprehensive molecular characterization of human colon and rectal cancer. Nature487 (7407), 330–337. 10.1038/nature11252
27
TorkkolaK. (2003). Feature extraction by non-parametric mutual information maximization. J. Mach. Learn. Res.3, 1415–1438. Mar.
28
UtazirubandaJ. C.León TM.NgomP. (2021). Variable selection with group LASSO approach: application to Cox regression with frailty model. Commun. Statistics - Simul. Comput.50 (3), 881–901. 10.1080/03610918.2019.1571605
29
WrightM. N.rangerZ. A. (2017). A fast implementation of random forests for high dimensional data in C++ and R. J. Stat. Soft77 (1). 10.18637/jss.v077.i01
30
YangQ.KhouryM. J.FriedmanJ.LittleJ.FlandersW. D. (2005). How many genes underlie the occurrence of common complex diseases in the population?Int. J. Epidemiol.34 (5), 1129–1137. 10.1093/ije/dyi130
31
Yilmaz-OzcanS.SadeA.KucukkaradumanB.KaygusuzY.SensesK. M.BanerjeeS.et al (2014). Epigenetic mechanisms underlying the dynamic expression of cancer-testis genes, PAGE2, -2B and SPANX-B, during mesenchymal-to-epithelial transition. PLoS ONE9(9), e107905. 10.1371/journal.pone.0107905
32
YinZ.YanX.WangQ.DengZ.TangK.CaoZ.et al (2020). Detecting prognosis risk biomarkers for colon cancer through multi-omics-based prognostic analysis and target regulation simulation modeling. Front. Genet.11, 524. 10.3389/fgene.2020.00524
33
ZhengY.ZhouJ.TongY. (2015). Gene signatures of drug resistance predict patient survival in colorectal cancer. Pharmacogenomics J.15 (2), 135–143. 10.1038/tpj.2014.45
34
ZhouZ. H. (2021). Machine learning. Singapore: Springer, 458.
Summary
Keywords
colorectal cancer, ensemble feature selection, high-dimensional data, time-to-event outcome, pseudo variables, group lasso
Citation
Le P, Gong X, Ung L, Yang H, Keenan BP, Zhang L and He T (2024) A robust ensemble feature selection approach to prioritize genes associated with survival outcome in high-dimensional gene expression data. Front. Syst. Biol. 4:1355595. doi: 10.3389/fsysb.2024.1355595
Received
14 December 2023
Accepted
05 March 2024
Published
21 March 2024
Volume
4 - 2024
Edited by
Rongling Wu, The Pennsylvania State University (PSU), United States
Reviewed by
Xiaoxi Shen, Texas State University, United States
Luis Diambra, National University of La Plata, Argentina
Updates
Copyright
© 2024 Le, Gong, Ung, Yang, Keenan, Zhang and He.
This is an open-access article distributed under the terms of the Creative Commons Attribution License (CC BY). The use, distribution or reproduction in other forums is permitted, provided the original author(s) and the copyright owner(s) are credited and that the original publication in this journal is cited, in accordance with accepted academic practice. No use, distribution or reproduction is permitted which does not comply with these terms.
*Correspondence: Li Zhang, li.zhang@ucsf.edu; Tao He, hetao@sfsu.edu
† These authors share senior authorship
Disclaimer
All claims expressed in this article are solely those of the authors and do not necessarily represent those of their affiliated organizations, or those of the publisher, the editors and the reviewers. Any product that may be evaluated in this article or claim that may be made by its manufacturer is not guaranteed or endorsed by the publisher.