Stable Learning via Sample Reweighting
Not on the map. Why?
Abstract (arXiv)
We consider the problem of learning linear prediction models with model misspecification bias. In such case, the collinearity among input variables may inflate the error of parameter estimation, resulting in instability of prediction results when training and test distributions do not match. In this paper we theoretically analyze this fundamental problem and propose a sample reweighting method that reduces collinearity among input variables. Our method can be seen as a pretreatment of data to improve the condition of design matrix, and it can then be combined with any standard learning method for parameter estimation and variable selection. Empirical studies on both simulation and real datasets demonstrate the effectiveness of our method in terms of more stable performance across different distributed data.
Key facts. Cited by 70 papers in the map (data through 14 Oct 2026). First posted on arXiv 28 Nov 2019.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2021: 8; 2022: 18; 2023: 18; 2024: 5; 2025: 14; 2026: 7.
70 papers.
Sort:
By area:
- 2021Towards Out-Of-Distribution Generalization: A Surveycited by 436
- 2021Deep Stable Learning for Out-Of-Distribution Generalizationcited by 153
- 2021OOD-GNN: Out-of-Distribution Generalized Graph Neural Networkcited by 78
- 2021Generalizing Graph Neural Networks on Out-Of-Distribution Graphscited by 71
- 2022Out-of-distribution Generalization with Causal Invariant Transformationscited by 45
- 2023MuggleMath: Assessing the Impact of Query and Response Augmentation on Math Reasoningcited by 45
- 2023Model Agnostic Sample Reweighting for Out-of-Distribution Learningcited by 44
- 2023Joint Learning of Label and Environment Causal Independence for Graph Out-of-Distribution Generalizationcited by 29
- 2022Debiased Graph Neural Networks with Agnostic Label Selection Biascited by 25
- 2024A Survey on Evaluation of Out-of-Distribution Generalizationcited by 21
- 2023Rethinking the Evaluation Protocol of Domain Generalizationcited by 20
- 2021A Theoretical Analysis on Independence-driven Importance Weighting for Covariate-shift Generalizationcited by 19
38 more
- 2022Exploring Optimal Substructure for Out-of-distribution Generalization via Feature-targeted Model Pruningcited by 17
- 2022Towards Domain Generalization in Object Detectioncited by 16
- 2022Label-Efficient Domain Generalization via Collaborative Exploration and Generalizationcited by 15
- 2021Distributionally Robust Learning with Stable Adversarial Trainingcited by 14
- 2022Stable Learning via Sparse Variable Independencecited by 13
- 2022ZooD: Exploiting Model Zoo for Out-of-Distribution Generalizationcited by 12
- 2022Instrumental Variables in Causal Inference and Machine Learning: A Surveycited by 10
- 2023ArCL: Enhancing Contrastive Learning with Augmentation-Robust Representationscited by 10
- 2023Graph Structure and Feature Extrapolation for Out-of-Distribution Generalizationcited by 10
- 2023SFP: Spurious Feature-targeted Pruning for Out-of-Distribution Generalizationcited by 9
- 2023Grounded Image Text Matching with Mismatched Relation Reasoningcited by 8
- 2021General Greedy De-bias Learningcited by 7
- 2021Understanding and Testing Generalization of Deep Networks on Out-of-Distribution Datacited by 5
- 2022Towards Class-agnostic Tracking Using Feature Decorrelation in Point Cloudscited by 5
- 2022RMT-Net: Reject-aware Multi-Task Network for Modeling Missing-not-at-random Data in Financial Credit Scoringcited by 5
- 2022PearNet: A Pearson Correlation-based Graph Attention Network for Sleep Stage Recognitioncited by 5
- 2023Mining Stable Preferences: Adaptive Modality Decorrelation for Multimedia Recommendationcited by 5
- 2022Decorrelate Irrelevant, Purify Relevant: Overcome Textual Spurious Correlations from a Feature Perspectivecited by 4
- 2022Regularization Penalty Optimization for Addressing Data Quality Variance in OoD Algorithmscited by 4
- 2023Quantifying Overfitting: Evaluating Neural Network Performance through Analysis of Null Spacecited by 4
- 2023Learning Generalizable Agents via Saliency-Guided Features Decorrelationcited by 4
- 2025Distribution Shift Aware Neural Tabular Learningcited by 4
- 2022Deep Stable Representation Learning on Electronic Health Recordscited by 3
- 2023Who Would be Interested in Services? An Entity Graph Learning System for User Targetingcited by 3
- 2025Causal invariant geographic network representations with feature and structural distribution shiftscited by 3
- 2022Time Will Change Things: An Empirical Study on Dynamic Language Understanding in Social Media Classificationcited by 2
- 2023Deep Stable Multi-Interest Learning for Out-of-distribution Sequential Recommendationcited by 2
- 2025Bridging Distribution Shift and AI Safety: Conceptual and Methodological Synergiescited by 2
- 2022EiHi Net: Out-of-Distribution Generalization Paradigmcited by 1
- 2023Invariant Random Forest: Tree-Based Model Solution for OOD Generalizationcited by 1
- 2024Stable Heterogeneous Treatment Effect Estimation across Out-of-Distribution Populationscited by 1
- 2024Establishing Causal Relationship Between Whole Slide Image Predictions and Diagnostic Evidence Subregions in Deep Learningcited by 1
- 2025Bayesian Cross-Modal Alignment Learning for Few-Shot Out-of-Distribution Generalizationcited by 1
- 2025Class-Conditional Distribution Balancing for Group Robust Classificationcited by 1
- 2025Stability Selection via Variable Decorrelationcited by 1
- 2025Decorrelated feature importance from local sample weightingcited by 1
- 2022Improving Multi-Interest Network with Stable Learningcited by 0
- 2022A Novel Plug-and-Play Approach for Adversarially Robust Generalizationcited by 0
50 most cited of 70 All 70 (JSON)
- 2026Parameter-Efficient Distributionally Robust Adaptation of Tabular Foundation Models under Subpopulation Shift1 Oct 2026
- 2026Decoupled Causal Discovery20 Sep 2026
- 2026PLSP (Pre-hoc Liminal Space Profiling): OOD Prediction over Detection -- An Anticipatory Approach for Machine Learning Model Reliability10 Sep 2026
- 2026StableRCA: Robust Graph-Agnostic Mechanism-Level Root Cause Analysis4 Jun 2026
- 20262D Stability Selection: Design Jittering for Doubly Stable Feature Selection4 May 2026
- 2026Environment-Conditioned Tail Reweighting for Invariant Learning under Mixed Shifts30 Jan 2026
- 2026Lightweight Edge Learning via Dataset Pruning19 Jan 2026
- 2025Generating Risky Samples with Conformity Constraints via Diffusion Models21 Dec 2025
- 2025Distribution Shift Aware Neural Tabular Learning27 Aug 2025
- 2025Decorrelated feature importance from local sample weighting8 Aug 2025
- 2025Incorporating Interventional Independence Improves Robustness against Interventional Distribution Shift7 Jul 2025
- 2025Rule Learning for Knowledge Graph Reasoning under Agnostic Distribution Shift7 Jul 2025
38 more
- 2025Learning task-specific predictive models for scientific computing4 Jun 2025
- 2025Data Heterogeneity Modeling for Trustworthy Machine Learning1 Jun 2025
- 2025Bridging Distribution Shift and AI Safety: Conceptual and Methodological Synergies28 May 2025
- 2025Stability Selection via Variable Decorrelation27 May 2025
- 2025Class-Conditional Distribution Balancing for Group Robust Classification24 Apr 2025
- 2025Bayesian Cross-Modal Alignment Learning for Few-Shot Out-of-Distribution Generalization13 Apr 2025
- 2025Causal invariant geographic network representations with feature and structural distribution shifts25 Mar 2025
- 2025Causally-informed Deep Learning towards Explainable and Generalizable Outcomes Prediction in Critical Care4 Feb 2025
- 2025Causal Learning for Heterogeneous Subgroups Based on Nonlinear Causal Kernel Clustering20 Jan 2025
- 2024DREAM: Domain-agnostic Reverse Engineering Attributes of Black-box Model8 Dec 2024
- 2024Establishing Causal Relationship Between Whole Slide Image Predictions and Diagnostic Evidence Subregions in Deep Learning24 Jul 2024
- 2024Stable Heterogeneous Treatment Effect Estimation across Out-of-Distribution Populations3 Jul 2024
- 2024IENE: Identifying and Extrapolating the Node Environment for Out-of-Distribution Generalization on Graphs2 Jun 2024
- 2024A Survey on Evaluation of Out-of-Distribution Generalization4 Mar 2024
- 2023FlexSSL : A Generic and Efficient Framework for Semi-Supervised Learning28 Dec 2023
- 2023GROOD: GRadient-Aware Out-of-Distribution Detection22 Dec 2023
- 2023Invariant Random Forest: Tree-Based Model Solution for OOD Generalization7 Dec 2023
- 2023MuggleMath: Assessing the Impact of Query and Response Augmentation on Math Reasoning9 Oct 2023
- 2023Learning Generalizable Agents via Saliency-Guided Features Decorrelation8 Oct 2023
- 2023Grounded Image Text Matching with Mismatched Relation Reasoning2 Aug 2023
- 2023DREAM: Domain-free Reverse Engineering Attributes of Black-box Model20 Jul 2023
- 2023Mining Stable Preferences: Adaptive Modality Decorrelation for Multimedia Recommendation25 Jun 2023
- 2023Graph Structure and Feature Extrapolation for Out-of-Distribution Generalization13 Jun 2023
- 2023Joint Learning of Label and Environment Causal Independence for Graph Out-of-Distribution Generalization1 Jun 2023
- 2023Quantifying Overfitting: Evaluating Neural Network Performance through Analysis of Null Space30 May 2023
- 2023Who Would be Interested in Services? An Entity Graph Learning System for User Targeting30 May 2023
- 2023Rethinking the Evaluation Protocol of Domain Generalization24 May 2023
- 2023SFP: Spurious Feature-targeted Pruning for Out-of-Distribution Generalization19 May 2023
- 2023Prediction with Incomplete Data under Agnostic Mask Distribution Shift18 May 2023
- 2023Deep Stable Multi-Interest Learning for Out-of-distribution Sequential Recommendation12 Apr 2023
- 2023ArCL: Enhancing Contrastive Learning with Augmentation-Robust Representations2 Mar 2023
- 2023Model Agnostic Sample Reweighting for Out-of-Distribution Learning24 Jan 2023
- 2022Exploring Optimal Substructure for Out-of-distribution Generalization via Feature-targeted Model Pruning19 Dec 2022
- 2022Instrumental Variables in Causal Inference and Machine Learning: A Survey12 Dec 2022
- 2022Stable Learning via Sparse Variable Independence2 Dec 2022
- 2022ZooD: Exploiting Model Zoo for Out-of-Distribution Generalization17 Oct 2022
- 2022Time Will Change Things: An Empirical Study on Dynamic Language Understanding in Social Media Classification6 Oct 2022
- 2022EiHi Net: Out-of-Distribution Generalization Paradigm29 Sep 2022
50 newest of 70 All 70 (JSON)
- 2021Towards Out-Of-Distribution Generalization: A Surveycited by 436
- 2021Deep Stable Learning for Out-Of-Distribution Generalizationcited by 153
- 2021OOD-GNN: Out-of-Distribution Generalized Graph Neural Networkcited by 78
- 2021Generalizing Graph Neural Networks on Out-Of-Distribution Graphscited by 71
- 2022Out-of-distribution Generalization with Causal Invariant Transformationscited by 45
- 2023Model Agnostic Sample Reweighting for Out-of-Distribution Learningcited by 44
- 2023Joint Learning of Label and Environment Causal Independence for Graph Out-of-Distribution Generalizationcited by 29
- 2022Debiased Graph Neural Networks with Agnostic Label Selection Biascited by 25
- 2024A Survey on Evaluation of Out-of-Distribution Generalizationcited by 21
- 2023Rethinking the Evaluation Protocol of Domain Generalizationcited by 20
- 2021A Theoretical Analysis on Independence-driven Importance Weighting for Covariate-shift Generalizationcited by 19
- 2022Exploring Optimal Substructure for Out-of-distribution Generalization via Feature-targeted Model Pruningcited by 17
12 most cited of 68 All 68 (JSON)
- 2026Parameter-Efficient Distributionally Robust Adaptation of Tabular Foundation Models under Subpopulation Shift1 Oct 2026
- 2026Decoupled Causal Discovery20 Sep 2026
- 2026PLSP (Pre-hoc Liminal Space Profiling): OOD Prediction over Detection -- An Anticipatory Approach for Machine Learning Model Reliability10 Sep 2026
- 2026StableRCA: Robust Graph-Agnostic Mechanism-Level Root Cause Analysis4 Jun 2026
- 20262D Stability Selection: Design Jittering for Doubly Stable Feature Selection4 May 2026
- 2026Environment-Conditioned Tail Reweighting for Invariant Learning under Mixed Shifts30 Jan 2026
- 2026Lightweight Edge Learning via Dataset Pruning19 Jan 2026
- 2025Distribution Shift Aware Neural Tabular Learning27 Aug 2025
- 2025Decorrelated feature importance from local sample weighting8 Aug 2025
- 2025Incorporating Interventional Independence Improves Robustness against Interventional Distribution Shift7 Jul 2025
- 2025Rule Learning for Knowledge Graph Reasoning under Agnostic Distribution Shift7 Jul 2025
- 2025Learning task-specific predictive models for scientific computing4 Jun 2025
12 newest of 68 All 68 (JSON)