跳到正文
arXiv · ML· Dongze Wu, David I. Inouye, Yao Xie·· 9 小时前AI 评分23

PO-Flow:用于潜在结果和反事实生成的生成流框架

Generative Flows for Potential Outcomes and Counterfactuals

AI 导读

PO-Flow 是一种连续归一化流框架,通过可逆传输映射表示处理特定的结果分布,支持潜在结果生成、条件平均处理效应估计及反事实预测。 该框架在损失函数中引入交叉拟合逆概率加权,以解决观察性数据中的处理不平衡问题。 实验表明,在 4 个标量基准测试中,PO-Flow 取得了最低的 PEHE 和潜在结果 RMSE,并在 35 个多元比较中的 26 个达到最低平均得分。

来源:arXiv · ML · arxiv.org