AdaHOP: Fast and Accurate Low-Precision Training via Outlier-Pattern-Aware Rotation
Organizations: Advanced Micro Devices, Inc. · POSTECH
Abstract
Hadamard transforms have become a key tool for stabilizing low-precision training, but existing methods apply them uniformly across tensors and computation paths. We show that this one-size-fits-all strategy is inherently limited: Hadamard smoothing reduces quantization error only when its direction is properly aligned with the operand's outlier structure. Through a systematic study of weights, activations, and gradients in LLM training, we identify three stable outlier patterns, Row-wise, Column-wise, and None, and show that each outlier pattern pair in matrix multiplication requires a distinct transform or outlier-handling strategy. We propose AdaHOP, Adaptive Hadamard transform with Outlier-Pattern-aware strategy, which applies Inner Hadamard Transform (IHT) when inner-dimension mixing properly suppresses the operands' outliers, and selectively applies Outlier Extraction (OE) that extracts dominant outlier rows or columns into a high-precision path when it does not. With fused, hardware-aware Triton kernels, AdaHOP enables training from scratch at MXFP4 precision with BF16-level quality, while achieving up to 3.6X memory compression, 1.46X end-to-end training speedup over BF16.
Figures & tables
| Llama3.2-1B | Instella-3B | Llama3.1-8B | ||||||||||
| Pattern Pair | Total | Fwd | Total | Fwd | Total | Fwd | ||||||
| RN | 23 | 0 | 15 | 8 | 79 | 0 | 33 | 46 | 65 | 0 | 27 | 38 |
| RC | 69 | 0 | 69 | 0 | 111 | 0 | 111 | 0 | 92 | 0 | 92 | 0 |
| NN | 35 | 15 | 0 | 20 | 100 | 36 | 2 | 62 | 103 | 32 | 4 | 67 |
| NC | 20 | 0 | 20 | 0 | 60 | 0 | 60 | 0 | 63 | 0 | 63 | 0 |
| CN | 181 | 97 | 0 | 84 | 361 | 216 | 1 | 144 | 312 | 192 | 1 | 119 |
| Model | Tokens | Method | PIQA | HellaSwag | ARC-E | Lambada | Avg. |
|---|---|---|---|---|---|---|---|
| Llama3.2-1B | 40B | BF16 | 72.69 | 52.33 | 47.85 | 38.17 | 52.76 |
| MXFP4 | 64.69 | 37.94 | 37.29 | 11.93 | 37.96 | ||
| MXFP4+Had. | 71.85 | 51.37 | 45.91 | 37.68 | 51.70 | ||
| Tseng et al. | 72.14 | 50.95 | 46.17 | 36.81 | 51.52 | ||
| HALO | 72.09 | 51.89 | 46.72 | 39.03 | 52.43 | ||
| AdaHOP-Lv1 | 73.04 | 52.24 | 46.91 | 38.21 | 52.60 |
| Method | Mem. | Speed | Accel. | Quality |
|---|---|---|---|---|
| BF16 | 76.00 | 10930 | 1.00 | ✓ |
| MXFP4 | 20.19 | 17640 | 1.61 | ✗ |
| MXFP4+Had. | 20.60 | 16305 | 1.49 | ✗ |
| Tseng et al. | 20.60 | 14881 | 1.36 | ▲ |
| HALO | 20.60 | 14251 | 1.30 | ▲ |
| AdaHOP-Lv1 | 20.94 | 15804 | 1.45 | ✓ |
Appendix figures & tables4 assets
Supplementary material from the paper’s appendix.