arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2008.07298cs.CRcs.DCcs.LGstat.ML
收藏

WAFFLE:联邦学习中的水印技术

WAFFLE: Watermarking in Federated Learning

Aalto University(阿尔托大学);F-Secure Corporation(F-Secure公司);University of Waterloo(滑铁卢大学)

Buse Gul Atli, Yuxi Xia, Samuel Marchal, N. Asokan

摘要:

联邦学习是一种分布式学习技术,其中在本地训练数据驻留的客户端设备上训练机器学习模型。训练通过中央服务器进行协调,中央服务器通常由结果模型的预期所有者控制。通过避免将训练数据传输到中央服务器,联邦学习提高了隐私和效率。但它增加了客户端窃取模型的风险,因为生成的模型在每个客户端设备上都可用。即使用于本地训练的应用软件可能试图阻止对模型的直接存取,恶意客户端也可以通过对应用软件进行逆向工程来绕过任何这样的限制。水印是一种众所周知的防止模型盗窃的方法,它为模型所有者提供了证明其模型所有权的手段。最近的几种深度神经网络(DNN)水印技术使用后门:用额外的错误标记数据训练模型。后门需要完全访问训练数据和控制训练过程。当一方以集中方式训练模型时,这是可行的,但在训练过程和训练数据分布在若干客户端设备之间的联邦学习设置中,这是不可行的。在本文中,我们提出了WAFFLE,第一种使用联邦学习训练的DNN模型的水印方法。它在每次将局部模型聚合到全局模型之后在服务器处引入再训练步骤。我们表明,WAFFLE有效地嵌入到模型中的弹性水印,只会导致测试精度(-0.17%)可以忽略不计的退化,并且不需要访问训练数据。我们还介绍了一种新的技术来产生后门用作水印。它优于以前的技术,不需要通信,计算开销低(+3.2%)。

英文摘要:

Federated learning is a distributed learning technique where machine learning models are trained on client devices in which the local training data resides. The training is coordinated via a central server which is, typically, controlled by the intended owner of the resulting model. By avoiding the need to transport the training data to the central server, federated learning improves privacy and efficiency. But it raises the risk of model theft by clients because the resulting model is available on every client device. Even if the application software used for local training may attempt to prevent direct access to the model, a malicious client may bypass any such restrictions by reverse engineering the application software. Watermarking is a well-known deterrence method against model theft by providing the means for model owners to demonstrate ownership of their models. Several recent deep neural network (DNN) watermarking techniques use backdooring: training the models with additional mislabeled data. Backdooring requires full access to the training data and control of the training process. This is feasible when a single party trains the model in a centralized manner, but not in a federated learning setting where the training process and training data are distributed among several client devices. In this paper, we present WAFFLE, the first approach to watermark DNN models trained using federated learning. It introduces a retraining step at the server after each aggregation of local models into the global model. We show that WAFFLE efficiently embeds a resilient watermark into models incurring only negligible degradation in test accuracy (-0.17%), and does not require access to training data. We also introduce a novel technique to generate the backdoor used as a watermark. It outperforms prior techniques, imposing no communication, and low computational (+3.2%) overhead.

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

↑