arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2401.06263cs.LG
收藏

FedTabDiff:面向合成混合类型表格数据生成的扩散概率模型联邦学习

FedTabDiff: Federated Learning of Diffusion Probabilistic Models for Synthetic Mixed-Type Tabular Data Generation

University of St.Gallen(圣加仑大学);Deutsche Bundesbank(德意志联邦银行);International Computer Science Institute (ICSI)(国际计算机科学研究所)

Timur Sattarov, Marco Schreyer, Damian Borth

AI总结:

本文提出FedTabDiff框架,将去噪扩散概率模型扩展到联邦学习场景,在无需集中访问原始数据的情况下,通过同步更新和加权平均实现多实体协作训练,生成高保真、高实用性且保护隐私的混合类型表格数据。

摘要:

现实感的合成表格数据生成在保护隐私方面面临重大挑战,尤其是在处理金融和医疗等领域的敏感信息时。本文提出了联邦表格扩散(FedTabDiff),用于在不集中访问原始表格数据集的情况下生成高保真混合类型表格数据。利用去噪扩散概率模型(DDPM)的优势,我们的方法解决了表格数据中的固有复杂性,如混合属性类型和隐式关系。更为关键的是,FedTabDiff实现了一种去中心化学习方案,允许多个实体在尊重数据隐私和本地性的前提下协作训练生成模型。我们将DDPM扩展到用于表格数据生成的联邦设置中,包括同步更新方案和用于有效模型聚合的加权平均。在真实金融和医疗数据集上的实验评估证明了该框架能够生成保持高保真度、实用性、隐私性和覆盖率的合成数据。

英文摘要:

Realistic synthetic tabular data generation encounters significant challenges in preserving privacy, especially when dealing with sensitive information in domains like finance and healthcare. In this paper, we introduce \textit{Federated Tabular Diffusion} (FedTabDiff) for generating high-fidelity mixed-type tabular data without centralized access to the original tabular datasets. Leveraging the strengths of \textit{Denoising Diffusion Probabilistic Models} (DDPMs), our approach addresses the inherent complexities in tabular data, such as mixed attribute types and implicit relationships. More critically, FedTabDiff realizes a decentralized learning scheme that permits multiple entities to collaboratively train a generative model while respecting data privacy and locality. We extend DDPMs into the federated setting for tabular data generation, which includes a synchronous update scheme and weighted averaging for effective model aggregation. Experimental evaluations on real-world financial and medical datasets attest to the framework's capability to produce synthetic data that maintains high fidelity, utility, privacy, and coverage.

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

↑