arXiv · 2104.07972
Language Models are Few-Shot Butlers
Abstract
Pretrained language models demonstrate strong performance in most NLP tasks when fine-tuned on small task-specific datasets. Hence, these autoregressive models constitute ideal agents to operate in text-based environments where language understanding and generative capabilities are essential. Nonetheless, collecting expert demonstrations in such environments is a time-consuming endeavour. We introduce a two-stage procedure to learn from a small set of demonstrations and further improve by interacting with an environment. We show that language models fine-tuned with only 1.2% of the expert demonstrations and a simple reinforcement learning algorithm achieve a 51% absolute improvement in success rate over existing methods in the ALFWorld environment.
Explore related subjects
Keep this discovery
Vincent Micheli, François Fleuret. 2021-04-16. Language Models are Few-Shot Butlers. https://arxiv.org/abs/2104.07972
Cite the original work for its findings. Save a collection to share your selection of sources.