Natural Language-conditioned Reinforcement Learning with Inside-out Task Language Development and Translation
This addresses the challenge of agents following human instructions in RL by reducing learning burden, though it is incremental as it builds on existing language-conditioned RL methods.
The paper tackles the problem of natural language-conditioned reinforcement learning by proposing an inside-out scheme that uses a task language to reduce complexity, resulting in a 13.4% improvement in success rate and better adaptation to unseen instructions.
Natural Language-conditioned reinforcement learning (RL) enables the agents to follow human instructions. Previous approaches generally implemented language-conditioned RL by providing human instructions in natural language (NL) and training a following policy. In this outside-in approach, the policy needs to comprehend the NL and manage the task simultaneously. However, the unbounded NL examples often bring much extra complexity for solving concrete RL tasks, which can distract policy learning from completing the task. To ease the learning burden of the policy, we investigate an inside-out scheme for natural language-conditioned RL by developing a task language (TL) that is task-related and unique. The TL is used in RL to achieve highly efficient and effective policy training. Besides, a translator is trained to translate NL into TL. We implement this scheme as TALAR (TAsk Language with predicAte Representation) that learns multiple predicates to model object relationships as the TL. Experiments indicate that TALAR not only better comprehends NL instructions but also leads to a better instruction-following policy that improves 13.4% success rate and adapts to unseen expressions of NL instruction. The TL can also be an effective task abstraction, naturally compatible with hierarchical RL.