动态量化漏洞可让攻击者窃取同批次用户数据
Quantamination: Dynamic Quantization Leaks Your Data Across the Batch

- 利用运行时动态量化中的参数泄漏,构建跨批次数据窃取通道
- 至少4个主流框架默认或可配置为存在批量数据泄露风险
- 适合关注AI服务隐私安全的研究者与工程师参考
动态量化作为一种提升机器学习推理效率的实用方法,通过运行时自适应调整量化参数,显著降低内存占用和计算负载,被主流框架广泛推荐用于模型部署。然而本文揭示了一个关键漏洞:若实现或配置不当,动态量化会形成侧信道,使攻击者能从同批次输入中窃取敏感用户数据。我们称之为「Quantamination」——量化导致的数据污染。分析表明,当前至少4个最流行的机器学习框架在默认设置或可配置选项下存在此类风险,理论上可使攻击者部分甚至完全恢复其他用户的输入数据,对现有ML服务架构构成严重隐私威胁。
原文摘要 · Abstract (English)
Dynamic quantization emerged as a practical approach to increase the utilization and efficiency of the machine learning serving flow. Unlike static quantization, which applies quantization offline, dynamic quantization operates on tensors at run-time, adapting its parameters to the actual input data. Today's mainstream machine learning frameworks, including ML compilers and inference engines, frequently recommend dynamic quantization as an initial step for optimizing model serving. This is because dynamic quantization can significantly reduce memory usage and computational load, leading to faster token generation and improved model serving efficiency without substantial loss in model accuracy. In this paper, we reveal a critical vulnerability in dynamic quantization: an adversary can exploit such quantization strategy to steal sensitive user data placed in the same batch as the adversary's input. Our analysis demonstrates that dynamic quantization, when improperly implemented or configured, can create side channels that expose information about other inputs within the same batch. We call this phenomenon Quantamination, describing contamination from quantization. Specifically, we show that at least 4 of the most popular ML frameworks in use today either default to or can use configurations that leak data across the batch boundary. This data leakage, in theory, allows attackers to partially or even fully recover other users' batched input data, representing a serious privacy risk for existing ML serving frameworks.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。