Smooth Multi-Policy Causal Effect Estimation in Longitudinal Settings.

TitleSmooth Multi-Policy Causal Effect Estimation in Longitudinal Settings.
Publication TypeJournal Article
Year of Publication2026
AuthorsChen W, Pan W, Gan K, Wang F
JournalProc Mach Learn Res
Volume306
Pagination16272-16294
Date Published2026
ISSN2640-3498
Abstract

Comparative evaluation of multiple dynamic treatment policies is essential for healthcare and policy decisions, yet conventional longitudinal causal inference methods estimate each in isolation, preventing information sharing across counterfactuals. We demonstrate that this separate estimation paradigm induces a structurally uncontrolled second-order bias, inflating finite-sample variance even after standard debiasing with longitudinal targeted maximum likelihood estimation (LTMLE). To address this, we propose a policy-aware reparameterization of Iterative Conditional Expectation (ICE) Q-functions that enables joint estimation through shared representations. We implement this approach in the Policy-Encoded Q Network (PEQ-Net), an architecture centered on a shared policy encoder. The encoder is trained using kernel mean embeddings, ensuring that the learned representation space reflects population-level policy dissimilarities. After applying an LTMLE correction step, we prove this design imposes a structural constraint on the second-order remainder, thereby stabilizing finite-sample variance. Experiments on semi-synthetic datasets demonstrate that PEQ-Net consistently outperforms existing ICE-based methods, achieving substantial reductions in root-mean-square error, particularly when evaluating closely related policies.

DOI10.13026/C2XW26
Alternate JournalProc Mach Learn Res
PubMed ID42866051
PubMed Central IDPMC13655700
Grant ListRF1 AG084178 / AG / NIA NIH HHS / United States
R01 AG080624 / AG / NIA NIH HHS / United States
R01 AG072449 / AG / NIA NIH HHS / United States
R01 NS140142 / NS / NINDS NIH HHS / United States
R01 AG076234 / AG / NIA NIH HHS / United States
RF1 AG072449 / AG / NIA NIH HHS / United States
R01 AG076448 / AG / NIA NIH HHS / United States
R01 AG080991 / AG / NIA NIH HHS / United States
Division: 
Institute of Artificial Intelligence for Digital Health
Category: 
Faculty Publication