用控制理论统一设计优化器,让训练更稳定高效
Riemannian Lyapunov Optimizer: A Unified Framework for Optimization
- 将优化视为黎曼流形上的动态系统,通过李雅普诺夫函数保证收敛
- 识别出两种训练阶段:快速对齐速度状态,再在目标流形上受控演化
- 可生成新优化器,适合追求理论严谨性的算法研究者
我们提出黎曼李雅普诺夫优化器(RLOs),一个将经典优化算法统一在几何框架下的算法家族。不同于对现有优化器的启发式改进,RLOs 系统性地源自一种新颖的控制理论框架,将优化重新诠释为定义在黎曼参数流形上的离散时间受控动力系统。该框架的核心是识别出一个正常吸引不变流形(NAIM),它将训练动态组织为两个阶段:速度状态快速对齐至目标图,随后在其中受控演化。我们通过构建严格李雅普诺夫函数,形式化证明其收敛至目标流形。这一视角催生了一个可构造的“优化器生成器”,不仅恢复了经典算法,还支持理性设计新优化器。我们通过几何诊断验证理论,并在大规模基准测试中展示其达到前沿性能。总体而言,RLOs 桥接了控制理论与现代机器学习优化,提供统一语言和系统化工具,用于设计稳定有效的优化器。
原文摘要 · Abstract (English)
We introduce Riemannian Lyapunov Optimizers (RLOs), a family of optimization algorithms that unifies classic optimizers within one geometric framework. Unlike heuristic improvements to existing optimizers, RLOs are systematically derived from a novel control-theoretic framework that reinterprets optimization as an extended state discrete-time controlled dynamical system on a Riemannian parameter manifold. Central to this framework is the identification of a Normally Attracting Invariant Manifold (NAIM), which organizes training dynamics into two distinct stages: rapid alignment of the speed state to a target graph, followed by controlled evolution within it. We formalize this by constructing a strict Lyapunov function that certifies convergence to a target manifold. This perspective yields a constructive ``optimizer generator" that not only recovers classic algorithms but enables the principled design of RLOs. We validate our theory via geometric diagnostics and demonstrate that grounding optimizer design in control theory yields state-of-the-art performance in large-scale benchmarks. Overall, RLOs bridge control theory and modern machine learning optimization, providing a unified language and a systematic toolkit for designing stable, effective optimizers.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。