智客 ZICQ
EN 登录 / 注册
智客前沿 算力架构 #MACaQ #W4A4量化 #MXFP4 #Hadamard变换 #后训练量化

MACaQ以逐层流式量化实现W4A4,32.5GB显存量化122B模型

Mr.Xu 的头像

文 / Mr.Xu 智客前沿编译 · 审校

发布时间:

中文阅读 (Chinese) English Version

摘要:MACaQ提出面向MXFP4 W4A4推理的模型无关后训练量化方法,将每个线性层独立处理:在线对激活执行分通道缩放与分块Hadamard变换,并将逆变换吸收到权重,从而摆脱跨层基变换依赖。配合Hessian感知的Flip-Refinement,方法在Qwen3-32B上较纯MXFP4仅增加2.3%端到端开销,在Llama-3-8B上以约11倍更短的量化时间达到DuQuant++与MR-GPTQ的最佳困惑度,且可在32.5GB显存内完成122B参数模型量化。


核心思路:把低精度量化拆成逐层独立问题

MACaQ针对权重与激活同时低精度的W4A4推理,目标是同时降低显存带宽占用和矩阵乘法计算成本。它不采用需要跨层传播基变换的方案,而是把每个线性层视为独立量化单元:激活在进入矩阵乘法前在线执行分通道缩放与分块Hadamard变换,权重则预先吸收相应的逆变换。

这种设计消除了层间依赖,使量化过程可以按层流式执行。处理时不必同时容纳完整模型,只需让一个层驻留在内存中,这也是其能够以32.5GB GPU内存量化122B参数模型的关键原因。

需要区分的是,Hadamard变换不是免费操作。MACaQ把变换放在推理路径中在线执行,因此仍需付出额外矩阵运算和内核开销。论文报告,在vLLM中运行Qwen3-32B时,相对纯MXFP4的端到端开销为2.3%,但该数字是否能够扩展到更小批量、更长上下文或不支持相关融合算子的硬件,仍需独立验证。

量化机制:在线变换吸收进权重

对于线性层,可将计算写成:

[ y = xW^T ]

MACaQ对激活 (x) 应用分通道尺度 (s) 和分块Hadamard变换 (H),得到:

[ y = (Hs)(HW^T)^T ]

其中,(HW^T) 可以在量化前折叠进权重。部署阶段只需对输入激活执行缩放与Hadamard变换,随后进入FP4矩阵乘法。分块Hadamard的作用是改善激活分布中的离群值和通道相关性,让更多数值落入FP4更容易表达的区间。

与全局基变换相比,逐层局部设计牺牲了跨层统一坐标系的理论最优性,却换取了工程上的独立性、可流式处理和无需针对架构定制传播逻辑。对于MoE、混合注意力结构或扩散模型,这种“局部近似”尤其具有部署吸引力。

代价也很明确:激活变换需要在线执行,低精度矩阵乘节省的带宽与算力未必能完全抵消变换开销;不同GPU、批量大小和序列长度下的实际收益,必须由硬件内核实测,而不能只依据理论FLOPs判断。

Flip-Refinement:用二阶信息优化FP4码值

普通round-to-nearest仅按数值距离把激活或权重映射到最近的FP4码值。MACaQ引入Hessian感知的Flip-Refinement:基于局部二阶近似评估当前FP4码值的损失,然后通过坐标下降逐个尝试将坐标移动到相邻的可表示值。

其关键不是简单扩大量化搜索空间,而是利用Hessian信息判断哪些坐标翻转真正能降低量化误差。给定输入分布 (X),量化误差通常可近似为:

[ \Delta L \approx \frac{1}{2}(x-\hat{x})^T H (x-\hat{x}) ]

其中 (H) 反映模型对输入扰动的敏感度。与均匀网格搜索相比,Flip-Refinement可以沿更有利的方向调整离散FP4码值,同时保留“每次只移动一个坐标”的贪心策略。

但坐标下降仍可能陷入局部最优,而且Hessian估计、敏感度计算和候选评估会增加离线量化时间。论文称,在Llama-3-8B上,该方法用约11倍更短的量化时间达到DuQuant++与MR-GPTQ的最佳困惑度,说明精度收益并不必然依赖冗长的全局优化。仍需核查不同随机种子、层宽度、模型分布和校准集规模下的稳定性。

工程表现与适用边界

MACaQ展示了三个有代表性的结果:

  • 在Qwen3-32B的vLLM部署中,相对纯MXFP4增加2.3%端到端开销;
  • 在Llama-3-8B上,以约11倍更少的量化时间达到DuQuant++与MR-GPTQ的最佳困惑度;
  • 使用32.5GB GPU内存,在64分钟内完成122B参数模型量化。

方法还被直接应用于图像扩散模型,且无需针对架构做专门修改。这支持其“模型无关”定位,但当前摘要没有给出扩散模型的量化质量、采样速度或FID等完整指标,因此不能据此断言所有生成模型都能获得同等收益。

此外,W4A4会同时放大权重和激活误差,尤其在长上下文、激活离群值较多或模型对数值扰动敏感的任务中,困惑度改善并不必然转化为下游任务能力稳定。上线前还应分别测试预填充、解码、小批量、高并发与不同上下文长度的端到端表现。

开发者落地建议

对于需要在有限显存部署超大模型的团队,MACaQ最有价值的部分不是单一困惑度纪录,而是逐层流式量化机制:量化阶段可以逐层处理权重,避免完整模型与完整校准数据同时驻留内存。

实际接入可按以下顺序推进:

  1. 先在目标vLLM与GPU型号上验证Hadamard变换、FP4矩阵乘和缩放操作是否真正融合,避免把论文中的算子开销隐藏为未优化Python路径。
  2. 分别测量预填充与解码阶段。W4A4的主要收益通常来自降低权重读取和矩阵乘成本,但在线激活变换可能更直接影响低批量解码延迟。
  3. 使用业务分布而非通用WikiText校准,并同时评估困惑度、任务准确率、输出稳定性与长上下文退化。
  4. 对MoE模型重点检查专家激活分布、路由稳定性和负载不均衡;路由器的微小数值偏移可能比普通线性层误差更敏感。
  5. 保留纯FP16、BF16、纯MXFP4以及现有W4A16方案作为基线,同时报告峰值显存、吞吐、首Token延迟和生成速度,而非只比较模型文件大小。

MACaQ的真正贡献,是用局部Hadamard坐标变换和吸收逆变换换取跨层解耦,再用流式执行把超大模型量化带入更可控的内存预算。能否从论文指标转化为稳定的生产收益,最终取决于量化内核实现、硬件支持和端到端工作负载测试。


消息来源:Hugging Face Daily Papers (2026-11-07)

—— 完 ——

主题标签: #MACaQ #W4A4量化 #MXFP4 #Hadamard变换 #后训练量化

编辑部与事实核查说明:本篇专刊由智客前沿资讯管线根据官方技术公告、学术论文及开源工程文档整理编译,经算法实体核验与人工编辑审校后发布。若发现技术事实纰漏,欢迎依据勘误方针或一键向编辑部发送勘误反馈。

社区整体评论区

正在加载实时智能评论与划词标注…