Performative Policy Gradient: Optimality in Performative Reinforcement Learning
Debabrota Basu, Udvas Das, Brahim Driss, Uddalak Mukherjee
TL;DR
This work tackles performative reinforcement learning where deployed policies shift the environment, and introduces PePG, a policy-gradient method that achieves performative optimality under softmax policies with or without entropy regularisation. By proving a performative policy gradient theorem and a performative performance-difference lemma, the authors derive a gradient-based algorithm that converges to performatively optimal policies in softmax PeMDPs, with iteration complexity scaling as $ ilde{O}ig(| ext{S}|| ext{A}|^2/(oldsymbol{ ext{epsilon}}^2(1-oldsymbol{eta})^3)ig)$. The paper also contrasts optimality-seeking and stability-seeking approaches, providing both theoretical bounds and empirical validation in gridworld environments where PePG outperforms existing methods. This advances the theoretical and practical understanding of optimization under performative feedback, enabling more robust decision-making in environments that react to deployed policies. Overall, PePG offers a principled path to achieve performative optimality and demonstrates practical advantages over stability-focused alternatives in dynamic, feedback-driven RL settings.
Abstract
Post-deployment machine learning algorithms often influence the environments they act in, and thus shift the underlying dynamics that the standard reinforcement learning (RL) methods ignore. While designing optimal algorithms in this performative setting has recently been studied in supervised learning, the RL counterpart remains under-explored. In this paper, we prove the performative counterparts of the performance difference lemma and the policy gradient theorem in RL, and further introduce the Performative Policy Gradient algorithm (PePG). PePG is the first policy gradient algorithm designed to account for performativity in RL. Under softmax parametrisation, and also with and without entropy regularisation, we prove that PePG converges to performatively optimal policies, i.e. policies that remain optimal under the distribution shifts induced by themselves. Thus, PePG significantly extends the prior works in Performative RL that achieves performative stability but not optimality. Furthermore, our empirical analysis on standard performative RL environments validate that PePG outperforms standard policy gradient algorithms and the existing performative RL algorithms aiming for stability.
