Hierarchical Ridge Regression for Incorporating Prior Information in Genomic Studies

Eric S. Kawaguchi ； Sisi Li ； Garrett M. Weaver ； Juan Pablo Lewinger

《Journal of Data Science》 20卷1期 (2022/01) Pp. 34-50

https://doi.org/10.6339/21-JDS1030

摘要

There is a great deal of prior knowledge about gene function and regulation in the form of annotations or prior results that, if directly integrated into individual prognostic or diagnostic studies, could improve predictive performance. For example, in a study to develop a predictive model for cancer survival based on gene expression, effect sizes from previous studies or the grouping of genes based on pathways constitute such prior knowledge. However, this external information is typically only used post-analysis to aid in the interpretation of any findings. We propose a new hierarchical two-level ridge regression model that can integrate external information in the form of ＂meta features＂ to predict an outcome. We show that the model can be fit efficiently using cyclic coordinate descent by recasting the problem as a single-level regression model. In a simulation-based evaluation we show that the proposed method outperforms standard ridge regression and competing methods that integrate prior information, in terms of prediction performance when the meta features are informative on the mean of the features, and that there is no loss in performance when the meta features are uninformative. We demonstrate our approach with applications to the prediction of chronological age based on methylation features and breast cancer mortality based on gene expression features.

關鍵字

high-dimensional regression ； meta-features ； penalization ； prediction ； regularization