arXiv:2605.19262cs.LGcs.CR2026-05

首次针对掩码扩散语言模型设计高效后门攻击,实现隐蔽控制。

Backdooring Masked Diffusion Language Models

论文配图:Backdooring Masked Diffusion Language Models
图 1 · 摘自论文原文
  • 通过替换掩码分布构造触发路径,诱导模型生成指定内容。
  • 在多个数据集上攻击成功率接近100%,且保持原始文本质量。
  • 适用于全量微调和参数高效微调,对防御手段有较强鲁棒性。

掩码扩散语言模型(MDLMs)是新兴的文本生成范式,但其训练阶段的安全性尚未被充分研究。现有针对高斯扩散模型或自回归语言模型的后门攻击不适用于MDLMs,因其依赖离散状态破坏与迭代去噪,而非连续加噪或从左到右预测。本文首次系统研究了对MDLMs的训练时后门攻击,提出SHADOWMASK攻击方法:通过将标准全掩码终止分布替换为触发-掩码混合先验,构建从触发污染状态到攻击者指定目标的专用去噪路径,同时保留正常去噪行为。我们进一步提供了严格的数学建模,定义后门前向过程,推导反向时间后验,并获得连续时间训练目标。在基于DiT的MDLM和LLaDA-8B-Instruct模型上,于WikiText-103、OpenWebText和Alpaca数据集上的评估表明,SHADOWMASK实现近100%攻击成功率,显著优于标准数据投毒,大幅保持干净文本性能,适用于全模型与参数高效微调,并对主流防御策略具有强鲁棒性。

原文摘要 · Abstract (English)

Masked diffusion language models (MDLMs) are emerging as a compelling new paradigm for text generation, but their training-time security remains largely unexplored. Existing backdoor attacks on Gaussian diffusion models or autoregressive language models do not directly apply to MDLMs because MDLMs rely on discrete state corruption and iterative denoising rather than continuous noising or left-to-right prediction. In this work, we present the first systematic study of training-time backdoor attacks on MDLMs. We propose SHADOWMASK, a backdoor attack that modifies the MDLM forward corruption process by replacing the standard all-mask terminal distribution with a trigger-mask mixture prior. This creates a dedicated denoising pathway from trigger-corrupted states to attacker-specified targets while preserving clean denoising behavior. We further provide a principled mathematical formulation by defining the backdoored forward process, deriving the reverse-time posterior, and obtaining the continuous-time training objective. Evaluations on DiT-based MDLM and LLaDA-8B-Instruct across WikiText-103, OpenWebText, and Alpaca show that SHADOWMASK achieves near-100% attack success, substantially outperforms standard data poisoning, largely preserves clean utility, remains effective under full-model and parameter-efficient fine-tuning, and is robust against representative defenses.

后门攻击扩散模型语言模型安全

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。