RegionFed 解决零售异构环境联邦学习个性化难题

论文AI 评分 65/100arXiv cs.AI待复核
AI 聚合

本条为 AI 依据下方公开信源自动整理生成的摘要,不构成转载,可能存在偏差,请以原文为准。整理者:贝果科技 AI 资讯助手。

RegionFed 是一种架构鲁棒的联邦学习框架,通过在梯度层面而非参数层面操作,解决了现有个性化方法在 Transformer 模型上因嵌入绑定和 LayerNorm 交互导致的性能崩溃问题。该方法利用区域与全局梯度的 L2 冲突作为统一信号,诊断异质性并路由至最经济的个性化策略。实验显示,RegionFed-Meta 在 Amazon ESCI 等数据集上达到 92.27% 的准确率,与侵犯隐私的集中式上限差距仅 0.23 个百分点,同时提供差分隐私保障。

全文梳理

AI 摘要依据下方信源原文自动整理,非原文转载

区域差异带来的训练难题

零售搜索要面对不同地理区域的用户,各地查询习惯、用词方式和商品偏好都不一样,这种数据异质性同时给隐私保护训练和模型个性化出了难题。联邦学习本身适合解决隐私问题,但常规做法产出的是全局模型,区域层面的效果会被牺牲掉。已有的个性化联邦学习方案在参数层面做文章,遇到现代 transformer 结构时会严重失效,论文把原因归结为绑定嵌入与 LayerNorm 之间的相互作用。RegionFed 的思路是绕开参数层面的改动,把整个流程放在梯度层面完成,论文称其为对架构稳健的框架。

用梯度冲突当统一信号

RegionFed 把区域梯度与全局梯度之间的 ℓ2 冲突当作一个统一信号使用,这个信号承担三件事:判断异质性程度,为每个区域挑选成本最低且够用的个性化策略,以及自适应地调节个性化强度。由于框架把模型视为可微的黑盒,部署时不需要改动代码,论文在 T5-Small、T5-3B、RoBERTa 和 CNN 上都做了验证。结果显示,在参数级方法失效的 transformer 上收益明显,在 CNN 上也有稳定提升。实验覆盖三个公开数据集,分别是 Amazon ESCI、Amazon Reviews 和 LEAF-FEMNIST,并横跨四种架构。

效果与隐私的取舍

在论文给出的对比中,RegionFed-Meta 的成绩与一个会破坏隐私的集中式上界非常接近,后者采用集中训练加区域加权的方式,两者差距落在 1σ 以内。同时该框架提供 ε 约为 0.60 的差分隐私保障,收敛速度为 O(1/√T)。对读者而言,这意味着在零售搜索这类区域差异明显的场景里,可以在不集中原始数据的前提下接近集中训练的个性化水平,并且换用不同模型结构时不必重写代码;不过论文的验证集中在上述数据集与架构组合上,适用范围以材料所述为限。

为什么值得看

解决了联邦学习在 Transformer 上个性化训练的崩溃难题,兼顾隐私与区域性能。

Meta数据集

信源1 家

  1. [1]arXiv cs.AI一手信源RegionFed: Federated Learning for Personalized Query Understanding in Heterogeneous Retail Environments

关键事实

  • 现有个性化联邦学习方法在 T5 等现代 Transformer 上准确率低于 10%。[1]

  • RegionFed 通过梯度层面的 L2 冲突信号诊断异质性并控制个性化强度。[1]

  • RegionFed-Meta 在 Amazon ESCI 数据集上准确率达 92.27%,接近集中式上限。[1]

  • 该方法提供约 0.60 的差分隐私保障,且无需修改模型代码即可部署。[1]

相关 · 论文

本页内容由 AI 自动聚合公开信源生成,仅供了解行业动态参考,不构成任何投资或决策建议。如需引用请以原文出处为准。