Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

A Distributional Lens for Multi-Aspect Controllable Text Generation

About

Multi-aspect controllable text generation is a more challenging and practical task than single-aspect 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 attribute-relevant 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.

Yuxuan Gu, Xiaocheng Feng, Sicheng Ma, Lingyuan Zhang, Heng Gong, Bing Qin• 2022

Related benchmarks

TaskDatasetResultRank
DetoxificationJigsaw (test)
Perplexity (PPL)46.2
29
Controllable Text GenerationYelp (test)
Perplexity (PPL)52.59
20
Sentiment ControlIMDB (test)
Sentiment Accuracy (Avg)92.5
11
Topic ControlAGNews (test)
Avg Topic Accuracy90.4
11
Multi-Aspect Controllable Text GenerationCompMCTG 1.0 (Original)
Aid Score85.46
10
Multi-Aspect Controllable Text GenerationCompMCTG 1.0 (Hold-Out)
Aid Score84.84
10
Multi-Aspect Controllable Text GenerationCompMCTG 1.0 (ACD)
Aid Score85.54
10
Multi-attribute Conditional Text GenerationCompMCTG In-distribution Few-shot 1.0 (test)
Accuracy87.81
10
Controllable Text GenerationMulti-Attribute Control
Average Score87.4
10
Multi-Aspect Controllable Text GenerationCompMCTG Overall Summary Average 1.0
Aavg Score72.19
10
Showing 10 of 21 rows

Other info

Follow for update