A Distributional Lens for Multi-Aspect Controllable Text Generation
Yuxuan Gu, Xiaocheng Feng, Sicheng Ma, Lingyuan Zhang, Heng Gong, Bing Qin
Abstract
Multi-aspect controllable text generation is a more challenging and practical task than singleaspect control. Existing methods achieve complex multi-aspect control by fusing multiple controllers learned from single-aspect, but suffer from attribute degeneration caused by the mutual interference of these controllers. To address this, we provide observations on attribute fusion from a distributional perspective and propose to directly search for the intersection areas of multiple attribute distributions as their combination for generation. Our method first estimates the attribute space with an autoencoder structure. Afterward, we iteratively approach the intersections by jointly minimizing distances to points representing different attributes. Finally, we map them to attributerelevant sentences with a prefix-tuning-based decoder. Experiments on the three-aspect control task, including sentiment, topic, and detoxification aspects, reveal that our method outperforms several strong baselines on attribute relevance and text quality and achieves the SOTA. Further analysis also supplies some explanatory support for the effectiveness of our approach 1 .
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Cited by top-tier papers7
- Generative Visual Prompt: Unifying Distributional Control of Pre-Trained Generative ModelsChen Henry Wu, Saman Motamed, Shaunak Srivastava, Fernando De la TorreNeurIPS 2022 · 43 citations
- Benchmarking Large Language Models on Controllable Generation under Diversified InstructionsYihan Chen, Benfeng Xu, Quan Wang, Yi Liu et al.AAAI 2024 · 42 citations
- An Extensible Plug-and-Play Method for Multi-Aspect Controllable Text GenerationXuancheng Huang, Zijun Liu, Peng Li, Tao Li et al.ACL 2023 · 4 citations
- FreeCtrl: Constructing Control Centers with Feedforward Layers for Learning-Free Controllable Text GenerationZijian Feng, Hanzhang Zhou, Kezhi Mao, Zixiao ZhuACL 2024 · 3 citations
- Semantic Space Grounded Weighted Decoding for Multi-Attribute Controllable Dialogue GenerationZhiling Zhang, Mengyue Wu, Kenny Q. ZhuEMNLP 2023 · 2 citations
Builds on13
- Language Models are Few-Shot LearnersTom B. Brown, Benjamin Mann, Nick Ryder, Melanie Subbiah et al.NeurIPS 2020 · 64,255 citations
- Plug and Play Language Models: A Simple Approach to Controlled Text GenerationSumanth Dathathri, Andrea Madotto, Janice Lan, Jane Hung et al.ICLR 2020 · 1,166 citations
- MAUVE: Measuring the Gap Between Neural Text and Human Text using Divergence FrontiersKrishna Pillutla, Swabha Swayamdipta, Rowan Zellers, John Thickstun et al.NeurIPS 2021 · 606 citations
- COLD Decoding: Energy-based Constrained Text Generation with Langevin DynamicsLianhui Qin, Sean Welleck, Daniel Khashabi, Yejin ChoiNeurIPS 2022 · 217 citations
- A Distributional Approach to Controlled Text GenerationMuhammad Khalifa, Hady Elsahar, Marc DymetmanICLR 2021 · 135 citations
Related papers
- Multi-Aspect Controllable Text Generation with Disentangled Counterfactual AugmentationYi Liu, Xiangyu Liu, Xiangrong Zhu, Wei HuACL 2024 · 6 citations
- Air-Decoding: Attribute Distribution Reconstruction for Decoding-Time Controllable Text GenerationTianqi Zhong, Quan Wang, Jingxuan Han, Yongdong Zhang et al.EMNLP 2023 · 7 citations
- Seen to Unseen: Exploring Compositional Generalization of Multi-Attribute Controllable Dialogue GenerationWeihao Zeng, Lulu Zhao, Keqing He, Ruotong Geng et al.ACL 2023 · 1 citation
- Controllable Text Generation via Probability Density Estimation in the Latent SpaceYuxuan Gu, Xiaocheng Feng, Sicheng Ma, Lingyuan Zhang et al.ACL 2023 · 8 citations
- DExperts: Decoding-Time Controlled Text Generation with Experts and Anti-ExpertsAlisa Liu, Maarten Sap, Ximing Lu, Swabha Swayamdipta et al.ACL 2021
