Paper ID: 2207.13813

Structural Similarity for Improved Transfer in Reinforcement Learning

C. Chace Ashcraft, Benjamin Stoler, Chigozie Ewulum, Susama Agarwala

Transfer learning is an increasingly common approach for developing performant RL agents. However, it is not well understood how to define the relationship between the source and target tasks, and how this relationship contributes to successful transfer. We present an algorithm called Structural Similarity for Two MDPS, or SS2, that calculates a state similarity measure for states in two finite MDPs based on previously developed bisimulation metrics, and show that the measure satisfies properties of a distance metric. Then, through empirical results with GridWorld navigation tasks, we provide evidence that the distance measure can be used to improve transfer performance for Q-Learning agents over previous implementations.

Submitted: Jul 27, 2022