中文

比海森矩阵累加更快地计算牛顿步

最优化与控制 2021-08-04 v1 机器学习 符号计算 系统与控制 系统与控制

摘要

计算具有 NN 个决策变量的通用函数的牛顿步需要 O(N3)O(N^3) 次浮点运算。在本文中,我们证明,给定函数的计算图,该界限可以降低到 O(mτ3)O(m\tau^3),其中 τ,m\tau, m 是该图树分解的宽度和大小。所提出的算法将基于 LQR 的非线性最优控制方法推广到一般优化问题,并且即使在 Hessian 矩阵稠密的情况下也能在迭代复杂度方面提供非平凡的增益。

关键词

引用

@article{arxiv.2108.01219,
  title  = {Computing the Newton-step faster than Hessian accumulation},
  author = {Akshay Srinivasan and Emanuel Todorov},
  journal= {arXiv preprint arXiv:2108.01219},
  year   = {2021}
}

备注

Presented at the Beyond First-order Methods workshop, ICML '21