arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2103.00301cs.LG
收藏

用于深度学习的神经网络控制的样条参数化

Spline parameterization of neural network controls for deep learning

Lawrence Livermore National Laboratory(劳伦斯利弗莫尔国家实验室);Cornell University(康奈尔大学)

Stefanie Günther, Will Pazner, Dongping Qi

摘要:

基于将深度学习转换作为最优控制问题的连续解释,本文研究了采用B样条基函数对各层神经网络控制进行参数化的好处。我们选择固定数量的B样条基函数,其系数是神经网络的可训练参数,而不是为离散ODE网络的每一层配备一组可训练的权重。将可训练参数从神经网络的层中解耦,使我们能够研究和适应与优化学习问题分离的网络传播的准确性。我们数值计算表明,基于样条的神经网络增加了鲁棒性的学习问题对超参数由于增加的稳定性和准确性的网络传播。此外,直接对B样条系数而不是层权重进行训练可以减少可训练参数的数量。

英文摘要:

Based on the continuous interpretation of deep learning cast as an optimal control problem, this paper investigates the benefits of employing B-spline basis functions to parameterize neural network controls across the layers. Rather than equipping each layer of a discretized ODE-network with a set of trainable weights, we choose a fixed number of B-spline basis functions whose coefficients are the trainable parameters of the neural network. Decoupling the trainable parameters from the layers of the neural network enables us to investigate and adapt the accuracy of the network propagation separated from the optimization learning problem. We numerically show that the spline-based neural network increases robustness of the learning problem towards hyperparameters due to increased stability and accuracy of the network propagation. Further, training on B-spline coefficients rather than layer weights directly enables a reduction in the number of trainable parameters.

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

↑