mlsanityData to 7 Oct 2026

Conformal Diffusion Models for Individual Treatment Effect Estimation and Inference

arXiv 2408.01582 ↗First posted 2 Aug 2024stat.ML cs.AI cs.LGcited by 1

On the map: Diffusion models → Diffusion foundations and sampling

Abstract (arXiv)

Estimating treatment effects from observational data is of central interest across numerous application domains. Individual treatment effect offers the most granular measure of treatment effect on an individual level, and is the most useful to facilitate personalized care. However, its estimation and inference remain underdeveloped due to several challenges. In this article, we propose a novel conformal diffusion model-based approach that addresses those intricate challenges. We integrate the highly flexible diffusion modeling, the model-free statistical inference paradigm of conformal inference, along with propensity score and covariate local approximation that tackle distributional shifts. We unbiasedly estimate the distributions of potential outcomes for individual treatment effect, construct an informative confidence interval, and establish rigorous theoretical guarantees. We demonstrate the competitive performance of the proposed method over existing solutions through extensive numerical studies.

Key facts. Cited by 1 paper in the map (data through 7 Oct 2026). Placed in Diffusion foundations and sampling, one of 14 directions in Diffusion models. Builds on 26 of its 63 references that are in the map. First posted on arXiv 2 Aug 2024.

Builds on

26 of 63 references.

Closest ancestors · 6

Same direction, newest first

  1. 2021Variational Diffusion Models1 Jul
  2. 2021Diffusion Models Beat GANs on Image Synthesis11 May
  3. 2021Improved Denoising Diffusion Probabilistic Models18 Feb
  4. 2020Score-Based Generative Modeling through Stochastic Differential Equations26 Nov
  5. 2020Denoising Diffusion Implicit Models6 Oct
  6. 2020Denoising Diffusion Probabilistic Models19 Jun

Other references · 20

  1. 2015Deep Unsupervised Learning using Nonequilibrium Thermodynamicscited by 7,005
  2. 2016Distribution-Free Predictive Inference For Regressioncited by 657
  3. 2019Conformal Prediction Under Covariate Shiftcited by 511
  4. 2017Meta-learners for Estimating Heterogeneous Treatment Effects using Machine Learningcited by 475
  5. 2017Causal Effect Inference with Deep Latent-Variable Modelscited by 408
  6. 2020Classification with Valid and Adaptive Coveragecited by 390
  7. 2016Learning Representations for Counterfactual Inferencecited by 327
  8. 2021Crystal Diffusion Variational Autoencoder for Periodic Material Generationcited by 268
  9. 2017Bayesian Inference of Individualized Treatment Effects using Multi-task Gaussian Processescited by 156
  10. 2020Conformal Inference of Counterfactuals and Individual Treatment Effectscited by 150
  11. 2021Localized Conformal Prediction: A Generalized Inference Framework for Conformal Predictioncited by 112
  12. 2019Distributional conformal predictioncited by 104
8 more
  1. 2021Conformal Prediction using Conditional Histogramscited by 77
  2. 2022Probabilistic Conformal Prediction Using Conditional Random Samplescited by 47
  3. 2023Conformal prediction with local weights: randomization enables local guaranteescited by 47
  4. 2020Exchangeability, Conformal Prediction, and Rank Testscited by 40
  5. 2023Conformal Meta-learners for Predictive Inference of Individual Treatment Effectscited by 23
  6. 2020Conformal prediction intervals for the individual treatment effectcited by 10
  7. 2016Conformalized Kernel Ridge Regressioncited by 6
  8. 2020Kernel Assisted Learning for Personalized Dose Findingcited by 3

Built on by

Cited by year

Citing papers per year: 2025: 1.

1 paper.

  1. 2025A Systematic Review of Conformal Inference Procedures for Treatment Effect Estimation: Methods and Challengescited by 1

Similar, not cited

Close in wording, no citation either way

  1. 2026Causal Diffusion Models for Counterfactual Outcome Distributions in Longitudinal Data
  2. 2025Deep Disentangled Representation Network for Treatment Effect Estimation
  3. 2024DiffPO: A causal diffusion model for learning distributions of potential outcomes
  4. 2025Technical Report: Facilitating the Adoption of Causal Inference Methods Through LLM-Empowered Co-Pilot
  5. 2025Flow-based Generative Modeling of Potential Outcomes and Counterfactuals
  6. 2023Counterfactual Generative Models for Time-Varying Treatments
  7. 2026Debiased Counterfactual Generation via Flow Matching from Observations
  8. 2025A Generative Framework for Causal Estimation via Importance-Weighted Diffusion Distillation