LAD: Language Augmented Diffusion for Reinforcement Learning

Edwin Zhang, Yujie Lu, William Yang Wang, Amy Zhang

Keywords: planning, diffusion, language, RL, robotics

TL;DR: We propose the usage of language augmented diffusion models for planning.

Abstract: Learning skills from language potentially provides a powerful avenue for generalization in RL, although it remains a challenging task as it requires agents to capture the complex interdependencies between language, actions and states, also known as language grounding. In this paper, we propose leveraging Language Augmented Diffusion models as a language-to-plan generator (LAD). We demonstrate comparable performance of LAD with the state of the art on the CALVIN benchmark with a much simpler architecture and conduct an analysis on the properties of language conditioned diffusion in reinforcement learning.

Community Implementations: [ 2 code implementations](https://www.catalyzex.com/paper/lad-language-augmented-diffusion-for/code)