arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2003.09772cs.LGcs.AIcs.CLstat.ML
收藏

Invariant Rationalization

MIT-IBM Watson AI Lab(麻省理工学院-IBM沃森人工智能实验室);IBM Research(IBM研究);CSAIL MIT(麻省理工学院计算机科学与人工智能实验室)

Shiyu Chang, Yang Zhang, Mo Yu, Tommi S. Jaakkola

英文摘要:

Selective rationalization improves neural network interpretability by identifying a small subset of input features -- the rationale -- that best explains or supports the prediction. A typical rationalization criterion, i.e. maximum mutual information (MMI), finds the rationale that maximizes the prediction performance based only on the rationale. However, MMI can be problematic because it picks up spurious correlations between the input features and the output. Instead, we introduce a game-theoretic invariant rationalization criterion where the rationales are constrained to enable the same predictor to be optimal across different environments. We show both theoretically and empirically that the proposed rationales can rule out spurious correlations, generalize better to different test scenarios, and align better with human judgments. Our data and code are available.

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

↑