Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Microscaling Data Formats for Deep Learning

About

Narrow bit-width data formats are key to reducing the computational and storage costs of modern deep learning applications. This paper evaluates Microscaling (MX) data formats that combine a per-block scaling factor with narrow floating-point and integer types for individual elements. MX formats balance the competing needs of hardware efficiency, model accuracy, and user friction. Empirical results on over two dozen benchmarks demonstrate practicality of MX data formats as a drop-in replacement for baseline FP32 for AI inference and training with low user friction. We also show the first instance of training generative language models at sub-8-bit weights, activations, and gradients with minimal accuracy loss and no modifications to the training recipe.

Bita Darvish Rouhani, Ritchie Zhao, Ankit More, Mathew Hall, Alireza Khodamoradi, Summer Deng, Dhruv Choudhary, Marius Cornea, Eric Dellinger, Kristof Denolf, Stosic Dusan, Venmugil Elango, Maximilian Golub, Alexander Heinecke, Phil James-Roxby, Dharmesh Jani, Gaurav Kolhe, Martin Langhammer, Ada Li, Levi Melnick, Maral Mesmakhosroshahi, Andres Rodriguez, Michael Schulte, Rasoul Shafipour, Lei Shao, Michael Siu, Pradeep Dubey, Paulius Micikevicius, Maxim Naumov, Colin Verrilli, Ralph Wittig, Doug Burger, Eric Chung• 2023

Related benchmarks

TaskDatasetResultRank
Language ModelingC4
Perplexity15.49
1688
Language UnderstandingMMLU
Accuracy66.16
844
Language ModelingWikiText
PPL9.75
740
Long-context language modelingLongBench
Average Score35.14
328
Language Model EvaluationWinogrande, ARC-C, ARC-E, Lambada, PIQA, Hellaswag, MMLU, IFEval, and GSM8K-CoT (Mixed standard 10-shot prompt)
Accuracy80.37
88
Language ModelingWikiText-103
PPL3.69
42
Zero-shot Language ModelingLM Evaluation Harness 0-shot
WG76.32
30
LLM Inference PerformanceLlama-3-8B
TTFT (ms)56.03
12
Quantization Distribution EvaluationC4 (calibration set)
KL Divergence (Top 10)0.0543
11
Quantization Distribution EvaluationWiki2 (calibration set)
KL Divergence0.1143
11
Showing 10 of 18 rows

Other info

Follow for update