English

Generative Bridging Network in Neural Sequence Prediction

Artificial Intelligence 2018-12-03 v6 Machine Learning Machine Learning

Abstract

In order to alleviate data sparsity and overfitting problems in maximum likelihood estimation (MLE) for sequence prediction tasks, we propose the Generative Bridging Network (GBN), in which a novel bridge module is introduced to assist the training of the sequence prediction model (the generator network). Unlike MLE directly maximizing the conditional likelihood, the bridge extends the point-wise ground truth to a bridge distribution conditioned on it, and the generator is optimized to minimize their KL-divergence. Three different GBNs, namely uniform GBN, language-model GBN and coaching GBN, are proposed to penalize confidence, enhance language smoothness and relieve learning burden. Experiments conducted on two recognized sequence prediction tasks (machine translation and abstractive text summarization) show that our proposed GBNs can yield significant improvements over strong baselines. Furthermore, by analyzing samples drawn from different bridges, expected influences on the generator are verified.

Keywords

Cite

@article{arxiv.1706.09152,
  title  = {Generative Bridging Network in Neural Sequence Prediction},
  author = {Wenhu Chen and Guanlin Li and Shuo Ren and Shujie Liu and Zhirui Zhang and Mu Li and Ming Zhou},
  journal= {arXiv preprint arXiv:1706.09152},
  year   = {2018}
}

Comments

Accepted to NAACL 2018

R2 v1 2026-06-22T20:31:51.029Z