Learning Cooperative Visual Dialog Agents with Deep Reinforcement Learning
This work addresses the challenge of developing cooperative AI agents for visual dialog, though it is incremental as it builds on existing RL and dialog methods.
The paper tackles the problem of training visual dialog agents to cooperate in an image guessing game using deep reinforcement learning, resulting in agents that invent their own communication protocol in a synthetic world and achieve significant performance gains over supervised learning agents on real-image data, with RL fine-tuning leading to more informative dialogs.
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.