跳到正文
原文

多任务学习共享表示上的黑盒隐私攻击

Black-Box Privacy Attacks on Shared Representations in Multitask Learning

arXiv· arXiv:2506.16460· ICLR 2026· John Abascal, Nicolás Berrios, Alina Oprea, Jonathan Ullman, Adam Smith, Matthew JagielskiKhoury College of Computer Sciences, Northeastern University; Department of Computer Science, Boston University; Google DeepMind本站收录· 16 天前· 原文发表攻击

论文速读

攻击

据论文 PDF 整理(AI 生成),以原文为准

作者提出黑盒 task-inference 威胁模型与两种攻击,推断某任务是否参与 MTL 共享表示训练;Stack Overflow 个性化设置下强对手 AUC 达 1.000(Figure 3c)。

威胁模型纯黑盒 task-inference:攻击者仅能查询共享表示获取 embedding,并持有来自挑战任务的样本;强对手持有该任务的训练样本,弱对手只有同分布的新鲜样本;不需要 shadow model 或带标签参考数据;受害系统是用 MTL 训练的共享表示加各任务线性头。

问题
多任务学习(MTL)的共享表示本应是各方需共享的最小信息单元,作者研究它是否会泄露哪些任务参与了训练,并提出任务级(而非样本级)的推断问题。
方法
提出黑盒 task-inference 威胁模型(强/弱两种对手),并设计两种无需 shadow model 的攻击:坐标方差攻击与成对内积攻击(含 whitening);另用高斯均值估计的 tracing 攻击做理论分析。
实验与结果
在 CelebA、FEMNIST、Stack Overflow 的个性化与多学习问题两种 MTL 设置下,强弱对手均取得非平凡 AUC;强对手在 Stack Overflow 个性化上 AUC 达 1.000,弱对手在多学习问题设置下与强对手差距变小。
局限与可以继续做的
作者称 user-level differential privacy 是潜在防御但未实验验证;攻击在梯度裁剪与归一化下仍成功。实验限于两种 MTL 用法、三个数据集与 ResNet/BERT 架构,未评测自适应攻击或实际防御效果。
实验设置ResNet34、ResNet8 等 · CelebA、FEMNIST 等 · AUC、TPR at fixed FPR 等
被测模型
ResNet34ResNet8ResNet50BERT SmallBERT Medium
基准
CelebAFEMNISTStack Overflowsynthetic Gaussian dataset
指标
AUCTPR at fixed FPRBalanced Accuracy
AI 导读研究者提出针对多任务学习共享表示的黑盒任务推断攻击,攻击者仅凭查询共享表示得到的嵌入向量,即可判断某个任务是否参与了训练。攻击利用同一任务嵌入之间的依赖关系,不需要影子模型或带标注的参考数据。在视觉和语言领域的多种多任务学习场景中,实验表明即使只掌握目标任务的新样本而非训练数据,黑盒攻击者也能成功推断该任务是否在训练集中。作者还对简化的学习设定给出理论分析,显示拥有训练样本与只有新样本的攻击者之间存在严格分离。

研究者提出针对多任务学习共享表示的黑盒任务推断攻击,攻击者仅凭查询共享表示得到的嵌入向量,即可判断某个任务是否参与了训练。攻击利用同一任务嵌入之间的依赖关系,不需要影子模型或带标注的参考数据。在视觉和语言领域的多种多任务学习场景中,实验表明即使只掌握目标任务的新样本而非训练数据,黑盒攻击者也能成功推断该任务是否在训练集中。作者还对简化的学习设定给出理论分析,显示拥有训练样本与只有新样本的攻击者之间存在严格分离。

深度解读

6个问题,点开问题读完整回答

  1. 这篇论文试图解决什么问题?

    MTL 的共享表示会泄露哪些任务参与了训练;作者提出黑盒 task-inference 威胁模型与攻击。

    核心问题:多任务学习(MTL)的共享表示是否会泄露某个任务是否参与了训练,以及泄露到什么程度。

    • 场景与重要性:MTL 让数据稀缺的多个任务(用户、机构)共享一个表示、各自保留任务头,作者称共享表示常是各方为获得泛化能力所必须贡献的最小信息单元,因此其泄露直接威胁协作学习的隐私前提。
    • 现有工作不足:作者称此前针对 MTL 的隐私攻击(如 MTL-Leak [Yan+24])限于样本级 membership inference 和模型窃取,且需要查询任务头、训练参考模型;许多 membership/property inference 攻击依赖 shadow model 或辅助数据。
    • 核心观察:作者假设共享表示对来自同一任务的样本会产生具有正负协依赖的 embedding,类似对比学习中同一图像增广之间的关系,这一信号无需参考模型即可利用。
    • 论文贡献:提出 task-inference 威胁模型(区分强/弱对手)、两个纯黑盒攻击算法、高斯均值估计下的理论分离,以及在视觉和语言两类 MTL 用法上的评测。
    • 威胁模型:
      • 目标:判定挑战任务 τ* 是否包含在 MTL 共享表示的训练任务集合中(IN/OUT 二分类)。
      • 知识:纯黑盒,仅能对共享表示做查询获得 embedding;不知道任务分布,无 shadow model、无带标签参考数据。
      • 能力:强对手持有挑战任务的训练样本;弱对手只有来自该任务分布、未参与训练的新鲜样本。
      • 受害系统:用 MTL 同时训练的共享表示(神经网络编码器)加各任务线性分类头。
  2. 有哪些相关研究?

    相关工作涵盖 membership/property/user inference、表示模型隐私与联邦多任务学习;本文定位为任务级、无 shadow model 的黑盒攻击。

    Membership-inference 攻击

    • Shokri 等 [SSS16]、Dwork 等 [Dwo+15] 等——判定单个样本是否在训练集中,是与差分隐私直接相关的基础泄露形式,已被扩展到编码器、LLM、联邦学习等设置。
    • [Hum+23] 与 [RL22]——研究攻击者了解训练数据子总体分布时的 membership inference,论文称区分子总体内的成员比区分 i.i.d. 样本更容易。

    Property/distribution inference

    • Ateniese 等 [Ate+15] 提出 property inference,判定子总体占比;[Cha+22] 的 property existence 与 [Har+23] 的 distributional membership inference 进一步判定某属性是否出现过。
    • 论文指出这些工作的对手访问水平与本文弱对手相近,但假设攻击者有足够样本和算力训练多个参考模型。

    表示模型与群体级攻击

    • [SR20; SS20] 研究 embedding 模型的信息泄露;EncoderMI [Liu+21b] 利用对比学习编码器对增广的鲁棒性做 membership inference,但需 shadow model 校准,本文攻击受其启发但不需要参考模型。
    • [Kan+23] 的 user inference 对用户分区的微调 LLM 做用户级判定,数据访问假设与本文弱对手相同,但用预训练参考模型校准置信度;[Che+23] 推断人脸数据贡献,[Sur+23a] 研究跨 silo 联邦学习中的主体泄露;[Mai+24; Ren+25] 研究 LLM 语料的数据集级推断。

    MTL 与联邦学习

    • Caruana [Car97] 的 MTL 框架与 [Smi+17] 的联邦多任务学习;[She+23] 称只共享表示可改善协作学习的收敛与性能,本文正是攻击这一"最小共享单元"。

    基线与基准:论文未实现对比基线攻击;评测使用 CelebA、FEMNIST、Stack Overflow 与合成高斯数据集。作者将本文定位为把样本级攻击推广到任务级、且首个(论文称 novel)在纯黑盒、无 shadow model 条件下针对 MTL 共享表示的任务级推断。

  3. 论文如何解决这个问题?

    用安全博弈形式化 task-inference,设计坐标方差与成对内积两个黑盒攻击,并用高斯均值估计的 tracing 攻击给出强弱对手的理论分离。

    整体思路:把"任务是否参与训练"形式化为一个 IN/OUT 安全博弈,利用同一任务样本在共享表示空间中的协依赖构造检验统计量,无需 shadow model。两个攻击算法(Coordinate-Wise Variance 与 Pairwise Inner Product)加一套高斯均值估计的理论分析构成方法主体。

    威胁模型与安全博弈

    • 挑战者从任务分布 Q 抽 T 个任务训练共享表示 h_θ,再以等概率抽取 IN(训练过的)或 OUT(未训练的)挑战任务,给攻击者一批该任务的样本。
    • 攻击者用这批样本查询 h_θ 得到 embedding,输出 IN/OUT 猜测;强对手的样本在 b=1 时来自训练集,弱对手则为同分布新鲜样本。
    • 作者称该威胁模型在任务为用户时对应 user inference、每任务单样本时退化为 membership inference、任务为学习问题时类似 property existence。

    理论分析:均值估计上的 tracing 攻击

    • 简化模型:T 个任务均值来自 N(μ̄, σ̄²I),每任务 N 个样本,发布多任务均值 μ̂ 作为共享表示的类比。
    • 检验统计量 z = ⟨ μ̂ − μ̄, μB − μ̄ ⟩,即发布统计量与攻击者批次均值的相关性,阈值化后判定 IN/OUT。
    • Theorem 4.1/4.2:强对手 IN 时期望为 (d/T)(σ̄² + σ²/N),弱对手为 (d/T)σ̄²;Theorem 4.3 给出方差界,作者据此称强对手严格占优,且追踪任务比追踪单样本更容易。

    攻击算法

    • Algorithm 1(坐标方差):对挑战集 k 个样本的 embedding 计算经验协方差矩阵,取迹(即各坐标方差之和)作为统计量,超阈值判 IN。
    • Algorithm 2(成对内积):先对 embedding 做 whitening(W(e − ē),W 由剔除当前任务后估计的正则化协方差得到),再取所有不同对的内积绝对值(或余弦相似度)的均值作为统计量。
    • 阈值选取保持纯黑盒:不用带标签 OUT 任务校准,直接按统计量的 50/75/90 分位数设阈值。

    MTL 训练与评测方式

    • 训练遵循 [Car97] 的原始 MTL:共享编码器加每任务线性头,用 AdamW、大 weight decay、梯度归一化与裁剪(C=1)、bottleneck 投影到 16 或 32 维,训练 200–300 epoch。
    • 成功判定:以 ROC/AUC 与固定低 FPR(如 1%)下的 TPR 为指标,另报告分位数阈值下的 (TPR, FPR) 与 balanced accuracy;每个实验区分强/弱对手各跑 128 次试验。
  4. 论文做了哪些实验?

    在 CelebA、FEMNIST、Stack Overflow 上评测两种 MTL 用法,强弱对手均非平凡;另做泛化差距分析与合成数据消融。

    实验设置

    • 被测模型:ResNet34(CelebA 个性化)、ResNet8(FEMNIST)、ResNet50(CelebA 多学习问题)、BERT Small 29M(Stack Overflow 个性化)、BERT Medium 41M(Stack Overflow 多学习问题),均加每任务线性头。
    • 数据集:CelebA(按人或按 40 个面部属性切任务)、FEMNIST(按书写者)、Stack Overflow(按用户或按 256 个主题切任务)、合成高斯数据集。
    • 任务规模:CelebA 个性化 256 任务×22 样本;FEMNIST 128 任务×128 样本;Stack Overflow 个性化 128 IN + 128 OUT 任务;多学习问题 CelebA 20 IN + 20 OUT、Stack Overflow 70 IN + 70 OUT(每任务 1024 帖)。
    • 攻击配置:每任务样本数强对手 8–32、弱对手 4–16(因数据集而异);每个设置跑 128 次试验,多学习问题 CelebA 另平均 8 次 MTL 训练。
    • 指标:AUC、固定 FPR 下 TPR、50/75/90 分位数阈值下的 (TPR, FPR) 与 balanced accuracy;无 LLM 评审,判定为纯统计阈值。
    • 效率:作者称每次黑盒查询加预测在单张 RTX 4090 上约 0.1 秒,最少每任务 4 个样本。

    主结果

    个性化设置 AUC(据 Figure 3 图注):

    表格较宽,可左右滑动

    数据集Variance 强Variance 弱Inner Product 强Inner Product 弱
    CelebA0.9170.5520.8780.533
    FEMNIST0.6910.5740.6570.544
    Stack Overflow1.0000.5560.9990.624

    多学习问题设置 AUC(据 Figure 5 图注):

    表格较宽,可左右滑动

    数据集Variance 强Variance 弱Inner Product 强Inner Product 弱
    CelebA0.7450.5670.7570.543
    Stack Overflow0.9180.9090.9430.935
    • 作者称个性化设置下弱对手 TPR 在所有固定 FPR 上都不超过强对手,与理论分离一致。
    • 固定 1% FPR 时:CelebA 个性化 Variance 攻击强/弱 TPR 为 61.2%/2.9%;Stack Overflow 个性化 Inner Product 强/弱为 98.5%/8.2%;FEMNIST 强对手 TPR 为 3%。
    • 多学习问题的 Stack Overflow 上强弱对手 (TPR, FPR) 几乎相等(Table 2),作者归因于任务与标签直接绑定、模型必须在训练任务上达到高效用。

    分位数阈值下的黑盒判定

    • Table 1(个性化,Inner Product):Stack Overflow 强对手 50 分位阈值达 (98.7%, 1.2%),75 与 90 分位 FPR 为 0%;弱对手 90 分位为 (16.3%, 3.6%)。
    • Table 2(多学习问题):Stack Overflow 弱对手 90 分位达 19.6% TPR、0.2% FPR;CelebA 强对手 90 分位 22.8% TPR、0% FPR。
    • 作者称这说明即使无参考知识、按分位数选阈值,攻击仍有非平凡成功率。

    泄露来源分析

    • 泛化差距(Figure 6):Stack Overflow 个性化上,强对手 AUC 随训练损失与"zero-shot"损失差距增大而上升,小差距时上升更快;弱对手 AUC 也随差距增长但方差更大。作者称模型既泛化到训练任务分布、也记忆了任务整体信息。
    • 任务头相关性(Figure 7、附录 C.1):ResNet 的任务头权重正相关而 BERT 接近正交;相关程度排序(FEMNIST > CelebA > Stack Overflow)与个性化实验 AUC 排序一致,作者称 Variance 攻击的主要信号来自表示在某些方向上的"过度自信"。

    其他消融与分析

    • 合成数据(Figure 8,Variance 攻击,4 次训练×64 次试验):总样本数(每任务样本数或任务数)增加时强对手 AUC 下降,强弱差距缩小。
    • 嵌入维度在合成数据消融中对 AUC 影响很小(Figure 8a)。
    • 理论模拟(Figure 2,T=256, d=256, N=8, k=4):强对手 AUC 0.922(T=64)降至 0.692(T=512),弱对手 0.871 降至 0.656,任务越多攻击越难。
    • 例外:FEMNIST 个性化上强对手 1% FPR 下 TPR 仅 3%,作者未给出单独解释;多学习问题 CelebA 弱对手 AUC(0.567/0.543)接近随机。
  5. 有什么可以进一步探索的点?

    作者讨论 user-level DP 作为潜在防御但未验证;实验限于两种 MTL 用法、三个数据集与 ResNet/BERT。

    作者指出的局限与后续方向

    • 潜在防御未验证(Section 7.1):作者提出 user-level(及 group-level)differential privacy 可能防御本攻击,并指出目前只有 [HWS23] 在 MTL 协作学习中给出 client-level 隐私算法;论文未做任何防御实验。
    • 梯度裁剪不构成防御(Section 7.1):作者称与 membership-inference 文献中经验防御的观察不同,本攻击在使用梯度裁剪与归一化训练的模型上仍成功(裁剪本为其训练收敛所用)。
    • 论文未设专门 Limitations 节,以上为其在 Conclusion and Discussion 中明确写出的内容。

    实验覆盖范围

    • 被测模型为 ResNet34/8/50 与 BERT Small/Medium 共 5 个实例,覆盖视觉与语言两个领域(Section 6.2)。
    • MTL 用法覆盖个性化(任务=用户/书写者)与多学习问题(任务=属性/主题)两类(Section 6.4、6.5)。
    • 数据集为 CelebA、FEMNIST、Stack Overflow 与合成高斯数据;每个设置攻击试验 128 次,多学习问题 CelebA 平均 8 次 MTL 训练,合成消融 4 次训练×64 次试验。
    • 理论分析限于高斯混合均值估计的简化设置,证明见附录 B。
    • 论文未报告攻击对自适应对手或任何实际防御(如 DP 训练)的评测,也未报告攻击统计量的置信区间或显著性检验。
  6. 总结一下论文的主要内容

    提出 MTL 共享表示的黑盒 task-inference 攻击,理论与实验均显示强对手占优、弱对手也可非平凡推断。

    一篇针对多任务学习隐私的攻击论文:作者问 MTL 的共享表示——各方为协作训练必须共享的最小信息单元——是否会泄露某个任务是否参与了训练。

    • 威胁模型:纯黑盒 task-inference,攻击者仅能查询共享表示并持有挑战任务的样本;强对手用训练样本,弱对手只用同分布新鲜样本,无需 shadow model 或参考数据。
    • 方法:两个攻击——坐标方差(同任务 embedding 的坐标方差之和)与成对内积(whitening 后 embedding 对内积均值);理论上把 MTL 类比为高斯混合均值估计,证明强对手统计量期望多一项 (d/T)(σ²/N),严格优于弱对手。
    • 结果一:个性化设置下,Stack Overflow(BERT Small)强对手 Variance 攻击 AUC=1.000,1% FPR 时 Inner Product 强/弱 TPR 为 98.5%/8.2%;CelebA(ResNet34)1% FPR 下强/弱为 61.2%/2.9%(Figure 3)。
    • 结果二:多学习问题设置下 Stack Overflow 强弱对手差距基本消失(AUC 0.918/0.909 与 0.943/0.935,Figure 5),弱对手 90 分位阈值下达 19.6% TPR、0.2% FPR(Table 2)。
    • 结果三:攻击 AUC 随模型泛化差距增大而上升(Figure 6);合成数据上总样本增多会缩小强弱对手差距,嵌入维度影响不大(Figure 8)。
    • 作者结论:共享表示在两种 MTL 用法下都会泄露任务级信息,梯度裁剪不能阻止攻击;作者认为 user-level differential privacy 是潜在防御方向,但未做验证。

相关论文

共同使用 CelebA
再看 1 篇
共同使用 CelebA
阅读原文ICLR 2026