arXiv · 2111.00565
Can we learn gradients by Hamiltonian Neural Networks?
Abstract
In this work, we propose a meta-learner based on ODE neural networks that learns gradients. This approach makes the optimizer is more flexible inducing an automatic inductive bias to the given task. Using the simplest Hamiltonian Neural Network we demonstrate that our method outperforms a meta-learner based on LSTM for an artificial task and the MNIST dataset with ReLU activations in the optimizee. Furthermore, it also surpasses the classic optimization methods for the artificial task and achieves comparable results for MNIST.
Explore related subjects
Keep this discovery
Aleksandr Timofeev, Andrei Afonin, Yehao Liu. 2021-10-31. Can we learn gradients by Hamiltonian Neural Networks?. https://arxiv.org/abs/2111.00565
Cite the original work for its findings. Save a collection to share your selection of sources.