arXiv Machine Learning

OrBIT: Structure-Guided Embedding Compression

OrBIT is a structure‑guided embedding compression framework that learns reusable local geometry from orbit dynamics to constrain a small set of shared codewords. It allocates a fixed coding budget based on a global reconstruction residual, allowing redundant overlapping charts to compensate local errors after gluing. Across four large language model embedding tables, OrBIT achieves 37.9× compression on GPT‑2 and over 23× on each 7B table compared to 16‑bit storage, while maintaining competitive rate‑distortion performance against existing quantization and low‑rank methods.

arXiv Machine Learning
Oct 16

N-Gram Memories at Two Bits: Quantizing Lookup Tables in Large Language Models

The paper introduces Conditional Program (CP), a compressed representation for large language model memory tables that reconstructs only the rows requested by an input. CP encodes each row with a short two‑bit code and a local scale, using a shared reconstruction program based on a Gaussian trellis. Experiments on three n‑gram memory families show CP reduces output KL divergence by 41–63% compared to 2‑bit round‑to‑nearest, and for DeepSeek it cuts table storage from 203 GB to 51 GB, enabling larger workloads on GPU.

By Donghoon Han, Minseop Jung, Dong Woo Kwak, SeungJae Lee, Junhoo Lee
arXiv Computation and Language
1d ago

Low-Bit N-Gram Embeddings for Language Models

The paper investigates reducing memory usage of n‑gram embedding tables in open‑weight language models by storing them in low precision. Experiments on nanochat, Llama, and GPT‑2 families show that four‑bit tables cut table memory by about 75% while retaining over 99% of the validation‑loss improvement. The study also demonstrates that re‑allocating the saved memory to larger tables can surpass the performance of full‑precision tables, and that similar savings are achievable in large models like DeepSeek‑V4.1‑Flash and Qwen3.8‑Flash‑Next with minimal impact on perplexity and downstream tasks.

By Stephen Chung, Wenyu Du
arXiv AI
Sep 10

FastE: Readout-Triggered Token Compression for LLM Embedding Inference

FastE is a training‑free, plug‑and‑play method that compresses token prefixes in large language model (LLM) embedding inference. It uses a shared fixed threshold on batch‑mean readout‑prefix alignment to decide when to compress and ranks prefix states by readout attention scores to keep the most important ones. Experiments on Qwen3‑Embedding models show that FastE can cut decoder‑backbone FLOPs by over 40% while preserving more than 99% of the original ranking quality across multiple benchmarks and tasks.

By Jinsong Shu, Jinyong Wen, Baokun Wang, Zhongle Xie, Lidan Shou, Weiqiang Wang, Gang Chen
arXiv Computation and Language
Oct 12

zip2zip++: Towards Inference-Time Token Compression with Minimal Quality Loss

The paper introduces zip2zip++, a method for inference‑time token compression that reduces token counts using a dynamic Lempel‑Ziv‑Welch (LZW) codec while preserving language model quality. It identifies four problems with prior LZW‑based compression—embedding conflict, improper RoPE application, harmful digit merges, and segmentation ambiguity—and proposes solutions: separate input/output embeddings, size‑aware RoPE, and a selective‑merge LZW codec. Experiments on four models show that these changes halve the quality gap in perplexity and reasoning accuracy compared to the uncompressed baseline, achieving similar compression ratios.

By Saibo Geng, Andrea Gentilini, Xinxian Ma, Nathan Ranchin, Yunzhen Yao, Robert West
arXiv Computer Vision
Sep 25

Towards Practical Compression of 3D Gaussian Splatting

The paper introduces COSA-GS, a new compression method for 3D Gaussian Splatting that avoids spatial aggregation by using anchor-wise causal factorization. It builds a compact learnable anchor latent from geometry context and fuses it with the geometry context to create an anchor context for attribute coding, employing only linear transformations and activations. The method is trained with rate–distortion optimization, adaptive Gaussian pruning, and quantization-aware training to ensure bit‑exact entropy decoding across platforms, achieving state‑of‑the‑art compression performance with fast, consistent cross‑platform decoding.

By Pengpeng Yu, Yueru Chen, Fei Song, Tai Qin, Qi Zhang, Jing Wang, Yulan Guo
arXiv AI
Jun 9

End-to-End Context Compression at Scale

arXiv:2606. 09659v1 Announce Type: cross Abstract: Long-context language model inference is bottlenecked by memory, as the KV cache grows with context length.

By Ang Li, Sean McLeish, Haozhe Chen, Nimit Kalra, Zaiqian Chen, Artem Gazizov, Venkata Anoop Suhas Kumar Morisetty, Bhavya Kailkhura, Harshitha Menon, Zhuang Liu, Brian R. Bartoldson, Tom Goldstein, Sanae Lotfi, Micah Goldblum, Pavel Izmailov