English

Learning Cooperative Visual Dialog Agents with Deep Reinforcement Learning

Computer Vision and Pattern Recognition 2017-03-22 v2 Artificial Intelligence Computation and Language Machine Learning

Abstract

We introduce the first goal-driven training for visual question answering and dialog agents. Specifically, we pose a cooperative 'image guessing' game between two agents -- Qbot and Abot -- who communicate in natural language dialog so that Qbot can select an unseen image from a lineup of images. We use deep reinforcement learning (RL) to learn the policies of these agents end-to-end -- from pixels to multi-agent multi-round dialog to game reward. We demonstrate two experimental results. First, as a 'sanity check' demonstration of pure RL (from scratch), we show results on a synthetic world, where the agents communicate in ungrounded vocabulary, i.e., symbols with no pre-specified meanings (X, Y, Z). We find that two bots invent their own communication protocol and start using certain symbols to ask/answer about certain visual attributes (shape/color/style). Thus, we demonstrate the emergence of grounded language and communication among 'visual' dialog agents with no human supervision. Second, we conduct large-scale real-image experiments on the VisDial dataset, where we pretrain with supervised dialog data and show that the RL 'fine-tuned' agents significantly outperform SL agents. Interestingly, the RL Qbot learns to ask questions that Abot is good at, ultimately resulting in more informative dialog and a better team.

Keywords

Cite

@article{arxiv.1703.06585,
  title  = {Learning Cooperative Visual Dialog Agents with Deep Reinforcement Learning},
  author = {Abhishek Das and Satwik Kottur and José M. F. Moura and Stefan Lee and Dhruv Batra},
  journal= {arXiv preprint arXiv:1703.06585},
  year   = {2017}
}

Comments

11 pages, 4 figures, 2 tables, webpage: http://visualdialog.org/

R2 v1 2026-06-22T18:50:26.525Z