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

Guiding Neural Network Initialization via Marginal Likelihood Maximization

Indiana University(印第安纳大学)

Anthony S. Tai, Chunfeng Huang

英文摘要:

We propose a simple, data-driven approach to help guide hyperparameter selection for neural network initialization. We leverage the relationship between neural network and Gaussian process models having corresponding activation and covariance functions to infer the hyperparameter values desirable for model initialization. Our experiment shows that marginal likelihood maximization provides recommendations that yield near-optimal prediction performance on MNIST classification task under experiment constraints. Furthermore, our empirical results indicate consistency in the proposed technique, suggesting that computation cost for the procedure could be significantly reduced with smaller training sets.

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

↑