A physics-inspired nonlinear momentum method for gradient descent with applications to inverse photonic design

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Zhang, Jianing, Liu, Rumei
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866915779517087744
author Zhang, Jianing
Liu, Rumei
author_facet Zhang, Jianing
Liu, Rumei
contents In this work, a nonlinear momentum method is introduced to enhance the convergence performance of momentum-based gradient optimization algorithms. Classical momentum methods, such as the Heavy Ball method, can be viewed as a dynamical system with quadratic kinetic energy and linear damping. By extending this analogy to non-Newtonian dynamical systems, we construct a Hamiltonian framework for optimization problems. In this framework, nonlinear kinetic energy and nonlinear damping effects naturally emerge. It provides a more flexible and physically interpretable mechanism for optimization algorithms. Specifically, we employ an anharmonic kinetic energy function to capture the inertial effects of accumulated gradient information during the optimization process, while the nonlinear damping mechanism effectively regulates the contribution of momentum during convergence. Numerical experiments show that the proposed method achieves faster convergence compared to classical momentum algorithms, making it particularly suitable for inverse design tasks. Moreover, the Hamiltonian based algorithmic framework may offer physical insights for the development of efficient physics-inspired optimization algorithms.
format Preprint
id arxiv_https___arxiv_org_abs_2511_16441
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle A physics-inspired nonlinear momentum method for gradient descent with applications to inverse photonic design
Zhang, Jianing
Liu, Rumei
Computational Physics
Optics
In this work, a nonlinear momentum method is introduced to enhance the convergence performance of momentum-based gradient optimization algorithms. Classical momentum methods, such as the Heavy Ball method, can be viewed as a dynamical system with quadratic kinetic energy and linear damping. By extending this analogy to non-Newtonian dynamical systems, we construct a Hamiltonian framework for optimization problems. In this framework, nonlinear kinetic energy and nonlinear damping effects naturally emerge. It provides a more flexible and physically interpretable mechanism for optimization algorithms. Specifically, we employ an anharmonic kinetic energy function to capture the inertial effects of accumulated gradient information during the optimization process, while the nonlinear damping mechanism effectively regulates the contribution of momentum during convergence. Numerical experiments show that the proposed method achieves faster convergence compared to classical momentum algorithms, making it particularly suitable for inverse design tasks. Moreover, the Hamiltonian based algorithmic framework may offer physical insights for the development of efficient physics-inspired optimization algorithms.
title A physics-inspired nonlinear momentum method for gradient descent with applications to inverse photonic design
topic Computational Physics
Optics
url https://arxiv.org/abs/2511.16441