2024
SiT: Symmetry-invariant Transformers for Generalisation in Reinforcement Learning
ICML 2024poster
An open challenge in reinforcement learning (RL) is the effective deployment of a trained policy to new or slightly different situations as well as semantically-similar environments. We introduce **S**ymmetry-**I**nvariant **T**ransformer (**SiT**), a scalable vision transformer (ViT) that leverages…