mlsanityData to 14 Oct 2026

Learning Counterfactual Outcomes Under Rank Preservation

arXiv 2502.06398 ↗First posted 10 Feb 2025cs.LG stat.MLcited by 0

Not on the map. Why?

Abstract (arXiv)

Counterfactual inference aims to estimate the counterfactual outcome at the individual level given knowledge of an observed treatment and the factual outcome, with broad applications in fields such as epidemiology, econometrics, and management science. Previous methods rely on a known structural causal model (SCM) or assume the homogeneity of the exogenous variable and strict monotonicity between the outcome and exogenous variable. In this paper, we propose a principled approach for identifying and estimating the counterfactual outcome. We first introduce a simple and intuitive rank preservation assumption to identify the counterfactual outcome without relying on a known structural causal model. Building on this, we propose a novel ideal loss for theoretically unbiased learning of the counterfactual outcome and further develop a kernel-based estimator for its empirical estimation. Our theoretical analysis shows that the rank preservation assumption is not stronger than the homogeneity and strict monotonicity assumptions, and shows that the proposed ideal loss is convex, and the proposed estimator is unbiased. Extensive semi-synthetic and real-world experiments are conducted to demonstrate the effectiveness of the proposed method.

Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 64 of its 118 references that are in the map. First posted on arXiv 10 Feb 2025.

Builds on

64 of 118 references.

References · 64

  1. 2017Counterfactual Fairnesscited by 804
  2. 2015Estimation and Inference of Heterogeneous Treatment Effects using Random Forestscited by 600
  3. 2017Meta-learners for Estimating Heterogeneous Treatment Effects using Machine Learningcited by 478
  4. 2017Causal Effect Inference with Deep Latent-Variable Modelscited by 410
  5. 2016Learning Representations for Counterfactual Inferencecited by 329
  6. 2019Adapting Neural Networks for the Estimation of Treatment Effectscited by 287
  7. 2013Causal Discovery with Continuous Additive Noise Modelscited by 255
  8. 2020Conformal Inference of Counterfactuals and Individual Treatment Effectscited by 151
  9. 2019Double Reinforcement Learning for Efficient Off-Policy Evaluation in Markov Decision Processescited by 119
  10. 2013Probabilities of Causation: Bounds and Identificationcited by 90
  11. 2023Fast Model Debias with Machine Unlearningcited by 72
  12. 2013Counterfactual Probabilities: Computational Methods, Bounds and Applicationscited by 67
38 more
  1. 2019Causal structure based root cause analysis of outlierscited by 65
  2. 2020Counterfactual Credit Assignment in Model-Free Reinforcement Learningcited by 64
  3. 2021Partial Counterfactual Identification from Observational and Experimental Datacited by 64
  4. 2021Sensitivity Analysis of Individual Treatment Effects: A Robust Conformal Inference Approachcited by 62
  5. 2020Sample-Efficient Reinforcement Learning via Counterfactual-Based Data Augmentationcited by 59
  6. 2018Perfect Match: A Simple Method for Learning Representations For Counterfactual Inference With Neural Networkscited by 56
  7. 2023High Fidelity Image Counterfactuals with Probabilistic Causal Modelscited by 53
  8. 2023LANCE: Stress-testing Visual Models by Generating Language-guided Counterfactual Imagescited by 46
  9. 2023Robust Counterfactual Explanations for Neural Networks With Probabilistic Guaranteescited by 45
  10. 2023Counterfactual Identifiability of Bijective Causal Modelscited by 43
  11. 2021Nested Counterfactual Identification from Arbitrary Surrogate Experimentscited by 38
  12. 2023Counterfactual Conservative Q Learning for Offline Multi-agent Reinforcement Learningcited by 37
  13. 2023Optimal Transport for Treatment Effect Estimationcited by 32
  14. 2022Counterfactual harmcited by 29
  15. 2022Causal Proxy Models for Concept-Based Model Explanationscited by 29
  16. 2022On the Opportunity of Causal Learning in Recommendation Systems: Foundation, Estimation, Prediction and Challengescited by 28
  17. 2023GLOBE-CE: A Translation-Based Approach for Global Counterfactual Explanationscited by 27
  18. 2022StableDR: Stabilized Doubly Robust Learning for Recommendation on Data Missing Not at Randomcited by 24
  19. 2023Self-Interpretable Time Series Prediction with Counterfactual Explanationscited by 24
  20. 2020Principal Fairness for Human and Algorithmic Decision-Makingcited by 20
  21. 2020Probabilistic Reasoning across the Causal Hierarchycited by 19
  22. 2023Causal Context Connects Counterfactual Fairness to Robust Prediction and Group Fairnesscited by 17
  23. 2022DESCN: Deep Entire Space Cross Networks for Individual Treatment Effect Estimationcited by 16
  24. 2023Partial Counterfactual Identification of Continuous Outcomes with a Curvature Sensitivity Modelcited by 14
  25. 2023Data Augmentations for Improved (Large) Language Model Generalizationcited by 14
  26. 2023Finding Counterfactually Optimal Action Sequences in Continuous State Spacescited by 13
  27. 2023Would I have gotten that reward? Long-term credit assignment by counterfactual contribution analysiscited by 13
  28. 2023Disentangled Counterfactual Learning for Physical Audiovisual Commonsense Reasoningcited by 13
  29. 2022Adversarial Counterfactual Environment Model Learningcited by 12
  30. 2023Counterfactual Evaluation of Peer-Review Assignment Policiescited by 11
  31. 2022Counterfactual Analysis in Dynamic Latent State Modelscited by 10
  32. 2022On counterfactual inference with unobserved confoundingcited by 10
  33. 2023ACAT: Adversarial Counterfactual Attention for Classification and Detection in Medical Imagingcited by 9
  34. 2023Fairness Aware Counterfactuals for Subgroupscited by 9
  35. 2023Counterfactually Fair Representationcited by 9
  36. 2016Decomposing Treatment Effect Variationcited by 8
  37. 2020Deriving Bounds and Inequality Constraints Using LogicalRelations Among Counterfactualscited by 8
  38. 2023The Utility of "Even if..." Semifactual Explanation to Optimise Positive Outcomescited by 8

50 most cited of 64 All 64 (JSON)

Built on by

Not cited yet.