A Plug-and-Play Method for Controlled Text Generation

Abstract

Large pre-trained language models have repeatedly shown their ability to produce fluent text. Yet even when starting from a prompt, generation can continue in many plausible directions. Current decoding methods with the goal of controlling generation, e.g., to ensure specific words are included, either require additional models or fine-tuning, or work poorly when the task at hand is semantically unconstrained, e.g., story generation. In this work, we present a plug-and-play decoding method for controlled language generation that is so simple and intuitive, it can be described in a single sentence: given a topic or keyword, we add a shift to the probability distribution over our vocabulary towards semantically similar words. We show how annealing this distribution can be used to impose hard constraints on language generation, something no other plug-and-play method is currently able to do with SOTA language generators. Despite the simplicity of this approach, we see it works incredibly well in practice: decoding from GPT-2 leads to diverse and fluent sentences while guaranteeing the appearance of given guide words. We perform two user studies, revealing that (1) our method outperforms competing methods in human evaluations; and(2) forcing the guide words to appear in the generated text has no impact on the fluency of the generated text. 1

Using a Language Model to Generate Music in its Symbolic Domain while Controlling its Perceived Emotion

“…Later research then proposed using different methods for text generation control that do not demand re-training the large-scale NLP models, one of which is Keyword2Text (K2T) [18]. K2T uses keywords as hard and soft constraints for text generation.…”

DFX: A Low-latency Multi-FPGA Appliance for Accelerating Transformer-based Text Generation

“…Transformer [1] is a deep learning language model that uses the mechanism of attention, which gives a different weight of significance to each part of the input data. By solving the recursion and lack of global dependency problem of recurrent neural network (RNN) [2] and long short-term memory (LSTM) [3], the transformer is becoming the de facto standard for natural language processing (NLP) applications such as text generation [4], [5], text classification [6], [7], and machine translation [8], [9]. Among them, text generation, broadly referred to as natural language generation (NLG), is related to the automatic generation of human-readable text by a computer.…”