Current browse context:
cs.LG
Change to browse by:
References & Citations
Computer Science > Machine Learning
Title: Large-scale Pretraining for Visual Dialog: A Simple State-of-the-Art Baseline
(Submitted on 5 Dec 2019 (v1), last revised 31 Mar 2020 (this version, v2))
Abstract: Prior work in visual dialog has focused on training deep neural models on VisDial in isolation. Instead, we present an approach to leverage pretraining on related vision-language datasets before transferring to visual dialog. We adapt the recently proposed ViLBERT (Lu et al., 2019) model for multi-turn visually-grounded conversations. Our model is pretrained on the Conceptual Captions and Visual Question Answering datasets, and finetuned on VisDial. Our best single model outperforms prior published work (including model ensembles) by more than 1% absolute on NDCG and MRR. Next, we find that additional finetuning using "dense" annotations in VisDial leads to even higher NDCG -- more than 10% over our base model -- but hurts MRR -- more than 17% below our base model! This highlights a trade-off between the two primary metrics -- NDCG and MRR -- which we find is due to dense annotations not correlating well with the original ground-truth answers to questions.
Submission history
From: Vishvak Murahari [view email][v1] Thu, 5 Dec 2019 04:51:11 GMT (5552kb,D)
[v2] Tue, 31 Mar 2020 03:12:26 GMT (6161kb,D)
Link back to: arXiv, form interface, contact.