Policy Learning with a Language Bottleneck - Unité d'Informatique et d'Ingénierie des Systèmes (U2IS)
Pré-Publication, Document De Travail Année : 2024

Policy Learning with a Language Bottleneck

Résumé

Modern AI systems such as self-driving cars and game-playing agents achieve superhuman performance, but often lack human-like features such as generalization, interpretability and human inter-operability. Inspired by the rich interactions between language and decision-making in humans, we introduce Policy Learning with a Language Bottleneck (PLLB), a framework enabling AI agents to generate linguistic rules that capture the strategies underlying their most rewarding behaviors. PLLB alternates between a rule generation step guided by language models, and an update step where agents learn new policies guided by rules. In a two-player communication game, a maze solving task, and two image reconstruction tasks, we show that PLLB agents are not only able to learn more interpretable and generalizable behaviors, but can also share the learned rules with human users, enabling more effective human-AI coordination.
Fichier principal
Vignette du fichier
2405.04118v1.pdf (1.74 Mo) Télécharger le fichier
Origine Fichiers produits par l'(les) auteur(s)

Dates et versions

hal-04810194 , version 1 (28-11-2024)

Licence

Identifiants

Citer

Megha Srivastava, Cédric Colas, Dorsa Sadigh, Jacob Andreas. Policy Learning with a Language Bottleneck. 2024. ⟨hal-04810194⟩
0 Consultations
0 Téléchargements

Altmetric

Partager

More