Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Reward Transfer from Inverse Reinforcement Learning: A Coupled Minimax Approach

About

We study the transfer of rewards learned using inverse reinforcement learning from expert demonstrations in one environment to reinforcement learning in a new, different environment. This arises naturally when demonstrations are collected in a controlled environment. We formulate the problem as a joint system of Bellman equations across the source and target environments and develop minimax estimators for the target soft-$q$-function. Whereas a sequential solution approach first estimates the source reward and then plugs it into the target control problem, a coupled approach solves the source and target system of equations jointly. We show that, in contrast to the sequential approach, the coupled approach removes the first-order influence of source Bellman residual error. We characterize the local behavior of each approach, develop finite-sample soft-$q$-function error bounds, and prove regret guarantees for the resulting soft-control policy. An empirical investigation using a sepsis simulator validates the theoretical comparison.

Guang-Yuan Hao, Lars van der Laan, Aur\'elien Bibaut, Nathan Kallus• 2026

Related benchmarks

TaskDatasetResultRank
Reward transferSepsis Simulator (D1 fraction 0.2)
Regret0.0546
9
Showing 1 of 1 rows

Other info

Follow for update