{"main_file_link":[{"open_access":"1","url":" https://doi.org/10.48550/arXiv.2107.05686"}],"status":"public","citation":{"ama":"Dittadi A, Träuble F, Wüthrich M, et al. The role of pretrained representations for the OOD generalization of reinforcement learning agents. In: 10th International Conference on Learning Representations. ; 2022.","ista":"Dittadi A, Träuble F, Wüthrich M, Widmaier F, Gehler P, Winther O, Locatello F, Bachem O, Schölkopf B, Bauer S. 2022. The role of pretrained representations for the OOD generalization of reinforcement learning agents. 10th International Conference on Learning Representations. ICLR: International Conference on Learning Representations.","apa":"Dittadi, A., Träuble, F., Wüthrich, M., Widmaier, F., Gehler, P., Winther, O., … Bauer, S. (2022). The role of pretrained representations for the OOD generalization of reinforcement learning agents. In 10th International Conference on Learning Representations. Virtual.","mla":"Dittadi, Andrea, et al. “The Role of Pretrained Representations for the OOD Generalization of Reinforcement Learning Agents.” 10th International Conference on Learning Representations, 2022.","chicago":"Dittadi, Andrea, Frederik Träuble, Manuel Wüthrich, Felix Widmaier, Peter Gehler, Ole Winther, Francesco Locatello, Olivier Bachem, Bernhard Schölkopf, and Stefan Bauer. “The Role of Pretrained Representations for the OOD Generalization of Reinforcement Learning Agents.” In 10th International Conference on Learning Representations, 2022.","ieee":"A. Dittadi et al., “The role of pretrained representations for the OOD generalization of reinforcement learning agents,” in 10th International Conference on Learning Representations, Virtual, 2022.","short":"A. Dittadi, F. Träuble, M. Wüthrich, F. Widmaier, P. Gehler, O. Winther, F. Locatello, O. Bachem, B. Schölkopf, S. Bauer, in:, 10th International Conference on Learning Representations, 2022."},"department":[{"_id":"FrLo"}],"conference":{"end_date":"2022-04-29","location":"Virtual","start_date":"2022-04-25","name":"ICLR: International Conference on Learning Representations"},"language":[{"iso":"eng"}],"day":"25","publication":"10th International Conference on Learning Representations","date_created":"2023-08-22T14:02:13Z","type":"conference","year":"2022","oa_version":"Preprint","quality_controlled":"1","abstract":[{"text":"Building sample-efficient agents that generalize out-of-distribution (OOD) in real-world settings remains a fundamental unsolved problem on the path towards achieving higher-level cognition. One particularly promising approach is to begin with low-dimensional, pretrained representations of our world, which should facilitate efficient downstream learning and generalization. By training 240 representations and over 10,000 reinforcement learning (RL) policies on a simulated robotic setup, we evaluate to what extent different properties of\r\npretrained VAE-based representations affect the OOD generalization of downstream agents. We observe that many agents are surprisingly robust to realistic distribution shifts, including the challenging sim-to-real case. In addition, we find that the generalization performance of a simple downstream proxy task reliably predicts the generalization performance of our RL agents\r\nunder a wide range of OOD settings. Such proxy tasks can thus be used to select pretrained representations that will lead to agents that generalize.","lang":"eng"}],"publication_status":"published","title":"The role of pretrained representations for the OOD generalization of reinforcement learning agents","extern":"1","author":[{"first_name":"Andrea","last_name":"Dittadi","full_name":"Dittadi, Andrea"},{"full_name":"Träuble, Frederik","last_name":"Träuble","first_name":"Frederik"},{"full_name":"Wüthrich, Manuel","last_name":"Wüthrich","first_name":"Manuel"},{"full_name":"Widmaier, Felix","last_name":"Widmaier","first_name":"Felix"},{"first_name":"Peter","full_name":"Gehler, Peter","last_name":"Gehler"},{"first_name":"Ole","last_name":"Winther","full_name":"Winther, Ole"},{"full_name":"Locatello, Francesco","last_name":"Locatello","id":"26cfd52f-2483-11ee-8040-88983bcc06d4","orcid":"0000-0002-4850-0683","first_name":"Francesco"},{"first_name":"Olivier","last_name":"Bachem","full_name":"Bachem, Olivier"},{"first_name":"Bernhard","last_name":"Schölkopf","full_name":"Schölkopf, Bernhard"},{"first_name":"Stefan","last_name":"Bauer","full_name":"Bauer, Stefan"}],"month":"04","date_published":"2022-04-25T00:00:00Z","_id":"14174","date_updated":"2023-09-11T09:48:36Z","external_id":{"arxiv":["2107.05686"]},"user_id":"2DF688A6-F248-11E8-B48F-1D18A9856A87","oa":1,"article_processing_charge":"No"}