Invariant Rationalization
Not on the map. Why?
Abstract (arXiv)
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.
Key facts. Cited by 131 papers in the map (data through 14 Oct 2026). First posted on arXiv 22 Mar 2020.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2021: 25; 2022: 29; 2023: 36; 2024: 27; 2025: 11; 2026: 3.
131 papers.
Sort:
By area:
- 2021Towards Out-Of-Distribution Generalization: A Surveycited by 436
- 2021Invariance Principle Meets Information Bottleneck for Out-of-Distribution Generalizationcited by 220
- 2022Improving Out-of-Distribution Robustness via Selective Augmentationcited by 202
- 2022Discovering Invariant Rationales for Graph Neural Networkscited by 188
- 2021Fishr: Invariant Gradient Variances for Out-of-Distribution Generalizationcited by 186
- 2022Interpretable and Generalizable Graph Learning via Stochastic Attention Mechanismcited by 171
- 2022Handling Distribution Shifts on Graphs: An Invariance Perspectivecited by 164
- 2022Learning Causally Invariant Representations for Out-of-Distribution Generalization on Graphscited by 144
- 2022Causality Inspired Representation Learning for Domain Generalizationcited by 138
- 2021Does Invariant Risk Minimization Capture Invariance?cited by 116
- 2022Out-Of-Distribution Generalization on Graphs: A Surveycited by 97
- 2021Towards a Theoretical Framework of Out-of-Distribution Generalizationcited by 94
38 more
- 2021Heterogeneous Risk Minimizationcited by 90
- 2021Invariant Information Bottleneck for Domain Generalizationcited by 86
- 2021Causal Attention for Unbiased Visual Recognitioncited by 80
- 2021Can Subnetwork Structure be the Key to Out-of-Distribution Generalization?cited by 75
- 2022Graph Rationalization with Environment-based Augmentationscited by 70
- 2022ZIN: When and How to Learn Invariance Without Environment Partition?cited by 58
- 2022Let Invariant Rationale Discovery Inspire Graph Contrastive Learningcited by 57
- 2022DrugOOD: Out-of-Distribution (OOD) Dataset Curator and Benchmark for AI-aided Drug Discovery -- A Focus on Affinity Prediction Problems with Noise Annotationscited by 54
- 2022Out-of-distribution Generalization with Causal Invariant Transformationscited by 45
- 2023Model Agnostic Sample Reweighting for Out-of-Distribution Learningcited by 44
- 2024On Large Language Models' Hallucination with Regard to Known Factscited by 43
- 2021Out-of-distribution Generalization in the Presence of Nuisance-Induced Spurious Correlationscited by 41
- 2022On the Explainability of Natural Language Processing Deep Modelscited by 38
- 2022Unleashing the Power of Graph Data Augmentation on Covariate Distribution Shiftcited by 38
- 2021Understanding Interlocking Dynamics of Cooperative Rationalizationcited by 36
- 2022Can Rationalization Improve Robustness?cited by 36
- 2023Robust Generalization against Photon-Limited Corruptions via Worst-Case Sharpness Minimizationcited by 29
- 2021Out-of-distribution Prediction with Invariant Risk Minimization: The Limitation and An Effective Fixcited by 27
- 2022Interpretable Geometric Deep Learning via Learnable Randomness Injectioncited by 27
- 2024Time-Series Forecasting for Out-of-Distribution Generalization Using Invariant Learningcited by 27
- 2021Explain, Edit, and Understand: Rethinking User Study Design for Evaluating Model Explanationscited by 26
- 2023Environment-Aware Dynamic Graph Learning for Out-of-Distribution Generalizationcited by 26
- 2021What to Learn, and How: Toward Effective Learning from Rationalescited by 25
- 2023Multi-view Adversarial Discriminator: Mine the Non-causal Factors for Object Detection in Unseen Domainscited by 24
- 2023Rationalization for Explainable NLP: A Surveycited by 22
- 2023Data-centric Graph Learning: A Surveycited by 22
- 2021Kernelized Heterogeneous Risk Minimizationcited by 21
- 2023A Survey on the Robustness of Computer Vision Models against Common Corruptionscited by 21
- 2023D-Separation for Causal Self-Explanationcited by 21
- 2021Rationalization through Conceptscited by 19
- 2021SPECTRA: Sparse Structured Text Rationalizationcited by 19
- 2022Localized Adversarial Domain Generalizationcited by 19
- 2023Towards Trustworthy Explanation: On Causal Rationalizationcited by 19
- 2023Diversifying Spatial-Temporal Perception for Video Domain Generalizationcited by 19
- 2024Spectral Invariant Learning for Dynamic Graphs under Distribution Shiftscited by 19
- 2022FR: Folded Rationalization with a Unified Encodercited by 18
- 2021Learning Representations that Support Robust Transfer of Predictorscited by 17
- 2021Predict then Interpolate: A Simple Algorithm to Learn Stable Classifierscited by 16
50 most cited of 131 All 131 (JSON)
- 2026Robust Representation Learning through Explicit Environment Modeling28 Apr 2026
- 2026Adversarial Label Invariant Graph Data Augmentations for Out-of-Distribution Generalization9 Apr 2026
- 2026Causality-inspired Federated Learning for Dynamic Spatio-Temporal Graphs31 Mar 2026
- 2025Evolving Graph Learning for Out-of-Distribution Generalization in Non-stationary Environments4 Nov 2025
- 2025When LRP Diverges from Leave-One-Out in Transformers21 Oct 2025
- 2025Learnable Game-theoretic Policy Optimization for Data-centric Self-explanation Rationalization15 Oct 2025
- 2025Out-of-Distribution Graph Models Merging4 Jun 2025
- 2025Bridging Distribution Shift and AI Safety: Conceptual and Methodological Synergies28 May 2025
- 2025Adversarial Cooperative Rationalization: The Risk of Spurious Correlations in Even Clean Datasets4 May 2025
- 2025Breaking Free from MMI: A New Frontier in Rationalization by Probing Input Utilization8 Mar 2025
- 2025Why Domain Generalization Fail? A View of Necessity and Sufficiency15 Feb 2025
- 2025Generative Risk Minimization for Out-of-Distribution Generalization on Graphs11 Feb 2025
38 more
- 2025Enhancing Distribution and Label Consistency for Graph Out-of-Distribution Generalization7 Jan 2025
- 2025Boosting Explainability through Selective Rationalization in Pre-trained Language Models3 Jan 2025
- 2024Interlocking-free Selective Rationalization Through Genetic-based Learning13 Dec 2024
- 2024Invariant Shape Representation Learning For Image Classification19 Nov 2024
- 2024DISCO: DISCovering Overfittings as Causal Rules for Text Classification Models7 Nov 2024
- 2024Quantum Rationale-Aware Graph Contrastive Learning for Jet Discrimination3 Nov 2024
- 2024Is the MMI Criterion Necessary for Interpretability? Degenerating Non-causal Features to Plain Noise for Self-Rationalization8 Oct 2024
- 2024MARE: Multi-Aspect Rationale Extractor on Unsupervised Rationale Extraction4 Oct 2024
- 2024Navigating the Shortcut Maze: A Comprehensive Analysis of Shortcut Learning in Text Classification by Language Models26 Sep 2024
- 2024Enhancing Fairness through Reweighting: A Path to Attain the Sufficiency Rule26 Aug 2024
- 2024Adversarial Attack for Explanation Robustness of Rationalization Models20 Aug 2024
- 2024Leveraging Invariant Principle for Heterophilic Graph Structure Distribution Shifts18 Aug 2024
- 2024Time-Series Forecasting for Out-of-Distribution Generalization Using Invariant Learning13 Jun 2024
- 2024How Interpretable Are Interpretable Graph Neural Networks?12 Jun 2024
- 2024Automated Trustworthiness Testing for Machine Learning Classifiers7 Jun 2024
- 2024Position Debiasing Fine-Tuning for Causal Perception in Long-Term Dialogue4 Jun 2024
- 2024IENE: Identifying and Extrapolating the Node Environment for Out-of-Distribution Generalization on Graphs2 Jun 2024
- 2024Invariant Graph Representations for Continuous-Time Dynamic Graphs Under Distribution Shifts29 May 2024
- 2024Causal-aware Graph Neural Architecture Search under Distribution Shifts26 May 2024
- 2024MetaCoCo: A New Few-Shot Classification Benchmark with Spurious Correlation30 Apr 2024
- 2024A robust assessment for invariant representations7 Apr 2024
- 2024On Large Language Models' Hallucination with Regard to Known Facts29 Mar 2024
- 2024Graph Partial Label Learning with Potential Cause Discovering18 Mar 2024
- 2024Towards Faithful Explanations: Boosting Rationalization with Shortcuts Discovery12 Mar 2024
- 2024Cooperative Classification and Rationalization for Graph Generalization10 Mar 2024
- 2024Spectral Invariant Learning for Dynamic Graphs under Distribution Shifts8 Mar 2024
- 2024CaT-GNN: Enhancing Credit Card Fraud Detection via Causal Temporal Graph Neural Networks22 Feb 2024
- 2024HYPO: Hyperspherical Out-of-Distribution Generalization12 Feb 2024
- 2024InvariantOODG: Learning Invariant Features of Point Clouds for Out-of-Distribution Generalization8 Jan 2024
- 2023Diagnosing and Rectifying Fake OOD Invariance: A Restructured Causal Approach15 Dec 2023
- 2023Fine-grained Graph Rationalization13 Dec 2023
- 2023Invariant Representation via Decoupling Style and Spurious Features from Images11 Dec 2023
- 2023Enhancing the Rationale-Input Alignment for Self-explaining Rationalization7 Dec 2023
- 2023Class Distribution Shifts in Zero-Shot Learning: Learning Robust Representations30 Nov 2023
- 2023Out-of-Distribution Generalized Dynamic Graph Neural Network for Human Albumin Prediction27 Nov 2023
- 2023Environment-Aware Dynamic Graph Learning for Out-of-Distribution Generalization18 Nov 2023
- 2023You Only Forward Once: Prediction and Rationalization in A Single Forward Pass4 Nov 2023
- 2023Diversifying Spatial-Temporal Perception for Video Domain Generalization27 Oct 2023
50 newest of 131 All 131 (JSON)
- 2021Towards Out-Of-Distribution Generalization: A Surveycited by 436
- 2021Invariance Principle Meets Information Bottleneck for Out-of-Distribution Generalizationcited by 220
- 2022Improving Out-of-Distribution Robustness via Selective Augmentationcited by 202
- 2022Discovering Invariant Rationales for Graph Neural Networkscited by 188
- 2021Fishr: Invariant Gradient Variances for Out-of-Distribution Generalizationcited by 186
- 2022Interpretable and Generalizable Graph Learning via Stochastic Attention Mechanismcited by 171
- 2022Handling Distribution Shifts on Graphs: An Invariance Perspectivecited by 164
- 2022Learning Causally Invariant Representations for Out-of-Distribution Generalization on Graphscited by 144
- 2022Causality Inspired Representation Learning for Domain Generalizationcited by 138
- 2021Does Invariant Risk Minimization Capture Invariance?cited by 116
- 2022Out-Of-Distribution Generalization on Graphs: A Surveycited by 97
- 2021Towards a Theoretical Framework of Out-of-Distribution Generalizationcited by 94
12 most cited of 130 All 130 (JSON)
- 2026Robust Representation Learning through Explicit Environment Modeling28 Apr 2026
- 2026Adversarial Label Invariant Graph Data Augmentations for Out-of-Distribution Generalization9 Apr 2026
- 2026Causality-inspired Federated Learning for Dynamic Spatio-Temporal Graphs31 Mar 2026
- 2025Evolving Graph Learning for Out-of-Distribution Generalization in Non-stationary Environments4 Nov 2025
- 2025When LRP Diverges from Leave-One-Out in Transformers21 Oct 2025
- 2025Learnable Game-theoretic Policy Optimization for Data-centric Self-explanation Rationalization15 Oct 2025
- 2025Out-of-Distribution Graph Models Merging4 Jun 2025
- 2025Bridging Distribution Shift and AI Safety: Conceptual and Methodological Synergies28 May 2025
- 2025Adversarial Cooperative Rationalization: The Risk of Spurious Correlations in Even Clean Datasets4 May 2025
- 2025Breaking Free from MMI: A New Frontier in Rationalization by Probing Input Utilization8 Mar 2025
- 2025Why Domain Generalization Fail? A View of Necessity and Sufficiency15 Feb 2025
- 2025Generative Risk Minimization for Out-of-Distribution Generalization on Graphs11 Feb 2025
12 newest of 130 All 130 (JSON)