arXiv Open Access 2024

LingGen: Scalable Multi-Attribute Linguistic Control via Power-Law Masking

Mohamed Elgaar Hadi Amiri
Lihat Sumber

Abstrak

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.

Topik & Kata Kunci

Penulis (2)

M

Mohamed Elgaar

H

Hadi Amiri

Format Sitasi

Elgaar, M., Amiri, H. (2024). LingGen: Scalable Multi-Attribute Linguistic Control via Power-Law Masking. https://arxiv.org/abs/2410.24201

Akses Cepat

Lihat di Sumber
Informasi Jurnal
Tahun Terbit
2024
Bahasa
en
Sumber Database
arXiv
Akses
Open Access ✓