用秘密探针检测分布式大模型推理中的恶意节点
Integrity of peer-to-peer distributed LLM inference under malicious nodes

- 通过混入已知正确输出的隐藏输入检测节点篡改
- 在408种配置下均实现1.0的AUROC,精准识别恶意节点
- 适合关注去中心化AI安全的开发者与研究者
点对点分布式推理将大型语言模型(LLM)分摊到多个由不同独立方拥有的消费级硬件节点上运行。每个请求需经过多方控制的节点,任何一方都可能篡改其处理层的输出以破坏最终结果。虽可通过在可信硬件上重新计算前向传播来检测,但会增加额外开销。现有完整性验证方法如已知答案陷阱和密码承诺仅测试精确正确性,未考虑良性节点间的正常波动。本文提出一种新方法:通过测量各节点传递给下一节点的激活值变化来检查输出完整性。使用少量秘密的探针输入,其正确激活值预先已知,混入正常流量中。由于无法区分探针与真实请求,任何篡改节点也必污染探针。与已知参考值的偏差可揭示恶意行为:良性节点仅表现出硬件噪声引起的微小波动,而篡改节点则显著偏离。我们将恶意节点识别为概率性检验,分离两种漂移分布,无需固定阈值。在408种配置下,所有指标与成功标准均在实验前确定,检测器达到AUROC 1.0,每次实验中所有探针的恶意分片均被正确排在所有良性分片之上。
原文摘要 · Abstract (English)
Peer-to-peer distributed inference executes a Large Language Model (LLM) on pooled consumer hardware by spreading its layers across many nodes. Every request passes through nodes that are owned and controlled by multiple independent parties. However, in this setting, any party can tamper with the output of its layers to corrupt the end result. Recomputing the forward pass on trusted hardware can catch this, but it introduces additional computational cost. The scientific literature includes several prior integrity-checking approaches, such as known-answer traps for image classifiers and cryptographic commitments. However, these solutions test only the exact correctness and do not account for the ordinary variation that may arise between benign nodes. In this paper, we propose a method that checks the output integrity by measuring the variation in the activations that each node passes to the next. A peer who wants to use the network selects a small set of secret canary inputs whose correct activations are known in advance and mixes them into regular traffic. Because the peers cannot tell a canary from a real query, any tampering node corrupts them as well. The deviation from the known reference then reveals malicious activity: benign nodes exhibit only minor variation from hardware-induced noise, whereas tampered nodes deviate far more. We treat the identification of malicious nodes as a probabilistic test that separates two drift distributions, without relying on a fixed threshold. We study 408 configurations with metrics and success criteria fixed before any experiment ran; the detector reaches AUROC 1.0, correctly ranking the malicious shard above every benign shard on every canary in every configuration.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。