提出NxFP压缩大模型,显著提升低比特下的精度与内存效率。
Nanoscaling Floating-Point (NxFP): NanoMantissa, Adaptive Microexponents, and Code Recycling for Direct-Cast Compression of Large Language Models
- 引入纳米尾数、自适应微指数和代码复用技术解决低比特量化问题。
- 在多个LLM上实现最高0.64的困惑度提升,MMLU准确率提升30%。
- 适合追求高精度低内存占用的大模型部署场景。
随着前沿大语言模型(LLMs)持续推动各行业变革,其模型规模与序列长度的快速增长带来了内存带宽与容量的挑战。近期AMD、Arm、Intel、Meta、Microsoft、NVIDIA和Qualcomm提出了微尺度标准(Mx),通过引入微指数增强块浮点,实现了令人瞩目的困惑度-存储空间权衡。然而,该方法在比特数低于六位时出现显著困惑度下降。本文对现代LLMs进行分析,识别出低比特微尺度格式的三大挑战:异常值追踪不准、量化等级空置以及二进制代码浪费。针对这些问题,本文提出纳米尺度(NxFP)三项技术:纳米尾数(NanoMantissa)、自适应微指数(Adaptive Microexponent)与代码复用(Code Recycling),在直接转换推理中实现优于现有最先进MxFP的效果。实验结果表明,在多个现代LLM上,所提方法在困惑度上最高提升0.64,在MMLU基准上准确率最高提升30%。此外,NxFP将内存占用减少最多16%,同时保持与MxFP相当的困惑度水平。
原文摘要 · Abstract (English)
As cutting-edge large language models (LLMs) continue to transform various industries, their fast-growing model size and sequence length have led to memory traffic and capacity challenges. Recently, AMD, Arm, Intel, Meta, Microsoft, NVIDIA, and Qualcomm have proposed a Microscaling standard (Mx), which augments block floating-point with microexponents to achieve promising perplexity-to-footprint trade-offs. However, the Microscaling suffers from significant perplexity degradation on modern LLMs with less than six bits. This paper profiles modern LLMs and identifies three main challenges of low-bit Microscaling format, i.e., inaccurate tracking of outliers, vacant quantization levels, and wasted binary code. In response, Nanoscaling (NxFP) proposes three techniques, i.e., NanoMantissa, Adaptive Microexponent, and Code Recycling to enable better accuracy and smaller memory footprint than state-of-the-art MxFP. Experimental results on direct-cast inference across various modern LLMs demonstrate that our proposed methods outperform state-of-the-art MxFP by up to 0.64 in perplexity and by up to 30% in accuracy on MMLU benchmarks. Furthermore, NxFP reduces memory footprint by up to 16% while achieving comparable perplexity as MxFP.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。