PICLE-PFN:首个面向动态策略的纵向因果效应基础模型

论文AI 评分 72/100arXiv cs.AI待复核
AI 聚合

本条为 AI 依据下方公开信源自动整理生成的摘要,不构成转载,可能存在偏差,请以原文为准。整理者:贝果科技 AI 资讯助手。

研究者提出 PICLE-PFN,用先验数据拟合网络做纵向因果效应估计,支持动态治疗策略。它把目标策略表示为上下文中的历史-决策示例,从而为多种治疗策略提供统一接口。模型在由纵向结构因果模型采样的合成数据上预训练,把因果效应估计摊销到不同数据集、策略与预测期上。给定新的观测数据集与目标策略,它一次前向即可直接预测条件平均潜在结果,无需任务特定训练或超参数调优。在合成与半合成纵向基准上,其 RMSE 与现有方法相当,不确定性校准明显更好,部署耗时仅为后者的一小部分。

全文梳理

AI 摘要依据下方信源原文自动整理,非原文转载

要解决的问题

纵向因果效应估计长期受两类困难牵制:随时间变化的混杂因素,以及治疗与混杂因素之间形成的反馈回路。材料指出,现有方法通常要为每个数据集单独拟合 nuisance 模型,其中不少还只针对预先指定的治疗序列或预测期设计,导致部署阶段计算耗时。PICLE-PFN 的出发点是把这套逐数据集、逐策略的准备工作摊销掉,让模型面对新数据与新策略时不必重新训练。

方法如何运作

PICLE-PFN 全称 Policy In-Context Longitudinal causal Effect estimation with Prior-data Fitted Networks,即用先验数据拟合网络做上下文内的纵向因果效应估计。它把目标治疗策略表达为上下文中的历史-决策示例,由此为一大类治疗策略提供统一输入接口,而不必为每种策略单独建模。训练数据来自纵向结构因果模型采样的合成数据,模型在这些数据上预训练,从而把估计能力摊销到不同数据集、策略与预测期上。使用时,给定一份新的观测数据集和目标策略,模型在一次前向传播中直接输出条件平均潜在结果(CAPO),不需要任务特定训练,也不需要超参数调优。

评测结果与边界

在合成与半合成纵向基准上,PICLE-PFN 的 RMSE 与那些为每个数据集单独训练并调优的现有估计器相当,不确定性校准则有明显改善,同时部署运行时间只占后者的一小部分。材料给出的对比对象是分别训练调优的常规估计器,比较维度包括误差、校准与运行耗时。需要留意的是,这些结论建立在合成与半合成基准之上,材料未说明真实观测数据上的表现,也未给出具体数值、参数量或开放范围。

为什么值得看

把基础模型与上下文学习引入纵向因果推断,免去逐数据集拟合与调参,值得关注因果建模落地效率的人一看。

数据集

信源1 家

  1. [1]arXiv cs.AI一手信源A Foundation Model for Policy In-Context Longitudinal Causal Effect Estimation

关键事实

  • PICLE-PFN 被描述为首个支持动态治疗策略的纵向因果效应估计基础模型[1]

  • 模型在由纵向结构因果模型采样的合成数据上预训练,将估计能力摊销到数据集、策略与预测期[1]

  • 给定新数据集与目标策略,模型一次前向预测条件平均潜在结果,无需任务特定训练或超参数调优[1]

  • 在合成与半合成纵向基准上,其 RMSE 与分别训练调优的现有估计器相当,不确定性校准明显更好,部署耗时更少[1]

相关 · 论文

本页内容由 AI 自动聚合公开信源生成,仅供了解行业动态参考,不构成任何投资或决策建议。如需引用请以原文出处为准。