NEURAL DAES: CONSTRAINED NEURAL NETWORKS

被引:0
作者
Boesen, Tue [1 ]
Haber, Eldad [1 ]
Ascher, Uri M. [2 ]
机构
[1] Univ British Columbia, Dept Earth Ocean & Atmospher Sci, Vancouver, BC V6T 1Z4, Canada
[2] Univ British Columbia, Dept Comp Sci, Vancouver, BC V6T 1Z4, Canada
关键词
constraints; projection; neural network; differential algebraic equation; auxiliary trajectory information; neural odes; PHYSICS;
D O I
10.1137/23M1574051
中图分类号
O29 [应用数学];
学科分类号
070104 ;
摘要
This article investigates the effect of explicitly adding auxiliary algebraic trajectory information to neural networks for dynamical systems. We draw inspiration from the field of differential algebraic equations and differential equations on manifolds and implement related methods in residual neural networks, despite some fundamental scenario differences. Constraint or auxiliary information effects are incorporated through stabilization as well as projection methods, and we show when to use which method based on experiments involving simulations of multibody pendulums and molecular dynamics scenarios. Several of our methods are easy to implement in existing code and have limited impact on training performance while giving significant boosts in terms of inference.
引用
收藏
页码:C291 / C312
页数:22
相关论文
共 34 条
[1]  
ALLEN MP, 2004, NIC SERIES, V23, P1, DOI DOI 10.1021/MA00095A001
[2]  
Arnold M, 2017, SURVEYS IN DIFFERENTIAL-ALGEBRAIC EQUATIONS IV, P41, DOI 10.1007/978-3-319-46618-7_2
[3]  
Ascher U. M., 1998, Computer methods for ordinary differential equations and differential-algebraic equations, DOI 10.1137/1.9781611971392
[4]   Stabilization of invariants of discretized differential systems [J].
Ascher, UM .
NUMERICAL ALGORITHMS, 1997, 14 (1-3) :1-24
[5]   E(3)-equivariant graph neural networks for data-efficient and accurate interatomic potentials [J].
Batzner, Simon ;
Musaelian, Albert ;
Sun, Lixin ;
Geiger, Mario ;
Mailoa, Jonathan P. ;
Kornbluth, Mordechai ;
Molinari, Nicola ;
Smidt, Tess E. ;
Kozinsky, Boris .
NATURE COMMUNICATIONS, 2022, 13 (01)
[6]   IMAGE DENOISING: LEARNING THE NOISE MODEL VIA NONSMOOTH PDE-CONSTRAINED OPTIMIZATION [J].
Carlos De los Reyes, Juan ;
Schoenlieb, Carola-Bibiane .
INVERSE PROBLEMS AND IMAGING, 2013, 7 (04) :1183-1214
[7]  
Chen R. T. Q., 2019, arXiv, DOI [10.48550/arXiv.1806.07366, DOI 10.48550/ARXIV.1806.07366]
[8]   Coupling Molecular Dynamics and Deep Learning to Mine Protein Conformational Space [J].
Degiacomi, Matteo T. .
STRUCTURE, 2019, 27 (06) :1034-+
[9]   Mimetic Neural Networks: A Unified Framework for Protein Design and Folding [J].
Eliasof, Moshe ;
Boesen, Tue ;
Haber, Eldad ;
Keasar, Chen ;
Treister, Eran .
FRONTIERS IN BIOINFORMATICS, 2022, 2
[10]   Brief review of image denoising techniques [J].
Fan, Linwei ;
Zhang, Fan ;
Fan, Hui ;
Zhang, Caiming .
VISUAL COMPUTING FOR INDUSTRY BIOMEDICINE AND ART, 2019, 2 (01)