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

一致稀疏深度学习:理论与计算

Consistent Sparse Deep Learning: Theory and Computation

Purdue University(普渡大学)

Yan Sun, Qifan Song, Faming Liang

摘要:

深度学习一直是推动数据科学取得许多成功的引擎。然而,深度神经网络(DNN)作为深度学习的基础模型,往往过度参数化,造成训练、预测和解释的诸多困难。我们提出了一种学习稀疏DNN的类频率论方法,并在贝叶斯框架下证明了其一致性:所提出的方法可以学习最多$O(n/\log(n))$连接的稀疏DNN,并具有良好的理论保证,如后验一致性,变量选择一致性和渐近最优泛化界。特别是,我们建立后验一致性的稀疏DNN与混合高斯先验,表明稀疏DNN的结构可以一致地确定使用基于拉普拉斯近似的边缘后验包含概率的方法,并使用贝叶斯证据,以引起稀疏DNN的学习优化方法,如随机梯度下降在多个运行不同的初始化。对于大规模稀疏DNN,所提出的方法在计算上比标准贝叶斯方法更有效。数值结果表明,该方法在大规模网络压缩和高维非线性变量选择方面都有很好的表现,都推进了可解释的机器学习.

英文摘要:

Deep learning has been the engine powering many successes of data science. However, the deep neural network (DNN), as the basic model of deep learning, is often excessively over-parameterized, causing many difficulties in training, prediction and interpretation. We propose a frequentist-like method for learning sparse DNNs and justify its consistency under the Bayesian framework: the proposed method could learn a sparse DNN with at most $O(n/\log(n))$ connections and nice theoretical guarantees such as posterior consistency, variable selection consistency and asymptotically optimal generalization bounds. In particular, we establish posterior consistency for the sparse DNN with a mixture Gaussian prior, show that the structure of the sparse DNN can be consistently determined using a Laplace approximation-based marginal posterior inclusion probability approach, and use Bayesian evidence to elicit sparse DNNs learned by an optimization method such as stochastic gradient descent in multiple runs with different initializations. The proposed method is computationally more efficient than standard Bayesian methods for large-scale sparse DNNs. The numerical results indicate that the proposed method can perform very well for large-scale network compression and high-dimensional nonlinear variable selection, both advancing interpretable machine learning.

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

↑