arXiv AI

SSDi8: Accurate and Efficient 8-bit Quantization for State Space Duality

arXiv Machine Learning
Aug 27

Beyond Dense Adam States: Adaptive Log-Space Quantization for Memory-Efficient Optimizers

The paper introduces Adaptive Log‑Space (AL) quantization, a block‑wise representation that adapts the non‑zero range per block and preserves the exact‑zero invariant for non‑negative optimizer states. AL8 and AL16 are combined with signed‑momentum encodings and state‑specific precision choices, rather than a single policy for all states. Experiments on TinyLlama‑1.1B and GPT‑2 show that AL‑based quantization can match or exceed full‑precision performance while dramatically reducing optimizer‑state storage.

By Yan Wang