Back to AI News
arxiv

AutoCompress: Hardware-Guided Model Compression for LLMs

AutoCompress is a new framework that integrates hardware and workload constraints directly into differentiable architecture search for LLM pruning. It jointly optimizes global and per-layer structure using serving-engine cost models, improving accuracy over Minitron across multiple GPUs and workloads. The method also compresses a 120B model to 74B parameters while retaining 95% of benchmark performance.

AI Brief

This introduces a practical approach to LLM compression that directly targets real-world serving constraints, offering measurable gains over a state-of-the-art method.

arXiv:2611.08290v1 Announce Type: cross Abstract: LLM pruning must preserve quality while meeting hardware- and workload-specific constraints on latency, throughput, and memory. Existing pruning workflows often target parameter counts, requiring practitioners to manually select model sizes that meet latency, throughput, and GPU memory limits. However, parameter count alone does not predict these metrics, which depend on the model architecture, workload, hardware, and serving engine. We introduce AutoCompress, a hardware-guided compression framework that directly incorporates such constraints into differentiable architecture search. AutoCompress jointly optimizes global and per-layer structure to produce heterogeneous architectures, guided by serving-engine cost models validated against end-to-end latency, throughput, and memory measurements. We evaluate AutoCompress on long-context local coding, batched datacenter serving, and latency-sensitive in-game interaction. Compared to the state-of-the-art Minitron pruning framework, AutoCompress improves average accuracy across eight benchmarks by 6.57, 2.35, and 0.57 percentage points on RTX 5070, RTX 5080, and H100, respectively. AutoCompress also compresses Nemotron 3.5 Super from 120B to 74B parameters for single-B200 BF16 serving at 128K context, retaining 95% of its mean benchmark score.
Be the first to rate this article.

Comments & Questions (0)

Captcha
Please be respectful — let's keep the conversation friendly.

No comments yet

Be the first to comment!