← 返回资源分享
Learning Cooperative Visual Dialog Agents with Deep Reinforcement Learning
论文
论文
发布时间2017-03-20
发表ICCV 2017 10 · arXiv:1703.06585
作者:Abhishek Das,Dhruv Batra,Satwik Kottur,José M. F. Moura,Stefan Lee
详细介绍
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.
代码仓库 (7)
zilongzheng/visdial-gnn官方PyTorch
batra-mlp-lab/visdial-rlPyTorch
vmurahari3/visdial-diversityPyTorch
kdexd/lang-emerge-parlaiPyTorch
Cremiy/visdial-rlPyTorch
naver/aqm-plusPyTorch
sea-snell/implicit-language-q-learningPyTorch
