Title: Toward Agents That Reason About Their Computation
Primary area: reinforcement learning
Keywords: Reinforcement Learning

Abstract: While reinforcement learning agents can achieve superhuman performance in many complex tasks, they typically do not become more computationally efficient as they improve. 
In contrast, humans gradually require less cognitive effort as they become more proficient at a task.
If agents could reason about their compute as they learn, could they similarly reduce their computation footprint? 
If they could, we could have more energy efficient agents or free up compute cycles for other processes like planning.
In this paper, we experiment with showing agents the cost of their computation and giving them the ability to control when they use compute.
We conduct our experiments on the Arcade Learning Environment, and our results demonstrate that with the same training compute budget, agents that reason about their compute perform better on 75\% of games.
Furthermore, these agents use 3 times less compute on average.
We analyze individual games and show where agents gain these efficiencies.

TL;DR: Agents that reason about their compute achieve higher scores under identical training budgets, and learn strategies in each Atari game that use less compute.