arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2002.10243stat.MLcs.LG

Informative Bayesian Neural Network Priors for Weak Signals

Aalto University(阿尔托大学);Helsinki Institute for Information Technology HIIT(赫尔辛基信息技术研究所(HIIT));Finnish Institute for Health and Welfare(芬兰健康与福利研究所);Institute for Molecular Medicine Finland(芬兰分子医学研究所);University of Manchester(曼彻斯特大学)

Tianyu Cui, Aki Havulinna, Pekka Marttinen, Samuel Kaski

英文摘要:

Encoding domain knowledge into the prior over the high-dimensional weight space of a neural network is challenging but essential in applications with limited data and weak signals. Two types of domain knowledge are commonly available in scientific applications: 1. feature sparsity (fraction of features deemed relevant); 2. signal-to-noise ratio, quantified, for instance, as the proportion of variance explained (PVE). We show how to encode both types of domain knowledge into the widely used Gaussian scale mixture priors with Automatic Relevance Determination. Specifically, we propose a new joint prior over the local (i.e., feature-specific) scale parameters that encodes knowledge about feature sparsity, and a Stein gradient optimization to tune the hyperparameters in such a way that the distribution induced on the model's PVE matches the prior distribution. We show empirically that the new prior improves prediction accuracy, compared to existing neural network priors, on several publicly available datasets and in a genetics application where signals are weak and sparse, often outperforming even computationally intensive cross-validation for hyperparameter tuning.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑