arXiv:2506.14460cs.LG2025-06被引 5

ZOO优化其实等价于单步策略优化,揭示了其背后的强化学习机制。

Zeroth-Order Optimization is Secretly Single-Step Policy Optimization

  • 将随机有限差分的ZOO方法视为单步策略优化,统一理论框架
  • 新算法ZoAR通过平均基线和查询复用,收敛更快、性能更优
  • 适合研究优化算法或强化学习交叉领域的研究人员

零阶优化(ZOO)在无法获取显式梯度或梯度计算成本高昂时极具价值。然而,主流ZOO方法(尤其是基于随机有限差分的)与强化学习(RL)等优化范式的内在联系尚未完全阐明。本文首次揭示:使用有限差分的ZOO等价于特定形式的单步策略优化(PO)。我们证明,常见ZOO算法所隐式优化的平滑目标函数,正是单步PO目标。进一步地,广泛使用的ZOO梯度估计器在数学上等价于带特定基线的REINFORCE估计器,从策略优化视角揭示了ZOO的方差降低机制。基于此统一框架,我们提出ZoAR(带平均基线与查询复用的零阶优化),引入受策略优化启发的方差减少技术:利用近期评估结果的平均基线,以及类似经验回放的查询复用。理论分析表明这些技术可降低方差并提升收敛性。大量实验证实理论正确性,且ZoAR在收敛速度与最终性能上显著优于现有方法。本工作为理解ZOO提供了新理论视角,并带来了切实可行的算法改进。

原文摘要 · Abstract (English)

Zeroth-Order Optimization (ZOO) provides powerful tools for optimizing functions where explicit gradients are unavailable or expensive to compute. However, the underlying mechanisms of popular ZOO methods, particularly those employing randomized finite differences, and their connection to other optimization paradigms like Reinforcement Learning (RL) are not fully elucidated. This paper establishes a fundamental and previously unrecognized connection: ZOO with finite differences is equivalent to a specific instance of single-step Policy Optimization (PO). We formally unveil that the implicitly smoothed objective function optimized by common ZOO algorithms is identical to a single-step PO objective. Furthermore, we show that widely used ZOO gradient estimators, are mathematically equivalent to the REINFORCE gradient estimator with a specific baseline function, revealing the variance-reducing mechanism in ZOO from a PO perspective.Built on this unified framework, we propose ZoAR (Zeroth-Order Optimization with Averaged Baseline and Query Reuse), a novel ZOO algorithm incorporating PO-inspired variance reduction techniques: an averaged baseline from recent evaluations and query reuse analogous to experience replay. Our theoretical analysis further substantiates these techniques reduce variance and enhance convergence. Extensive empirical studies validate our theory and demonstrate that ZoAR significantly outperforms other methods in terms of convergence speed and final performance. Overall, our work provides a new theoretical lens for understanding ZOO and offers practical algorithmic improvements derived from its connection to PO.

优化算法强化学习零阶优化策略优化

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。