提出动态激活量化框架FreeAct,提升大模型压缩效果
FreeAct: Freeing Activations for LLM Quantization
- 用可变变换矩阵应对不同令牌的动态特征差异
- 在扩散与多模态大模型上最高提升5.3%性能
- 适合关注大模型高效部署的研究者
量化是缓解大语言模型(LLMs)内存与计算开销的关键技术。现有基于变换的方法通过正交矩阵将特征空间投影到更平滑的流形上,但通常强制采用固定的单对一变换,难以捕捉输入激活中的动态模式,尤其在扩散型大模型(dLLMs)和多模态大模型(MLLMs)中,不同类型的标记具有显著不同的分布。为此,我们提出FreeAct,一种新型量化框架,放宽静态一对一约束,以适应动态激活差异。理论上,利用激活的秩亏特性,推导出超越简单逆矩阵的解空间,实现激活变换与权重解耦。方法上,FreeAct识别特定令牌动态(如视觉与文本、掩码标记),为激活侧分配独立变换矩阵,而权重侧保持统一静态变换。在dLLMs和MLLMs上的广泛实验表明,FreeAct显著优于基线,最高提升达5.3%,并进行了深入分析。代码将公开发布。
原文摘要 · Abstract (English)
Quantization is pivotal for mitigating the significant memory and computational overhead of Large Language Models (LLMs). While emerging transformation-based methods have successfully enhanced quantization by projecting feature spaces onto smoother manifolds using orthogonal matrices, they typically enforce a rigid one-to-one transformation constraint. This static approach fails to account for the dynamic patterns inherent in input activations, particularly within diffusion LLMs (dLLMs) and Multimodal LLMs (MLLMs), where varying token types exhibit distinct distributions. To advance this, we propose FreeAct, a novel quantization framework that relaxes the static one-to-one constraint to accommodate dynamic activation disparities. Theoretically, we leverage the rank-deficient nature of activations to derive a solution space that extends beyond simple inverse matrices, enabling the decoupling of activation transformations from weights. Methodologically, FreeAct identifies token-specific dynamics (i.e., vision v.s. text, or masked tokens) and allocates distinct transformation matrices to the activation side, while maintaining a unified, static transformation for the weights. Extensive experiments across dLLMs and MLLMs demonstrate that FreeAct significantly outperforms baselines, up to 5.3% performance improvement, with in-depth analyses. Our code will be publicly released.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。