Posterior Control of Blackbox Generation

Code

• github.com/XiangLi1999/PosteriorControl-NLG
  OfficialIn paperpytorch★ 23
• github.com/FranxYao/Gumbel-CRF
  pytorch★ 56

Abstract

Text generation often requires high-precision output that obeys task-specific rules. This fine-grained control is difficult to enforce with off-the-shelf deep learning models. In this work, we consider augmenting neural generation models with discrete control states learned through a structured latent-variable approach. Under this formulation, task-specific knowledge can be encoded through a range of rich, posterior constraints that are effectively trained into the model. This approach allows users to ground internal model decisions based on prior knowledge, without sacrificing the representational power of neural generative models. Experiments consider applications of this approach for text generation. We find that this method improves over standard benchmarks, while also providing fine-grained control.

Tasks

Reproductions