Applying NSN-Based 3-Bit KV Compression to Looped Transformers — Evaluating Actual Storage, Prediction Quality, and Fused Readout Costs

Can a looped Transformer use a smaller KV cache without losing the time it saves to reconstructing the stored values? We integrated NSN-based 3-bit compression into Ouro-1.4B and Nanbeige4.2-3B-Base, keeping model weights fixed. The implementation tracks keys and values by recurrence and layer, stores actual packed 3-bit codes together with reconstruction metadata, and uses the cache for subsequent predictions. We evaluated four texts per model using teacher forcing: 255 predictions per text for Ouro, and 8 then 256 predictions on the same four texts for Nanbeige. The fused path met the prespecified 5% diagnostic threshold for relative perplexity degradation in every designated window. Final KV storage, including metadata and the uncompressed tail, fell by 67.68% for Ouro and by 64.39% in the extended Nanbeige evaluation. To isolate the cost of materializing a full-length V array, we compared two paths that used the same packed values and the same reconstruction arithmetic. Reconstructing each block inside attention and using it directly in the weighted sum reduced component time by 28.28%. In Nanbeige, initial compression and four subsequent compression updates offset most of the forward-pass savings; total time over 256 steps was approximately equal to the checked uncompressed route. This companion paper provides the implementation and measurement details for the looped-model experiments in the integrated NSN study (DOI: 10.5281/zenodo.23938878). Readable code, experiment records and reproduction instructions: https://github.com/metaSATOKEN/nsn.

Authors

Publication Details

Journal
Zenodo (CERN European Organization for Nuclear Research)
Published
2026-11-06
DOI
https://doi.org/10.5281/zenodo.23970815
Primary Topic
Advanced Neural Network Applications
Type
preprint
Controls
|||
ALL TIME
JAN
FEB
MAR
APR
MAY
JUN
JUL
AUG
SEP
OCT
NOV
preprint

Applying NSN-Based 3-Bit KV Compression to Looped Transformers — Evaluating Actual Storage, Prediction Quality, and Fused Readout Costs

Kentaro Sato
4 links
Zenodo (CERN European Organization for Nuclear Research)
Advanced Neural Network Applications
preprint

Applying NSN-Based 3-Bit KV Compression to Looped Transformers — Evaluating Actual Storage, Prediction Quality, and Fused Readout Costs

Kentaro Sato
preprint en

Abstract

Can a looped Transformer use a smaller KV cache without losing the time it saves to reconstructing the stored values? We integrated NSN-based 3-bit compression into Ouro-1.4B and Nanbeige4.2-3B-Base, keeping model weights fixed. The implementation tracks keys and values by recurrence and layer, stores actual packed 3-bit codes together with reconstruction metadata, and uses the cache for subsequent predictions. We evaluated four texts per model using teacher forcing: 255 predictions per text for Ouro, and 8 then 256 predictions on the same four texts for Nanbeige. The fused path met the prespecified 5% diagnostic threshold for relative perplexity degradation in every designated window. Final KV storage, including metadata and the uncompressed tail, fell by 67.68% for Ouro and by 64.39% in the extended Nanbeige evaluation. To isolate the cost of materializing a full-length V array, we compared two paths that used the same packed values and the same reconstruction arithmetic. Reconstructing each block inside attention and using it directly in the weighted sum reduced component time by 28.28%. In Nanbeige, initial compression and four subsequent compression updates offset most of the forward-pass savings; total time over 256 steps was approximately equal to the checked uncompressed route. This companion paper provides the implementation and measurement details for the looped-model experiments in the integrated NSN study (DOI: 10.5281/zenodo.23938878). Readable code, experiment records and reproduction instructions: https://github.com/metaSATOKEN/nsn.

Zenodo (CERN European Organization for Nuclear Research)
Advanced Neural Network Applications
AI Navigator

Ask Laika to Summarize, Analyze, and Connect papers live on the map.

Summarize Papers & Methodologies

Extract key findings, datasets, and comparative methods across publications.

Benchmark Rankings & Visual Analytics

Rank top research institutions, authors, funders, topics, and journals by Field-Weighted Citation Impact (FWCI) and paper volume with instant charts.

Connect Distant Disciplines

Bridge topological clusters on the map to find hidden collaborative intersections.

Laika access coming soon