arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2201.09119cs.CLcs.AIcs.LGstat.ML

A Causal Lens for Controllable Text Generation

UC San Diego(加州大学圣迭戈分校);AWS AI, Amazon(亚马逊云科技人工智能)

Zhiting Hu, Li Erran Li

英文摘要:

Controllable text generation concerns two fundamental tasks of wide applications, namely generating text of given attributes (i.e., attribute-conditional generation), and minimally editing existing text to possess desired attributes (i.e., text attribute transfer). Extensive prior work has largely studied the two problems separately, and developed different conditional models which, however, are prone to producing biased text (e.g., various gender stereotypes). This paper proposes to formulate controllable text generation from a principled causal perspective which models the two tasks with a unified framework. A direct advantage of the causal formulation is the use of rich causality tools to mitigate generation biases and improve control. We treat the two tasks as interventional and counterfactual causal inference based on a structural causal model, respectively. We then apply the framework to the challenging practical setting where confounding factors (that induce spurious correlations) are observable only on a small fraction of data. Experiments show significant superiority of the causal approach over previous conditional models for improved control accuracy and reduced bias.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑