arXiv · 2411.06371
LLM Vocabulary Compression for Low-Compute Environments
Abstract
We present a method to compress the final linear layer of language models, reducing memory usage by up to 3.4x without significant performance loss. By grouping tokens based on Byte Pair Encoding (BPE) merges, we prevent materialization of the memory-intensive logits tensor. Evaluations on the TinyStories dataset show that our method performs on par with GPT-Neo and GPT2 while significantly improving throughput by up to 3x, making it suitable for low-compute environments.
Explore related subjects
Keep this discovery
Sreeram Vennam, Anish Joishy, Ponnurangam Kumaraguru. 2024-11-10. LLM Vocabulary Compression for Low-Compute Environments. https://arxiv.org/abs/2411.06371
Cite the original work for its findings. Save a collection to share your selection of sources.