Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Exploiting LLM Quantization

About

Quantization leverages lower-precision weights to reduce the memory usage of large language models (LLMs) and is a key technique for enabling their deployment on commodity hardware. While LLM quantization's impact on utility has been extensively explored, this work for the first time studies its adverse effects from a security perspective. We reveal that widely used quantization methods can be exploited to produce a harmful quantized LLM, even though the full-precision counterpart appears benign, potentially tricking users into deploying the malicious quantized model. We demonstrate this threat using a three-staged attack framework: (i) first, we obtain a malicious LLM through fine-tuning on an adversarial task; (ii) next, we quantize the malicious model and calculate constraints that characterize all full-precision models that map to the same quantized model; (iii) finally, using projected gradient descent, we tune out the poisoned behavior from the full-precision model while ensuring that its weights satisfy the constraints computed in step (ii). This procedure results in an LLM that exhibits benign behavior in full precision but when quantized, it follows the adversarial behavior injected in step (i). We experimentally demonstrate the feasibility and severity of such an attack across three diverse scenarios: vulnerable code generation, content injection, and over-refusal attack. In practice, the adversary could host the resulting full-precision model on an LLM community hub such as Hugging Face, exposing millions of users to the threat of deploying its malicious quantized version on their devices.

Kazuki Egashira, Mark Vero, Robin Staab, Jingxuan He, Martin Vechev• 2024

Related benchmarks

TaskDatasetResultRank
Code GenerationHumanEval (test)--
701
Code GenerationMBPP (test)--
411
Language UnderstandingMMLU
MMLU Accuracy37.9
307
Backdoor DefenseTiny-ImageNet
Accuracy61.91
267
Image ClassificationCIFAR-10
Clean Accuracy92.31
175
Language UnderstandingMMLU
MMLU Accuracy25.7
144
Code GenerationHumanEval
HumanEval Score11.3
140
Multi-task Language UnderstandingMMLU (test)
Normalized Accuracy52.9
107
Backdoor DefenseCIFAR-10 (test)
Clean Accuracy93.99
94
Ad Injection Attack Resistance EvaluationAd Injection
MMLU65.92
60
Showing 10 of 25 rows

Other info

Follow for update