paper-with-me

홈 › Papers

Sample-Efficient Hypergradient Estimation for Decentralized Bi-Level Reinforcement Learning

2026-03-16 · Mikoto Kudo, Takumi Tanabe, Akifumi Wachi, Youhei Akimoto arxiv

Many strategic decision-making problems, such as environment design for warehouse robots, can be naturally formulated as bi-level reinforcement learning (RL), where a leader agent optimizes its objective while a follower solves a Markov decision process (MDP) conditioned on the leader's decisions. In many situations, a fundamental challenge arises when the leader cannot intervene in the follower's optimization process; it can only observe the optimization outcome. We address this decentralized setting by deriving the hypergradient of the leader's objective, i.e., the gradient of the leader's strategy that accounts for changes in the follower's optimal policy. Unlike prior hypergradient-based methods that require extensive data for repeated state visits or rely on gradient estimators whose complexity can increase substantially with the high-dimensional leader's decision space, we leverage the Boltzmann covariance trick to derive an alternative hypergradient formulation. This enables efficient hypergradient estimation solely from interaction samples, even when the leader's decision space is high-dimensional. Additionally, to our knowledge, this is the first method that enables hypergradient-based optimization for 2-player Markov games in decentralized settings. Experiments highlight the impact of hypergradient updates and demonstrate our method's effectiveness in both discrete and continuous state tasks.

📄 PDF Abstract BibTeX arXiv:2603.14867

Code (0)

등록된 구현이 없습니다.

Tasks

Reinforcement Learning

Similar Papers 제목 키워드 기반

A Communication and Computation Efficient Fully First-order Method for Decentralized Bilevel Optimization

2024-10-18 · Min Wen, Chengchang Liu, Ahmed Abdelmoniem, Yipeng Zhou 외

Bilevel optimization, crucial for hyperparameter tuning, meta-learning and reinforcement learning, remains less explored in the decentralized learning paradigm, such as decentralized federated learning (DFL). Typically, …

Bilevel OptimizationFederated LearningMeta-Learning

On The Sample Complexity Bounds In Bilevel Reinforcement Learning

2025-03-22 · Mudit Gaur, Amrit Singh Bedi, Raghu Pasupathu, Vaneet Aggarwal

Bilevel reinforcement learning (BRL) has emerged as a powerful mathematical framework for studying generative AI alignment and related problems. While several principled algorithmic frameworks have been proposed, key the…

Bilevel Optimizationreinforcement-learningReinforcement Learning

Communication-Efficient Federated Hypergradient Computation via Aggregated Iterative Differentiation

2023-02-09 · Peiyao Xiao, Kaiyi Ji

Federated bilevel optimization has attracted increasing attention due to emerging machine learning and communication applications. The biggest challenge lies in computing the gradient of the upper-level objective functio…

Bilevel Optimization

Understanding the Generalization of Bilevel Programming in Hyperparameter Optimization: A Tale of Bias-Variance Decomposition

2026-02-20 · Yubo Zhou, Jun Shu, Junmin Liu, Deyu Meng arxiv

Gradient-based hyperparameter optimization (HPO) have emerged recently, leveraging bilevel programming techniques to optimize hyperparameter by estimating hypergradient w.r.t. validation loss. Nevertheless, previous theo…

Hyperparameter OptimizationFew-Shot Learning

Glocal Hypergradient Estimation with Koopman Operator

2024-02-05 · Ryuichiro Hataya, Yoshinobu Kawahara

Gradient-based hyperparameter optimization methods update hyperparameters using hypergradients, gradients of a meta criterion with respect to hyperparameters. Previous research used two distinct update strategies: optimi…

Hyperparameter Optimization