English

Reinforced Video Captioning with Entailment Rewards

Computation and Language 2017-08-09 v1 Artificial Intelligence Computer Vision and Pattern Recognition Machine Learning

Abstract

Sequence-to-sequence models have shown promising improvements on the temporal task of video captioning, but they optimize word-level cross-entropy loss during training. First, using policy gradient and mixed-loss methods for reinforcement learning, we directly optimize sentence-level task-based metrics (as rewards), achieving significant improvements over the baseline, based on both automatic metrics and human evaluation on multiple datasets. Next, we propose a novel entailment-enhanced reward (CIDEnt) that corrects phrase-matching based metrics (such as CIDEr) to only allow for logically-implied partial matches and avoid contradictions, achieving further significant improvements over the CIDEr-reward model. Overall, our CIDEnt-reward model achieves the new state-of-the-art on the MSR-VTT dataset.

Keywords

Cite

@article{arxiv.1708.02300,
  title  = {Reinforced Video Captioning with Entailment Rewards},
  author = {Ramakanth Pasunuru and Mohit Bansal},
  journal= {arXiv preprint arXiv:1708.02300},
  year   = {2017}
}

Comments

EMNLP 2017 (9 pages)