This paper presents a groundbreaking approach to causal inference by integrating continuous normalizing flows (CNFs) with parametric submodels, enhancing their geometric sensitivity and improving upon traditional Targeted Maximum Likelihood Estimation (TMLE). Our method employs CNFs to refine TMLE, optimizing the Cram\'er-Rao bound and transitioning from a predefined distribution $p_0$ to a data-driven distribution $p_1$. We innovate further by embedding Wasserstein gradient flows within Fokker-Planck equations, thus imposing geometric structures that boost the robustness of CNFs, particularly in optimal transport theory. Our approach addresses the disparity between sample and population distributions, a critical factor in parameter estimation bias. We leverage optimal transport and Wasserstein gradient flows to develop causal inference methodologies with minimal variance in finite-sample settings, outperforming traditional methods like TMLE and AIPW. This novel framework, centered on Wasserstein gradient flows, minimizes variance in efficient influence functions under distribution $p_t$. Preliminary experiments showcase our method's superiority, yielding lower mean-squared errors compared to standard flows, thereby demonstrating the potential of geometry-aware normalizing Wasserstein flows in advancing statistical modeling and inference.

翻译：本文提出了一种突破性的因果推断方法，通过将连续归一化流（CNFs）与参数子模型融合，增强了其几何敏感性，并改进了传统的目标最大似然估计（TMLE）。我们的方法利用CNFs优化TMLE，以优化Cramér-Rao界为目标，实现了从预设分布$p_0$到数据驱动分布$p_1$的转换。我们进一步创新性地将Wasserstein梯度流嵌入Fokker-Planck方程中，从而引入几何结构以提升CNFs的鲁棒性，特别是在最优输运理论中。该方法解决了样本分布与总体分布之间的差异问题，这是参数估计偏差的关键因素。我们利用最优输运和Wasserstein梯度流开发了在有限样本环境下具有最小方差的因果推断方法，性能优于TMLE和AIPW等传统方法。这一以Wasserstein梯度流为核心的新型框架，能够在分布$p_t$下最小化有效影响函数的方差。初步实验表明，我们的方法具有优越性，与标准流相比获得了更低的均方误差，从而展示了几何感知归一化Wasserstein流在推进统计建模与推断方面的潜力。