一种用于异构因果效应联邦学习的自适应核方法
机器学习
2023-01-03 v1 人工智能
统计方法学
机器学习
摘要
我们提出一种新的因果推断框架,用于在联邦设置下从多个分散数据源学习因果效应。我们引入一种自适应迁移算法,利用随机傅里叶特征将损失函数解耦为多个分量,每个分量与一个数据源相关联,从而学习数据源之间的相似性。数据源可能具有不同的分布;因果效应被独立且系统地纳入。所提方法通过迁移系数估计源间相似性,因此无需关于相似性度量的先验信息。异构因果效应可在源间不共享原始训练数据的情况下进行估计,从而将隐私泄露风险降至最低。我们还给出了极小极大下界,以评估从异质源学到的参数质量。实验表明,所提方法在具有相异分布的分散数据源上优于基线方法。
引用
@article{arxiv.2301.00346,
title = {An Adaptive Kernel Approach to Federated Learning of Heterogeneous Causal Effects},
author = {Thanh Vinh Vo and Arnab Bhattacharyya and Young Lee and Tze-Yun Leong},
journal= {arXiv preprint arXiv:2301.00346},
year = {2023}
}
备注
NeurIPS 2022