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

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Kudo, Mikoto, Tanabe, Takumi, Wachi, Akifumi, Akimoto, Youhei
Formato: Preprint
Publicado: 2026
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866910087114653696
author Kudo, Mikoto
Tanabe, Takumi
Wachi, Akifumi
Akimoto, Youhei
author_facet Kudo, Mikoto
Tanabe, Takumi
Wachi, Akifumi
Akimoto, Youhei
contents 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.
format Preprint
id arxiv_https___arxiv_org_abs_2603_14867
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle Sample-Efficient Hypergradient Estimation for Decentralized Bi-Level Reinforcement Learning
Kudo, Mikoto
Tanabe, Takumi
Wachi, Akifumi
Akimoto, Youhei
Machine Learning
Artificial Intelligence
Computer Science and Game Theory
Multiagent Systems
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.
title Sample-Efficient Hypergradient Estimation for Decentralized Bi-Level Reinforcement Learning
topic Machine Learning
Artificial Intelligence
Computer Science and Game Theory
Multiagent Systems
url https://arxiv.org/abs/2603.14867