arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2103.15419math.NAcs.LGcs.NA

Translating Numerical Concepts for PDEs into Neural Architectures

Saarland University(萨尔兰大学)

Tobias Alt, Pascal Peter, Joachim Weickert, Karl Schrader

英文摘要:

We investigate what can be learned from translating numerical algorithms into neural networks. On the numerical side, we consider explicit, accelerated explicit, and implicit schemes for a general higher order nonlinear diffusion equation in 1D, as well as linear multigrid methods. On the neural network side, we identify corresponding concepts in terms of residual networks (ResNets), recurrent networks, and U-nets. These connections guarantee Euclidean stability of specific ResNets with a transposed convolution layer structure in each block. We present three numerical justifications for skip connections: as time discretisations in explicit schemes, as extrapolation mechanisms for accelerating those methods, and as recurrent connections in fixed point solvers for implicit schemes. Last but not least, we also motivate uncommon design choices such as nonmonotone activation functions. Our findings give a numerical perspective on the success of modern neural network architectures, and they provide design criteria for stable networks.

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

↑