arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 1906.04596cs.LGstat.ML
收藏

ANODEV2:一个耦合的神经ODE进化框架

ANODEV2: A Coupled Neural ODE Evolution Framework

University of California at Berkeley(加州大学伯克利分校);University of Texas at Austin(德克萨斯大学奥斯汀分校);ICSI(国际计算机科学研究所)

Tianjun Zhang, Zhewei Yao, Amir Gholami, Kurt Keutzer, Joseph Gonzalez, George Biros, Michael Mahoney

摘要:

已经观察到,残差网络可以被视为常微分方程(ODE)的显式欧拉离散化。这一观察激发了所谓的神经常微分方程的引入,它允许更一般的离散化方案与自适应时间步进。在这里,我们提出了ANODEV 2,这是一个扩展的这种方法,也允许演变的神经网络参数,在一个耦合的ODE为基础的配方。前面介绍的神经ODE方法实际上是这个新的更一般框架的特例。我们提出了ANODEV 2的公式,推导出最优性条件,并在PyTorch中实现了该框架的耦合反应-扩散-平流版本。我们使用几种不同配置的ANODEV 2,在CIFAR-10上的多个模型上测试它们,给出了实证结果。我们报告的结果表明,这种耦合的基于ODE的框架确实是可训练的,并且与基线模型以及最近提出的神经ODE方法相比,它实现了更高的准确性。

英文摘要:

It has been observed that residual networks can be viewed as the explicit Euler discretization of an Ordinary Differential Equation (ODE). This observation motivated the introduction of so-called Neural ODEs, which allow more general discretization schemes with adaptive time stepping. Here, we propose ANODEV2, which is an extension of this approach that also allows evolution of the neural network parameters, in a coupled ODE-based formulation. The Neural ODE method introduced earlier is in fact a special case of this new more general framework. We present the formulation of ANODEV2, derive optimality conditions, and implement a coupled reaction-diffusion-advection version of this framework in PyTorch. We present empirical results using several different configurations of ANODEV2, testing them on multiple models on CIFAR-10. We report results showing that this coupled ODE-based framework is indeed trainable, and that it achieves higher accuracy, as compared to the baseline models as well as the recently-proposed Neural ODE approach.

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

↑