arXiv Machine Learning By Nikita Dhawan, Arnav Paruthi, Andrew Kim, Lovedeep Gondara, Jekaterina Novikova, Chris J. Maddison

Supervised Learning Objectives for High-Dimensional Treatment Effects

Read the original on arXiv Machine Learning →

The paper proposes supervised learning objectives for estimating causal effects of high‑dimensional treatments, such as text or continuous variables, by recasting causal inference as a learning problem under realizability and no unobserved confounding. It shows that causal error can be decomposed into moment‑balancing errors of increasing order and designs objectives that directly improve causal estimation by exploiting the structure of conditional outcomes. The authors also demonstrate how to project high‑dimensional treatment effects onto coarser attributes, enabling a single model to answer multiple causal questions without extra attribute‑specific training, and validate their approach empirically across continuous, discrete, and text treatments.

Machine-generated by The Flow from the publisher's headline and feed description — not written or checked by a human. The full article lives at arXiv Machine Learning.

arXiv Machine Learning
Sep 4

Causal Foundation Models

Causal Foundation Models (CFMs) are pretrained neural networks designed to estimate causal quantities—such as the average treatment effect—across new datasets using in‑context learning, eliminating the need for bespoke pipelines or model updates. The paper introduces CFMs, reviews foundational concepts in causal inference and machine learning, and provides practical code examples and Jupyter notebooks to illustrate their application.

By Christopher Stith, Hossein Rahmani, Jesse C. Cresswell
arXiv Machine Learning
Aug 4

Causal Inference with Unstructured Treatments

arXiv:2608. 00657v1 Announce Type: cross Abstract: Causal inference usually concerns a scalar treatment, yet in many problems the treatment is unstructured: a text, an image, or a sequence of clinical decisions.

By Kevin Christian Wibisono, Yixin Wang