arXiv Machine Learning By Yunied Puig, Amit Kumar Jaiswal

OrBIT: Structure-Guided Embedding Compression

Read the original on arXiv Machine Learning →

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.

Machine-generated by The Flow from the publisher's headline and feed description — not written or checked by a human. The full article lives at arXiv Machine Learning.

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