arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2003.02738cs.LGcs.CLstat.ML

BERT as a Teacher: Contextual Embeddings for Sequence-Level Reward

ETH Zürich(苏黎世联邦理工学院)

Florian Schmidt, Thomas Hofmann

英文摘要:

Measuring the quality of a generated sequence against a set of references is a central problem in many learning frameworks, be it to compute a score, to assign a reward, or to perform discrimination. Despite great advances in model architectures, metrics that scale independently of the number of references are still based on n-gram estimates. We show that the underlying operations, counting words and comparing counts, can be lifted to embedding words and comparing embeddings. An in-depth analysis of BERT embeddings shows empirically that contextual embeddings can be employed to capture the required dependencies while maintaining the necessary scalability through appropriate pruning and smoothing techniques. We cast unconditional generation as a reinforcement learning problem and show that our reward function indeed provides a more effective learning signal than n-gram reward in this challenging setting.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑