arXiv · 2410.24201
LingGen: Scalable Multi-Attribute Linguistic Control via Power-Law Masking
Abstract
We present LingGen, a controlled text generation model that allows fine-grained control over a large number of real-valued linguistic attributes. It encodes target attribute values with a dedicated linguistic attribute encoder and conditions the language model by injecting the resulting representation into the language model using the beginning-of-sequence (BOS) embeddings. To improve robustness when controlling different attribute subsets, we introduce P-MASKING, which samples per-example attribute masking rates from a truncated Pareto distribution during training. Across 1-40 control attributes, LingGen achieves the lowest average control error among evaluated methods, while remaining efficient at inference and receiving the highest fluency scores in human evaluation. Ablations show that Pareto-sampled masking and BOS-based injection are effective choices compared to alternative masking and integration variants.
Explore related subjects
Keep this discovery
Mohamed Elgaar, Hadi Amiri. 2024-10-31. LingGen: Scalable Multi-Attribute Linguistic Control via Power-Law Masking. https://arxiv.org/abs/2410.24201
Cite the original work for its findings. Save a collection to share your selection of sources.