Training Hamiltonian neural networks without backpropagation

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: Rahma, Atamert, Datar, Chinmay, Dietrich, Felix
Format: Preprint
Publié: 2024
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866910130418745344
author Rahma, Atamert
Datar, Chinmay
Dietrich, Felix
author_facet Rahma, Atamert
Datar, Chinmay
Dietrich, Felix
contents Neural networks that synergistically integrate data and physical laws offer great promise in modeling dynamical systems. However, iterative gradient-based optimization of network parameters is often computationally expensive and suffers from slow convergence. In this work, we present a backpropagation-free algorithm to accelerate the training of neural networks for approximating Hamiltonian systems through data-agnostic and data-driven algorithms. We empirically show that data-driven sampling of the network parameters outperforms data-agnostic sampling or the traditional gradient-based iterative optimization of the network parameters when approximating functions with steep gradients or wide input domains. We demonstrate that our approach is more than 100 times faster with CPUs than the traditionally trained Hamiltonian Neural Networks using gradient-based iterative optimization and is more than four orders of magnitude accurate in chaotic examples, including the Hénon-Heiles system.
format Preprint
id arxiv_https___arxiv_org_abs_2411_17511
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Training Hamiltonian neural networks without backpropagation
Rahma, Atamert
Datar, Chinmay
Dietrich, Felix
Machine Learning
Numerical Analysis
Neural networks that synergistically integrate data and physical laws offer great promise in modeling dynamical systems. However, iterative gradient-based optimization of network parameters is often computationally expensive and suffers from slow convergence. In this work, we present a backpropagation-free algorithm to accelerate the training of neural networks for approximating Hamiltonian systems through data-agnostic and data-driven algorithms. We empirically show that data-driven sampling of the network parameters outperforms data-agnostic sampling or the traditional gradient-based iterative optimization of the network parameters when approximating functions with steep gradients or wide input domains. We demonstrate that our approach is more than 100 times faster with CPUs than the traditionally trained Hamiltonian Neural Networks using gradient-based iterative optimization and is more than four orders of magnitude accurate in chaotic examples, including the Hénon-Heiles system.
title Training Hamiltonian neural networks without backpropagation
topic Machine Learning
Numerical Analysis
url https://arxiv.org/abs/2411.17511