CAREL: Instruction-guided reinforcement learning with cross-modal auxiliary objectives

Autor: Saghafian, Armin, Izadi, Amirmohammad, Dijujin, Negin Hashemi, Baghshah, Mahdieh Soleymani
Rok vydání: 2024
Předmět:
Druh dokumentu: Working Paper
Popis: Grounding the instruction in the environment is a key step in solving language-guided goal-reaching reinforcement learning problems. In automated reinforcement learning, a key concern is to enhance the model's ability to generalize across various tasks and environments. In goal-reaching scenarios, the agent must comprehend the different parts of the instructions within the environmental context in order to complete the overall task successfully. In this work, we propose CAREL (Cross-modal Auxiliary REinforcement Learning) as a new framework to solve this problem using auxiliary loss functions inspired by video-text retrieval literature and a novel method called instruction tracking, which automatically keeps track of progress in an environment. The results of our experiments suggest superior sample efficiency and systematic generalization for this framework in multi-modal reinforcement learning problems. Our code base is available here.
Databáze: arXiv