Variational Bayes for high-dimensional proportional hazards models with applications within gene expression
File(s)btac416.pdf (432.95 KB)
Published version
Author(s)
Komodromos, Michael
Aboagye, Eric O
Evangelou, Marina
Filippi, Sarah
Ray, Kolyan
Type
Journal Article
Abstract
Motivation:
Few Bayesian methods for analyzing high-dimensional sparse survival data provide scalable variable selection, effect estimation and uncertainty quantification. Such methods often either sacrifice uncertainty quantification by computing maximum a posteriori estimates, or quantify the uncertainty at high (unscalable) computational expense.
Results:
We bridge this gap and develop an interpretable and scalable Bayesian proportional hazards model for prediction and variable selection, referred to as SVB. Our method, based on a mean-field variational approximation, overcomes the high computational cost of MCMC whilst retaining useful features, providing a posterior distribution for the parameters and offering a natural mechanism for variable selection via posterior inclusion probabilities. The performance of our proposed method is assessed via extensive simulations and compared against other state-of-the-art Bayesian variable selection methods, demonstrating comparable or better performance. Finally, we demonstrate how the proposed method can be used for variable selection on two transcriptomic datasets with censored survival outcomes, and how the uncertainty quantification offered by our method can be used to provide an interpretable assessment of patient risk.
Availability and implementation:
our method has been implemented as a freely available R package survival.svb (https://github.com/mkomod/survival.svb).
Few Bayesian methods for analyzing high-dimensional sparse survival data provide scalable variable selection, effect estimation and uncertainty quantification. Such methods often either sacrifice uncertainty quantification by computing maximum a posteriori estimates, or quantify the uncertainty at high (unscalable) computational expense.
Results:
We bridge this gap and develop an interpretable and scalable Bayesian proportional hazards model for prediction and variable selection, referred to as SVB. Our method, based on a mean-field variational approximation, overcomes the high computational cost of MCMC whilst retaining useful features, providing a posterior distribution for the parameters and offering a natural mechanism for variable selection via posterior inclusion probabilities. The performance of our proposed method is assessed via extensive simulations and compared against other state-of-the-art Bayesian variable selection methods, demonstrating comparable or better performance. Finally, we demonstrate how the proposed method can be used for variable selection on two transcriptomic datasets with censored survival outcomes, and how the uncertainty quantification offered by our method can be used to provide an interpretable assessment of patient risk.
Availability and implementation:
our method has been implemented as a freely available R package survival.svb (https://github.com/mkomod/survival.svb).
Date Issued
2022-06-25
Date Acceptance
2022-06-23
Citation
BIOINFORMATICS, 2022, 38 (16), pp.3918-3926
ISSN
1367-4803
Publisher
OXFORD UNIV PRESS
Start Page
3918
End Page
3926
Journal / Book Title
BIOINFORMATICS
Volume
38
Issue
16
Copyright Statement
© The Author(s) 2022. Published by Oxford University Press.
This is an Open Access article distributed under the terms of the Creative Commons Attribution License
(http://creativecommons.org/licenses/by/4.0/), which permits unrestricted reuse, distribution, and reproduction in any medium,
provided the original work is properly cited.
This is an Open Access article distributed under the terms of the Creative Commons Attribution License
(http://creativecommons.org/licenses/by/4.0/), which permits unrestricted reuse, distribution, and reproduction in any medium,
provided the original work is properly cited.
License URL
Sponsor
Cancer Research UK
Imperial College Healthcare NHS Trust- BRC Funding
Identifier
https://www.webofscience.com/api/gateway?GWVersion=2&SrcApp=PARTNER_APP&SrcAuth=LinksAMR&KeyUT=WOS:000821982200001&DestLinkType=FullRecord&DestApp=ALL_WOS&UsrCustomerID=a2bf6146997ec60c407a63945d4e92bb
Grant Number
25147
RDC04
Subjects
Science & Technology
Life Sciences & Biomedicine
Technology
Physical Sciences
Biochemical Research Methods
Biotechnology & Applied Microbiology
Computer Science, Interdisciplinary Applications
Mathematical & Computational Biology
Statistics & Probability
Biochemistry & Molecular Biology
Computer Science
Mathematics
VARIABLE SELECTION
SURVIVAL ANALYSIS
INFERENCE
REGRESSION
LASSO
REGULARIZATION
HORSESHOE
Publication Status
Published
Date Publish Online
2022-06-25