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

针对模型投毒攻击的鲁棒联邦对比推荐系统

Robust Federated Contrastive Recommender System against Model Poisoning Attack

Wei Yuan, Chaoqun Yang, Liang Qu, Guanhua Ye, Quoc Viet Hung Nguyen, Hongzhi Yin

AI总结:

本文提出 CL4FedRec 对比学习框架,利用嵌入增强缓解联邦推荐中的数据稀疏问题,并针对其加剧投毒攻击脆弱性的现象,进一步提出 rCL4FedRec 以提升模型性能与鲁棒性。

摘要:

联邦推荐系统(FedRecs)因其隐私保护优势近年来受到越来越多的关注。然而,当前 FedRecs 的去中心化和开放性特征带来了两个困境。第一,由于每个客户端的设备端数据高度稀疏,FedRecs 的性能受到影响。第二,系统鲁棒性因恶意用户发起的模型投毒攻击而受到削弱。在本文中,我们提出了一种新颖的对比学习框架,旨在通过嵌入增强充分利用客户端的稀疏数据,称为 CL4FedRec。与以往 FedRecs 中需要客户端共享私有参数的对比学习方法不同,我们的 CL4FedRec 符合基本的 FedRec 学习协议,确保与大多数现有 FedRec 实现兼容。随后,我们通过将配备 CL4FedRec 的 FedRecs 置于若干最先进的模型投毒攻击下,评估其鲁棒性。令人惊讶的是,我们的观察结果表明,对比学习往往会加剧 FedRecs 对这些攻击的脆弱性。这归因于增强的嵌入均匀性,使得被污染的目标物品嵌入容易接近热门物品。基于这一发现,我们提出了一种增强且鲁棒的 CL4FedRec 版本(rCL4FedRec),通过引入正则化器来保持不同流行度物品嵌入之间的距离。在四个常用推荐数据集上进行的大量实验表明,CL4FedRec 显著提升了模型性能以及 FedRecs 的鲁棒性。

英文摘要:

Federated Recommender Systems (FedRecs) have garnered increasing attention recently, thanks to their privacy-preserving benefits. However, the decentralized and open characteristics of current FedRecs present two dilemmas. First, the performance of FedRecs is compromised due to highly sparse on-device data for each client. Second, the system's robustness is undermined by the vulnerability to model poisoning attacks launched by malicious users. In this paper, we introduce a novel contrastive learning framework designed to fully leverage the client's sparse data through embedding augmentation, referred to as CL4FedRec. Unlike previous contrastive learning approaches in FedRecs that necessitate clients to share their private parameters, our CL4FedRec aligns with the basic FedRec learning protocol, ensuring compatibility with most existing FedRec implementations. We then evaluate the robustness of FedRecs equipped with CL4FedRec by subjecting it to several state-of-the-art model poisoning attacks. Surprisingly, our observations reveal that contrastive learning tends to exacerbate the vulnerability of FedRecs to these attacks. This is attributed to the enhanced embedding uniformity, making the polluted target item embedding easily proximate to popular items. Based on this insight, we propose an enhanced and robust version of CL4FedRec (rCL4FedRec) by introducing a regularizer to maintain the distance among item embeddings with different popularity levels. Extensive experiments conducted on four commonly used recommendation datasets demonstrate that CL4FedRec significantly enhances both the model's performance and the robustness of FedRecs.

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

↑