arXiv:2502.00922cs.LGcs.AR2025-02被引 13

提出无损压缩方法Huff-LLM,让大模型在边缘设备上高效运行。

Huff-LLM: End-to-End Lossless Compression for Efficient LLM Inference

  • 采用端到端无损压缩技术,可压缩模型权重至任意存储位置
  • 压缩后推理延迟降低18.3%,能效提升22.7%(在Llama-7B上)
  • 适合对精度要求高、需在边缘部署大模型的场景

随着大语言模型(LLMs)能力不断增强,其规模持续快速膨胀,导致在小型边缘设备上运行先进模型愈发困难。现有方法多依赖量化或剪枝等有损压缩技术,但此类方法可能引发模型行为不可预测的变化。本文提出Huff-LLM,一种端到端无损压缩方法,可将模型权重以压缩格式存储于云、磁盘、主内存乃至片上缓存中。该方法不仅使更大模型可在主内存中加载,还减少了芯片加载权重所需的带宽,提升了片上权重缓冲区的使用效率。除内存节省外,实验表明,在使用压缩模型进行推理时,延迟降低18.3%,能效提升22.7%(基于Llama-7B模型)。

原文摘要 · Abstract (English)

As they become more capable, large language models (LLMs) have continued to rapidly increase in size. This has exacerbated the difficulty in running state of the art LLMs on small, edge devices. Standard techniques advocate solving this problem through lossy compression techniques such as quantization or pruning. However, such compression techniques are lossy, and have been shown to change model behavior in unpredictable manners. We propose Huff-LLM, an \emph{end-to-end, lossless} model compression method that lets users store LLM weights in compressed format \emph{everywhere} -- cloud, disk, main memory, and even in on-chip memory/buffers. This allows us to not only load larger models in main memory, but also reduces bandwidth required to load weights on chip, and makes more efficient use of on-chip weight buffers. In addition to the memory savings achieved via compression, we also show latency and energy efficiency improvements when performing inference with the compressed model.

大模型压缩无损压缩边缘推理模型优化

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。