vix.ing · top · new · best · stats · spec

An Empirical Investigation of Pre-Trained Transformer Language Models for Open-Domain Dialogue Generation

2020/03/09 by Piji Li, Li, Piji
Computer Science · #Algorithm #Artificial Intelligence (cs.AI) #Artificial intelligence #Baseline (sea) #Beam search #Computation and Language (cs.CL) #Computer science #Context (archaeology) #Decoding methods #FOS: Computer and information sciences #Language model #Machine learning #Natural Language Processing Techniques #Natural language processing #Relevance (law) #Search algorithm #Speech and dialogue systems #Speech recognition #Topic Modeling #Transformer #cs.AI #cs.CL

paper · pdf · doi:10.48550/arxiv.2003.04195

26 pages

arxiv created 2020/03/09 · openalex publication_date 2020/03/09 · arxiv updated 2020/03/10 · openalex created_date 2020/03/13 · openalex updated_date 2026/07/28

Abstract

We present an empirical investigation of pre-trained Transformer-based auto-regressive language models for the task of open-domain dialogue generation. Training paradigm of pre-training and fine-tuning is employed to conduct the parameter learning. Corpora of News and Wikipedia in Chinese and English are collected for the pre-training stage respectively. Dialogue context and response are concatenated into a single sequence utilized as the input of the models during the fine-tuning stage. A weighted joint prediction paradigm for both context and response is designed to evaluate the performance of models with or without the loss term for context prediction. Various of decoding strategies such as greedy search, beam search, top-k sampling, etc. are employed to conduct the response text generation. Extensive experiments are conducted on the typical single-turn and multi-turn dialogue corpora such as Weibo, Douban, Reddit, DailyDialog, and Persona-Chat. Detailed numbers of automatic evaluation metrics on relevance and diversity of the generated results for the languages models as well as the baseline approaches are reported.

Citations

Related