Skip to main navigation Skip to search Skip to main content

Adversary-Free Counterfactual Prediction via Information-Regularized Representations

Research output: Chapters, Conference Papers, Creative and Literary WorksRGC 32 - Refereed conference paper (with host publication)peer-review

Abstract

We study counterfactual prediction under assignment bias and propose a mathematically grounded, information-theoretic approach that removes treatment–covariate dependence without adversarial training. Starting from a bound that links the counterfactual–factual risk gap to mutual information, we learn a stochastic representation Z that is predictive of outcomes while minimizing I(Z; T). We derive a tractable variational objective that upper-bounds the information term and couples it with a supervised decoder, yielding a stable, provably motivated training criterion. The framework extends naturally to dynamic settings by applying the information penalty to sequential representations at each decision time. We evaluate the method on controlled numerical simulations and a real-world clinical dataset, comparing against recent state-of-the-art balancing, reweighting, and adversarial baselines. Across metrics of likelihood, counterfactual error, and policy evaluation, our approach performs favorably while avoiding the training instabilities and tuning burden of adversarial schemes. © 2026 by the author(s).
Original languageEnglish
Title of host publicationProceedings of the 29th International Conference on Artificial Intelligence and Statistics (AISTATS) 2026
PublisherML Research Press
Number of pages18
Publication statusOnline published - 3 Feb 2026
Event29th International Conference on Artificial Intelligence and Statistics (AISTATS 2026) - Tangier, Morocco
Duration: 2 May 20265 May 2026
https://virtual.aistats.org/Conferences/2026

Publication series

NameProceedings of Machine Learning Research
Volume300
ISSN (Print)2640-3498

Conference

Conference29th International Conference on Artificial Intelligence and Statistics (AISTATS 2026)
PlaceMorocco
CityTangier
Period2/05/265/05/26
Internet address

Bibliographical note

Information for this record is supplemented by the author(s) concerned.

Research Keywords

  • counterfactual prediction
  • Casual Inference

Fingerprint

Dive into the research topics of 'Adversary-Free Counterfactual Prediction via Information-Regularized Representations'. Together they form a unique fingerprint.

Cite this