Learning to Compare for Better Training and Evaluation of Open Domain Natural Language Generation Models

By Wangchunshu Zhou, Ke Xu. 2020

50 citations recorded by the source corpus.

Automated evaluation of open domain natural language generation (NLG) models remains a challenge and widely used metrics such as BLEU and Perplexity can be misleading in some cases. In our paper, we propose to evaluate natural language generation models by learning to compare a pair of generated sentences by fine-tuning BERT, which has been shown to have good natural language understanding ability. We also propose to evaluate the model-level quality of NLG models with sample-level comparison results with skill rating system. While able to be trained in a fully self-supervised fashion, our model can be further fine-tuned with a little amount of human preference annotation to better imitate human judgment. In addition to evaluating trained models, we propose to apply our model as a performance indicator during training for better hyperparameter tuning and early-stopping. We evaluate our approach on both story generation and chit-chat dialogue response generation. Experimental results show that our model correlates better with human preference compared with previous automated evaluation approaches. Training with the proposed metric yields better performance in human evaluation, which further demonstrates the effectiveness of the proposed model.

Subfields: Artificial Intelligence, Natural Language Processing, Evaluation & Benchmarks

How researchers see this paper

The researchers whose trusted networks rank it highest, strongest advocate first. Every researcher gets a different order.

In Artificial Intelligence

  1. Jianfeng Gao
  2. Sergey Levine
  3. Mohit Bansal
  4. Percy Liang
  5. Yejin Choi
  6. Pieter Abbeel
  7. Luke S. Zettlemoyer
  8. Minlie Huang

In Natural Language Processing

  1. Jianfeng Gao
  2. Minlie Huang
  3. Elizabeth Clark
  4. Yejin Choi
  5. Asli Celikyilmaz
  6. Ming Zhou
  7. Zdeněk Kasner
  8. Jacob Hilton

In Evaluation & Benchmarks

  1. Pieter Abbeel
  2. Jianfeng Gao
  3. Shixiang Shane Gu
  4. Elizabeth Clark
  5. Asli Celikyilmaz
  6. Zdeněk Kasner
  7. Minlie Huang
  8. Ming Zhou

Citations in the corpus