Higher-Order Geometric Updates for Levenberg-Marquardt Method via Riemann Normal Coordinates
Summary
A new optimization method, Riemann-normal-coordinate Levenberg-Marquardt (RNC-LM), is proposed to enhance nonlinear least-squares optimization. This method addresses a limitation of the standard Levenberg-Marquardt (LM) approach, which applies tangent-space steps as straight updates in parameter coordinates, leading to inconsistencies due to parameter-effects curvature. RNC-LM extends geodesic acceleration by reformulating the geodesic equation, enabling arbitrary-order corrections and constructing finite-step updates with higher reparameterization consistency. It employs a line search along the RNC curve to control distance while maintaining cost close to standard LM. The technique eliminates the tangential component of residual acceleration order by order, making objective reduction more consistent with LM's linear model. RNC-LM demonstrates improved convergence and robustness on classical benchmarks, reduces relative L2 error to 1e-3 on a reaction-diffusion PINN failure-mode benchmark, and achieves a 34-fold speedup on a large-scale machine-learning potential-energy-surface fitting task.
Key takeaway
For research scientists optimizing nonlinear least-squares problems, particularly in physics-informed neural networks or large-scale machine learning, you should consider RNC-LM. This method offers significantly improved convergence and robustness over standard Levenberg-Marquardt, reducing L2 error to 1e-3 and achieving a 34-fold speedup in specific tasks. Evaluate RNC-LM to overcome parameter-effects curvature limitations and achieve more physically meaningful solutions.
Key insights
RNC-LM improves nonlinear least-squares optimization by using higher-order geometric updates for enhanced reparameterization consistency and performance.
Principles
- Parameter-effects curvature limits standard LM.
- Higher-order geometric updates improve optimization consistency.
- Riemann-normal-coordinates extend geodesic acceleration.
Method
RNC-LM reformulates the geodesic equation to extend geodesic acceleration, constructing finite-step updates with higher reparameterization consistency, and uses a line search along the RNC curve.
In practice
- Apply RNC-LM to regression problems.
- Use RNC-LM for physics-informed neural networks.
- Improve large-scale potential-energy-surface fitting.
Topics
- Nonlinear Least-Squares
- Levenberg-Marquardt Method
- Riemann Normal Coordinates
- Physics-Informed Neural Networks
- Optimization Algorithms
- Geodesic Acceleration
Best for: AI Scientist, Research Scientist
Related on AIssential
Editorial summary, takeaway, and curation by AIssential. Original article published by Machine Learning.