HyperAIHyperAI

Command Palette

Search for a command to run...

3 months ago

Ensemble of MRR and NDCG models for Visual Dialog

Idan Schwartz

Ensemble of MRR and NDCG models for Visual Dialog

Abstract

Assessing an AI agent that can converse in human language and understand visual content is challenging. Generation metrics, such as BLEU scores favor correct syntax over semantics. Hence a discriminative approach is often used, where an agent ranks a set of candidate options. The mean reciprocal rank (MRR) metric evaluates the model performance by taking into account the rank of a single human-derived answer. This approach, however, raises a new challenge: the ambiguity and synonymy of answers, for instance, semantic equivalence (e.g., yeah' andyes'). To address this, the normalized discounted cumulative gain (NDCG) metric has been used to capture the relevance of all the correct answers via dense annotations. However, the NDCG metric favors the usually applicable uncertain answers such as `I don't know. Crafting a model that excels on both MRR and NDCG metrics is challenging. Ideally, an AI agent should answer a human-like reply and validate the correctness of any answer. To address this issue, we describe a two-step non-parametric ranking approach that can merge strong MRR and NDCG models. Using our approach, we manage to keep most MRR state-of-the-art performance (70.41% vs. 71.24%) and the NDCG state-of-the-art performance (72.16% vs. 75.35%). Moreover, our approach won the recent Visual Dialog 2020 challenge. Source code is available at https://github.com/idansc/mrr-ndcg.

Code Repositories

idansc/mrr-ndcg
Official
pytorch
Mentioned in GitHub

Benchmarks

BenchmarkMethodologyMetrics
visual-dialog-on-visdial-v10-test-stdTwo-Step
MRR: 0.7041
Mean Rank: 3.66
NDCG: 72.16
R@1: 58.18
R@10: 90.83
R@5: 83.85
visual-dialog-on-visdial-v10-test-std5xFGA + LS*+
MRR: 0.7124
Mean Rank: 2.96
R@1: 58.28
R@10: 94.45
R@5: 87.55
visual-dialog-on-visdial-v10-test-std5xFGA + LS
NDCG: 64.04
visual-dialog-on-visual-dialog-v1-0-test-std2 Step: Factor Graph Attention + VD-Bert
MRR (x 100): 69.92
Mean: 3.84
NDCG (x 100): 72.83
R@1: 58.3
R@10: 89.6
R@5: 81.55

Build AI with AI

From idea to launch — accelerate your AI development with free AI co-coding, out-of-the-box environment and best price of GPUs.

AI Co-coding
Ready-to-use GPUs
Best Pricing
Get Started

Hyper Newsletters

Subscribe to our latest updates
We will deliver the latest updates of the week to your inbox at nine o'clock every Monday morning
Powered by MailChimp