攻击者可利用架构后门窃取同批用户数据并操控模型输出。
Architectural Backdoors for Within-Batch Data Stealing and Model Inference Manipulation
- 设计新型架构后门,利用批量推理漏洞实现跨用户信息泄露。
- 在Transformer等主流模型中可轻松注入,导致同批用户数据被窃取或篡改。
- 提出基于信息流控制的确定性防御方案,可证明多用户输入无干扰。
近十年来,学术界主要研究分类任务中的神经网络后门,攻击者通过操纵模型预测实现恶意行为。然而这类攻击的实际影响尚不明确。本文提出一种更强大且新颖的后门类型,基于近期架构后门进展,专门针对批量推理(batched inference)这一常见硬件优化技术。攻击者可通过修改模型结构,在同一批次内实现大规模用户数据操控与窃取。具体而言,该后门可导致并发请求间的信息泄露,并完全控制其他用户的模型响应。这意味着,只要能改写模型架构,攻击者即可获取并篡改同批其他用户的数据输入与输出。实验表明,此类攻击不仅可行,且极具破坏力,可在主流模型(如Transformer)中轻松植入。为应对这一威胁,我们提出一种确定性缓解策略,提供形式化保证,优于以往依赖大模型检测后门的方法。该策略采用新型信息流控制机制,分析模型图结构,证明同一批次内不同用户输入之间无干扰。基于此方法,我们对Hugging Face平台上的模型进行了大规模分析,发现超过200个模型因使用动态量化而引入了意外的批内信息泄露。
原文摘要 · Abstract (English)
For nearly a decade the academic community has investigated backdoors in neural networks, primarily focusing on classification tasks where adversaries manipulate the model prediction. While demonstrably malicious, the immediate real-world impact of such prediction-altering attacks has remained unclear. In this paper we introduce a novel and significantly more potent class of backdoors that builds upon recent advancements in architectural backdoors. We demonstrate how these backdoors can be specifically engineered to exploit batched inference, a common technique for hardware utilization, enabling large-scale user data manipulation and theft. By targeting the batching process, these architectural backdoors facilitate information leakage between concurrent user requests and allow attackers to fully control model responses directed at other users within the same batch. In other words, an attacker who can change the model architecture can set and steal model inputs and outputs of other users within the same batch. We show that such attacks are not only feasible but also alarmingly effective, can be readily injected into prevalent model architectures, (e.g. Transformers), and represent a truly malicious threat to user privacy and system integrity. Critically, to counteract this new class of vulnerabilities, we propose a deterministic mitigation strategy that provides formal guarantees against this new attack vector, unlike prior work that relied on LLMs to find the backdoors. Our mitigation strategy employs a novel Information Flow Control mechanism that analyzes the model graph and proves non-interference between different user inputs within the same batch. Using our mitigation strategy we perform a large scale analysis of models hosted through Hugging Face and find over 200 models that introduce (unintended) information leakage between batch entries due to the use of dynamic quantization.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。