Search papers, labs, and topics across Lattice.
This paper introduces a novel intrinsic reward for maximum entropy RL based on the entropy of discounted future state-action feature visitations. They prove that the expected sum of these rewards is a lower bound on the entropy of discounted state-action features visited from initial states, connecting it to a maximum entropy objective. Experiments show this approach improves feature visitation within individual trajectories and speeds up convergence for exploration-only agents, though it slightly reduces feature visitation across different trajectories.
Maximizing entropy of future state-action visitations boosts feature coverage within single RL trajectories, offering a new exploration strategy.
Maximum entropy reinforcement learning motivates agents to explore states and actions to maximize the entropy of some distribution, typically by providing additional intrinsic rewards proportional to that entropy function. In this paper, we study intrinsic rewards proportional to the entropy of the discounted distribution of state-action features visited during future time steps. This approach is motivated by two results. First, we show that the expected sum of these intrinsic rewards is a lower bound on the entropy of the discounted distribution of state-action features visited in trajectories starting from the initial states, which we relate to an alternative maximum entropy objective. Second, we show that the distribution used in the intrinsic reward definition is the fixed point of a contraction operator and can therefore be estimated off-policy. Experiments highlight that the new objective leads to improved visitation of features within individual trajectories, in exchange for slightly reduced visitation of features in expectation over different trajectories, as suggested by the lower bound. It also leads to improved convergence speed for learning exploration-only agents. Control performance remains similar across most methods on the considered benchmarks.
