arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2406.00535cs.LGstat.ME

面向时间维度反事实回归的因果对比学习

Causal Contrastive Learning for Counterfactual Regression Over Time

Paris-Saclay University(巴黎萨克雷大学);CentraleSupélec(中央高等电力学院);Saint-Gobain(圣戈班集团)

Mouad El Bouchattaoui, Myriam Tami, Benoit Lepetit, Paul-Henry Cournède

AI总结:

针对时间维度反事实回归的长期预测问题,提出结合RNN、对比预测编码与信息最大化准则的因果对比学习方法,在合成与真实数据上取得最优反事实估计效果,首次将对比预测编码引入因果推断。

摘要:

估计随时间变化的治疗效应在精准医疗、流行病学、经济学和营销学等多个领域具有重要意义。本文提出了一种独特的时间维度反事实回归方法,重点关注长期预测。与Causal Transformer(因果Transformer)等现有模型不同,本方法凸显了将RNN(循环神经网络)用于长期预测的有效性,并辅以Contrastive Predictive Coding(CPC,对比预测编码)与Information Maximization(InfoMax,信息最大化)。出于效率考量,我们无需使用计算成本高昂的Transformer。借助CPC,本方法可在存在时变混杂因素的情况下捕捉长期依赖关系。值得注意的是,近期的模型忽略了可逆表示的重要性,损害了识别假设。为解决这一问题,我们采用InfoMax准则,最大化序列数据与其表示之间互信息的下界。本方法在合成数据与真实世界数据上均取得了当前最优的反事实估计结果,是首个将Contrastive Predictive Encoding(对比预测编码)应用于因果推断的研究。

英文摘要:

Estimating treatment effects over time holds significance in various domains, including precision medicine, epidemiology, economy, and marketing. This paper introduces a unique approach to counterfactual regression over time, emphasizing long-term predictions. Distinguishing itself from existing models like Causal Transformer, our approach highlights the efficacy of employing RNNs for long-term forecasting, complemented by Contrastive Predictive Coding (CPC) and Information Maximization (InfoMax). Emphasizing efficiency, we avoid the need for computationally expensive transformers. Leveraging CPC, our method captures long-term dependencies in the presence of time-varying confounders. Notably, recent models have disregarded the importance of invertible representation, compromising identification assumptions. To remedy this, we employ the InfoMax principle, maximizing a lower bound of mutual information between sequence data and its representation. Our method achieves state-of-the-art counterfactual estimation results using both synthetic and real-world data, marking the pioneering incorporation of Contrastive Predictive Encoding in causal inference.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑