arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2103.10415cs.CLcs.LG
收藏

用组合推理精化语言模型

Refining Language Models with Compositional Explanations

Peking University(北京大学);Tsinghua University(清华大学);University of Southern California(南加州大学)

Huihan Yao, Ying Chen, Qinyuan Ye, Xisen Jin, Xiang Ren

摘要:

预先训练的语言模型在文本分类任务上已经取得了成功,但容易从有偏见的数据集中学习虚假的相关性,因此在新领域中进行推断时很容易受到攻击。先前的工作揭示了这种虚假的模式,通过事后解释算法计算输入功能的重要性。此外,模型被正则化以使重要性分数与人类知识一致,从而消除了非预期的模型行为。然而,这样的正则化技术缺乏灵活性和覆盖范围,因为只有对一个预定义的功能列表的重要性分数进行调整,而更复杂的人类知识,如功能的相互作用和模式的泛化几乎不能被纳入。在这项工作中,我们建议通过收集人类提供的关于观察到的偏见的组成解释来完善目标域的学习语言模型。通过将这些解释解析成可执行的逻辑规则,来自一小组解释的人类指定的精化建议可以被推广到更多的训练样本。我们还引入了一个正则化项,允许调整特征的重要性和相互作用,以更好地纠正模型行为。我们证明了所提出的方法的有效性在两个文本分类任务,显示改进的性能在目标域以及改进后的模型公平性。

英文摘要:

Pre-trained language models have been successful on text classification tasks, but are prone to learning spurious correlations from biased datasets, and are thus vulnerable when making inferences in a new domain. Prior work reveals such spurious patterns via post-hoc explanation algorithms which compute the importance of input features. Further, the model is regularized to align the importance scores with human knowledge, so that the unintended model behaviors are eliminated. However, such a regularization technique lacks flexibility and coverage, since only importance scores towards a pre-defined list of features are adjusted, while more complex human knowledge such as feature interaction and pattern generalization can hardly be incorporated. In this work, we propose to refine a learned language model for a target domain by collecting human-provided compositional explanations regarding observed biases. By parsing these explanations into executable logic rules, the human-specified refinement advice from a small set of explanations can be generalized to more training examples. We additionally introduce a regularization term allowing adjustments for both importance and interaction of features to better rectify model behavior. We demonstrate the effectiveness of the proposed approach on two text classification tasks by showing improved performance in target domain as well as improved model fairness after refinement.

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

↑