Timezone: »

Tractable Control for Autoregressive Language Generation
Honghua Zhang · Meihua Dang · Nanyun Peng · Guy Van den Broeck

Tue Jul 25 05:00 PM -- 06:30 PM (PDT) @ Exhibit Hall 1 #205
Event URL: https://github.com/UCLA-StarAI/GeLaTo »
Despite the success of autoregressive large language models in text generation, it remains a major challenge to generate text that satisfies complex constraints: sampling from the conditional distribution ${\Pr}(\text{text} | \alpha)$ is intractable for even the simplest lexical constraints $\alpha$. To overcome this challenge, we propose to use tractable probabilistic models (TPMs) to impose lexical constraints in autoregressive text generation models, which we refer to as GeLaTo (Generating Language with Tractable Constraints). To demonstrate the effectiveness of this framework, we use distilled hidden Markov models, where we can efficiently compute ${\Pr}(\text{text} | \alpha)$, to guide autoregressive generation from GPT2. GeLaTo achieves state-of-the-art performance on challenging benchmarks for constrained text generation (e.g., CommonGen), beating various strong baselines by a large margin. Our work not only opens up new avenues for controlling large language models but also motivates the development of more expressive TPMs.

Author Information

Honghua Zhang (University of California, Los Angeles)
Meihua Dang (University of California, Los Angeles)
Nanyun Peng (UCLA)
Guy Van den Broeck (University of California, Los Angeles)

Related Events (a corresponding poster, oral, or spotlight)

More from the Same Authors