A Doubly Robust Machine Learning Approach for Disentangling Treatment Effect Heterogeneity with Functional Outcomes

2026-02-11

AI Summary

Existing methods struggle to accurately estimate heterogeneous treatment effects for functional outcomes defined over continuous domains such as time or space, limiting personalized decision-making in complex settings. This work proposes the FOCaL framework, which introduces double robustness into functional causal inference for the first time. By combining functional regression modeling with pseudo-outcome reconstruction, FOCaL directly estimates the functional conditional average treatment effect (F-CATE). The approach integrates functional regression, doubly robust estimation, and a meta-learner architecture into an end-to-end causal learning pipeline. Extensive experiments on both synthetic and real-world functional datasets demonstrate that FOCaL significantly outperforms existing non-robust methods, offering a reliable tool for applications in personalized medicine and intelligent decision support.

Technology Category

• Machine Learning: Causal Learning
• Reasoning under Uncertainty: Causality
• Intelligent Robots: State Estimation

Application Category

• User Modeling, Personalization and Recommendation: Fairness-aware retrieval and ranking
• Search and Retrieval-Augmented AI: Web learning to rank, online learning, and counterfactual learning for ranking
• Graph Algorithms and Modeling for the Web: Algorithms and analysis for incomplete, noisy, or partially observed Web-related graphs

Abstract

Causal inference is paramount for understanding the effects of interventions, yet extracting personalized insights from increasingly complex data remains a significant challenge for modern machine learning. This is the case, in particular, when considering functional outcomes observed over a continuous domain (e.g., time, or space). Estimation of heterogeneous treatment effects, known as CATE, has emerged as a crucial tool for personalized decision-making, but existing meta-learning frameworks are largely limited to scalar outcomes, failing to provide satisfying results in scientific applications that leverage the rich, continuous information encoded in functional data. Here, we introduce FOCaL (Functional Outcome Causal Learning), a novel, doubly robust meta-learner specifically engineered to estimate a functional heterogeneous treatment effect (F-CATE). FOCaL integrates advanced functional regression techniques for both outcome modeling and functional pseudo-outcome reconstruction, thereby enabling the direct and robust estimation of F-CATE. We provide a rigorous theoretical derivation of FOCaL, demonstrate its performance and robustness compared to existing non-robust functional methods through comprehensive simulation studies, and illustrate its practical utility on diverse real-world functional datasets. FOCaL advances the capabilities of machine intelligence to infer nuanced, individualized causal effects from complex data, paving the way for more precise and trustworthy AI systems in personalized medicine, adaptive policy design, and fundamental scientific discovery.

Problem

• heterogeneous treatment effect
• functional outcomes
• causal inference
• CATE
• functional data

Innovation

• functional outcomes
• heterogeneous treatment effects
• doubly robust
• causal inference
• meta-learning