FlattenQuant: 通过 Per-tensor 量化来突破大型语言模型的推断计算限制
原文中文,约400字,阅读约需1分钟。发表于: 。使用 FlattenQuant 方法在大型语言模型中实现低比特量化,通过将张量中的大通道展平以显著减少张量的最大值,本方法能够直接使用 4 位来实现 LLMs 中线性层计算的 48.29%,并使用 8 位来处理其余层,从而实现高达 2 倍的速度提升和 2.3 倍的内存减少,而准确度损失微不足道。
为了解决大型语言模型的内存需求和推断成本问题,提出了一种高效的仅权重量化方法。通过减少内存消耗和加速推断,利用预训练模型的权重来确保最小质量降低。适用于混合专家模型和密集模型,无需额外微调。通过自适应的量化粒度解决挑战和问题,展示了方法的有效性。实现了高效的GPU矩阵乘法和解量化算法,支持不同激活和权重的乘法。在大规模开源模型上评估,展示了最小的准确性损失和高吞吐量。