arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2402.02672stat.MEcs.CRcs.LG
收藏

基于分布式机密数据的条件平均处理效应估计

Estimation of conditional average treatment effects on distributed confidential data

Center for Artificial Intelligence Research(人工智能研究中心);Tsukuba Institute for Advanced Research(茨鲁巴高级研究机构);University of Tsukuba(茨鲁巴大学);Graduate School of Science and Technology(科学与技术研究生院);Faculty of Engineering, Information and Systems(工程、信息与系统学部)

Yuji Kawamata, Ryoki Motai, Yukihiko Okada, Akira Imakura, Tetsuya Sakurai

AI总结:

针对多方机密数据难以集中的问题,提出 data collaboration double machine learning,利用隐私保护融合数据估计并检验半参数 CATE 模型,且支持跨时间、跨方协作。

摘要:

条件平均处理效应(CATEs)的估计在许多科学领域都是一个重要课题。如果将分布在多方的数据集中起来,就可以高精度估计 CATE。然而,由于保密或隐私方面的顾虑,很难汇总此类数据。为解决这一问题,我们提出了 data collaboration double machine learning(数据协作双重机器学习),这是一种利用由分布式来源构建的隐私保护融合数据来估计 CATE 模型的方法,并通过模拟评估其性能。我们作出三项主要贡献。第一,我们的方法无需在分布式数据上进行迭代通信,即可实现半参数 CATE 模型的估计与检验,并且与参数方法相比,对模型误设具有稳健性。第二,它通过积累知识库,实现跨不同时间点和参与方的协作估计。第三,在使用合成、半合成和真实世界数据集的模拟中,我们的方法表现与现有方法相当或更好。

英文摘要:

The estimation of conditional average treatment effects (CATEs) is an important topic in many scientific fields. CATEs can be estimated with high accuracy if data distributed across multiple parties are centralized. However, it is difficult to aggregate such data owing to confidentiality or privacy concerns. To address this issue, we propose data collaboration double machine learning, a method for estimating CATE models using privacy-preserving fusion data constructed from distributed sources, and evaluate its performance through simulations. We make three main contributions. First, our method enables estimation and testing of semi-parametric CATE models without iterative communication on distributed data, providing robustness to model mis-specification compared to parametric approaches. Second, it enables collaborative estimation across different time points and parties by accumulating a knowledge base. Third, our method performs as well as or better than existing methods in simulations using synthetic, semi-synthetic, and real-world datasets.

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

↑