用因果表示学习防住垂直联邦学习中的样本重建攻击
CausShield: Sample Reconstruction-Resilient Vertical FL via Causal Representation Learning

- 通过因果模型分离任务相关与隐私敏感特征
- 在保持模型性能前提下显著提升隐私保护能力
- 适合关注隐私安全的联邦学习系统开发者
垂直联邦学习(VFL)可在不共享原始样本的前提下,利用跨参与方的纵向特征进行分布式训练;然而仍易受主动样本重建攻击。现有防御方法难以兼顾模型效用与隐私保护,或因误删任务相关特征,或依赖端到端监督训练导致早期阶段漏洞。本文提出CausShield,基于结构因果模型(SCM)理论,将样本中的因果特征(直接相关于任务目标)与非因果特征(仅含样本私密信息)分离。该方法通过无监督表示学习求解优化问题,在保证模型收敛性的同时,实现全周期隐私保护。实验对比七种先进方法(包括InvL、URVFL),结果表明,CausShield在隐私保护、模型效用和计算效率上均表现更优。
原文摘要 · Abstract (English)
Vertical federated learning (VFL) is a distributed learning paradigm that leverages vertically partitioned features across isolated parties without sharing raw samples; however, it remains vulnerable to active sample reconstruction attacks. Existing defenses fail to achieve a satisfactory trade-off between model utility and privacy protection, due to either suppressing task-relevant information alongside privacy-sensitive features or relying on end-to-end supervised training to converge the defense module, which exposes the model to early-epoch vulnerability. To address this challenge, we adopt a structural causal model (SCM) insight and construct CausShield. From a task-learning standpoint, causal features within a raw sample are those that are directly relevant and contributory to the learning objective, whereas non-causal features are task-irrelevant but often encode sample-specific private information, thereby facilitating reconstruction. Importantly, we lay a theoretical foundation to prove this insight. CausShield thus decomposes the shared representations between the client and the coordinating server in VFL into task-relevant and task-irrelevant components to ensure full-cycle privacy protection. Nonetheless, the decomposition is inherently challenging due to the dual objectives of preserving model utility while mitigating privacy leakage. We address this via a carefully formulated optimization problem, which is solved through unsupervised representation learning. We further theoretically prove that CausShield preserves the convergence behavior of standard VFL. Extensive experiments compare CausShield against seven SOTAs, including InvL (USENIX Security'25), and evaluate robustness against advanced reconstruction attacks such as URVFL (NDSS'25). Results demonstrate that CausShield consistently outperforms in privacy protection, model utility, and computational efficiency.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。