Wendelin Böhmer

Training on Irrelevant States Implies Data Augmentation: Generalization in Contextual MDPs

2024 · arXiv (Cornell University)

In the zero-shot policy transfer (ZSPT) setting for contextual Markov decision processes (CMDP), agents train on a fixed, finite set of contexts and must generalize to new ones. Recent work has demonstrated that training on …