arXiv:2605.24391cs.ARcs.AI2026-05中稿 · DATE 2026

提出可同时支持训练与推理的新型量化格式,提升精度与能效。

MX-SAFE: Versatile Inference- and Training-Proof Microscaling Format with On-the-Fly Exponent and Mantissa Bit Allocation

论文配图:MX-SAFE: Versatile Inference- and Training-Proof Microscaling Format with On-the-Fly Exponent and Mantissa Bit Allocation
图 1 · 摘自论文原文
  • 动态分配指数与尾数位数,灵活适配不同计算场景。
  • 相比现有格式,推理与训练精度平均提升0.05%至11.1%。
  • 硬件设计优化减少重量化负担,能效降低24.9%。

随着深度学习需求增长,量化已成为降低成本的关键手段,尤其在训练与推理中。2022年,开放计算项目(OCP)标准化了用于深度学习的窄精度格式——微缩放(MX)格式。该格式是一种硬件友好的动态量化方案,通过共享8位指数以有效压缩数据大小。MX格式分为两类:(i) MXINT仅使用尾数位实现高精度;(ii) MXFP通过允许局部指数位实现更宽动态范围。本文提出一种多功能的MXFP格式——MX-SAFE(简称MXSF),可自适应切换两种模式:(FP8 E2M5)更宽尾数模式与子正常数(FP5 E3M2)模式,兼顾训练与直接转换推理。此外,我们设计了基于块的瓷砖结构,减轻训练中重量化负担,提升硬件效率。采用该格式后,相较于MXFP8 E2M5与MXFP8 E4M3,在推理与全量训练上平均精度分别提升0.05%/11.1%与3.55%/3.57%。同时,我们构建了一款支持MXSF格式的训练-推理加速器,其精度接近BF16基线,但总能耗降低24.9%。

原文摘要 · Abstract (English)

As the demand for deep learning grows, cost reduction through quantization has become essential for both training and inference. In 2022, the Open Compute Project (OCP) consortium standardized narrow precision formats for deep learning, called the microscaling (MX) format. The MX format is a hardware-friendly dynamic quantization scheme that effectively reduces the data size by sharing an 8-bit exponent across multiple operands. The MX format can be categorized into two types with their own strengths: (i) MXINT which focuses on a high precision consisting only of mantissa bits and (ii) MXFP which focuses on a wider dynamic range by allowing local exponent bits. In this work, we present a versatile MXFP format, called MX-SAFE (MXSF in short), that adaptively uses two modes, i.e., a wider mantissa mode (FP8 E2M5) and a subnormal FP mode (FP5 E3M2), to support both training and direct-cast inference. Furthermore, we propose a tile-based block design to increase hardware efficiency by reducing the burden of re-quantization process during the training with the MXSF format. Owing to the use of the proposed MXSF format, 0.05%/11.1% and 3.55%/3.57% improvements in accuracy, on average, for inference/full-training compared to MXFP8 E2M5 and MXFP8 E4M3 are observed, respectively. Moreover, we present a training-inference accelerator that supports the MXSF format and it achieves similar accuracy to the BF16 baseline while using 24.9% less total energy consumption.

量化硬件加速训练推理一体低功耗

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