arXiv:2410.06003cs.LG2024-10NeurIPS被引 17

提出新准则让模型自动忽略无关特征,提升解释性提取质量

Is the MMI Criterion Necessary for Interpretability? Degenerating Non-causal Features to Plain Noise for Self-Rationalization

  • 将伪相关特征视为普通噪声,统一处理以简化优化
  • 在6个数据集上使解释内容与人工标注重合度提升最高10.4%
  • 适合需要可靠解释的复杂数据场景,如医疗、金融决策

可解释性研究中,关键任务是从完整输入中提取少量重要理由。当前主流方法采用最大互信息(MMI)准则,但在某些数据集中,存在与标签非因果相关的伪特征,这些特征同样具有高互信息,使损失函数结构复杂化。尽管已有基于惩罚项的方法(如不变性惩罚、干预惩罚)试图抑制伪特征,但这些方法仍区分伪特征与普通噪声,阻碍了对因果理由的发现。本文提出一种新准则——最大化剩余差异(MRD),理论发现:从输入中移除普通噪声或伪特征不会改变剩余成分对任务标签的条件分布,而移除因果特征则会显著改变该分布。基于此,MRD准则将伪特征等同于噪声,使模型在含丰富伪特征的数据上表现如同在干净数据上一样。在六个常用数据集上的实验表明,相比多个先进的MMI变体,本方法使解释质量(与人工标注重合度)最高提升10.4%。代码已开源。

原文摘要 · Abstract (English)

An important line of research in the field of explainability is to extract a small subset of crucial rationales from the full input. The most widely used criterion for rationale extraction is the maximum mutual information (MMI) criterion. However, in certain datasets, there are spurious features non-causally correlated with the label and also get high mutual information, complicating the loss landscape of MMI. Although some penalty-based methods have been developed to penalize the spurious features (e.g., invariance penalty, intervention penalty, etc) to help MMI work better, these are merely remedial measures. In the optimization objectives of these methods, spurious features are still distinguished from plain noise, which hinders the discovery of causal rationales. This paper aims to develop a new criterion that treats spurious features as plain noise, allowing the model to work on datasets rich in spurious features as if it were working on clean datasets, thereby making rationale extraction easier. We theoretically observe that removing either plain noise or spurious features from the input does not alter the conditional distribution of the remaining components relative to the task label. However, significant changes in the conditional distribution occur only when causal features are eliminated. Based on this discovery, the paper proposes a criterion for \textbf{M}aximizing the \textbf{R}emaining \textbf{D}iscrepancy (MRD). Experiments on six widely used datasets show that our MRD criterion improves rationale quality (measured by the overlap with human-annotated rationales) by up to $10.4\%$ as compared to several recent competitive MMI variants. Code: \url{https://github.com/jugechengzi/Rationalization-MRD}.

可解释性理由提取伪特征优化准则

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。