NeuZip: Memory-Efficient Training and Inference with Dynamic Compression of Neural Networks
Not on the map. Why?
Abstract (arXiv)
The performance of neural networks improves when more parameters are used. However, the model sizes are constrained by the available on-device memory during training and inference. Although applying techniques like quantization can alleviate the constraint, they suffer from performance degradation. In this work, we introduce NeuZip, a new weight compression scheme based on the entropy of floating-point numbers in neural networks. With NeuZip, we are able to achieve memory-efficient training and inference without sacrificing performance. Notably, we significantly reduce the memory footprint of training a Llama-3 8B model from 31GB to less than 16GB, while keeping the training dynamics fully unchanged. In inference, our method can reduce memory usage by more than half while maintaining near-lossless performance. Our code is publicly available.
Key facts. Cited by 14 papers in the map (data through 14 Oct 2026). Builds on 34 of its 45 references that are in the map. First posted on arXiv 28 Oct 2024.
Builds on
34 of 45 references.
References · 34
- 2017Attention Is All You Needcited by 60,361
- 2020Language Models are Few-Shot Learnerscited by 32,419
- 2024The Llama 3 Herd of Modelscited by 14,715
- 2019Exploring the Limits of Transfer Learning with a Unified Text-to-Text Transformercited by 14,599
- 2021LoRA: Low-Rank Adaptation of Large Language Modelscited by 14,380
- 2023Llama 2: Open Foundation and Fine-Tuned Chat Modelscited by 13,028
- 2015Distilling the Knowledge in a Neural Networkcited by 10,324
- 2020Measuring Massive Multitask Language Understandingcited by 7,963
- 2020Scaling Laws for Neural Language Modelscited by 6,586
- 2019Parameter-Efficient Transfer Learning for NLPcited by 4,236
- 2015Delving Deep into Rectifiers: Surpassing Human-Level Performance on ImageNet Classificationcited by 3,611
- 2023QLoRA: Efficient Finetuning of Quantized LLMscited by 3,418
22 more
- 2016Pointer Sentinel Mixture Modelscited by 3,096
- 2015Deep Compression: Compressing Deep Neural Networks with Pruning, Trained Quantization and Huffman Codingcited by 2,329
- 2015Learning both Weights and Connections for Efficient Neural Networkscited by 2,031
- 2018A Call for Clarity in Reporting BLEU Scorescited by 1,799
- 2022SmoothQuant: Accurate and Efficient Post-Training Quantization for Large Language Modelscited by 1,386
- 2019ZeRO: Memory Optimizations Toward Training Trillion Parameter Modelscited by 1,301
- 2022Scaling Autoregressive Models for Content-Rich Text-to-Image Generationcited by 1,295
- 2021BitFit: Simple Parameter-efficient Fine-tuning for Transformer-based Masked Language-modelscited by 1,245
- 2018Spider: A Large-Scale Human-Labeled Dataset for Complex and Cross-Domain Semantic Parsing and Text-to-SQL Taskcited by 1,139
- 2018Adafactor: Adaptive Learning Rates with Sublinear Memory Costcited by 905
- 2016Training Deep Nets with Sublinear Memory Costcited by 848
- 2024Yi: Open Foundation Models by 01.AIcited by 719
- 2017Seq2SQL: Generating Structured Queries from Natural Language using Reinforcement Learningcited by 687
- 2017The Implicit Bias of Gradient Descent on Separable Datacited by 655
- 2024GaLore: Memory-Efficient LLM Training by Gradient Low-Rank Projectioncited by 463
- 2022The case for 4-bit precision: k-bit Inference Scaling Lawscited by 251
- 2019A Study of BFLOAT16 for Deep Learning Trainingcited by 232
- 2022A Fast Post-Training Pruning Framework for Transformerscited by 162
- 2020Exploring Versatile Generative Language Model Via Parameter-Efficient Transfer Learningcited by 139
- 2023Full Parameter Fine-tuning for Large Language Models with Limited Resourcescited by 134
- 2024Flora: Low-Rank Adapters Are Secretly Gradient Compressorscited by 108
- 2020Implicit Regularization in ReLU Networks with the Square Losscited by 45
Built on by
Citing papers per year: 2025: 3; 2026: 11.
14 papers.
Sort:
By area:
- 202570% Size, 100% Accuracy: Lossless LLM Compression for Efficient GPU Inference via Dynamic-Length Float (DFloat11)cited by 24
- 2026ZipServ: Fast and Memory-Efficient LLM Inference with Hardware-Aware Lossless Compressioncited by 8
- 2025Scalable Parameter and Memory Efficient Pretraining for LLM: Recent Algorithmic Advances and Benchmarkingcited by 5
- 2026ZipMoE: Efficient On-Device MoE Serving via Lossless Compression and Cache-Affinity Schedulingcited by 4
- 2026Float8@2bits: Entropy Coding Enables Data-Free Model Compressioncited by 2
- 2026UCCL-Zip: Lossless Compression Supercharged GPU Communicationcited by 2
- 2026ZipCCL: Efficient Lossless Data Compression of Communication Collectives for Accelerating LLM Trainingcited by 1
- 2026Lossless Tensor Compression as Program Synthesiscited by 1
- 2025Reducing Storage of Pretrained Neural Networks by Rate-Constrained Quantization and Entropy Codingcited by 0
- 2026The Thermodynamic Costs of Simple Linear Regressioncited by 0
- 2026Transforms for LLM Quantization: The Great Inversion and Format Co-Designcited by 0
- 2026CIERA: Cross-Iteration Exponent Reuse for Lossless Allgather in Sharded MoE Trainingcited by 0
- 2026NeuralZip: Reusable Setup for Fast Lossless Compression7 Oct 2026
- 2026EntroPack: Fast and Accurate Entropy-Coded Weight Compression at Arbitrary Bitrates28 Sep 2026
- 2026CIERA: Cross-Iteration Exponent Reuse for Lossless Allgather in Sharded MoE Training4 Sep 2026
- 2026Transforms for LLM Quantization: The Great Inversion and Format Co-Design25 Aug 2026
- 2026Lossless Tensor Compression as Program Synthesis3 Aug 2026
- 2026The Thermodynamic Costs of Simple Linear Regression18 May 2026
- 2026ZipCCL: Efficient Lossless Data Compression of Communication Collectives for Accelerating LLM Training30 Apr 2026
- 2026UCCL-Zip: Lossless Compression Supercharged GPU Communication19 Apr 2026
- 2026ZipServ: Fast and Memory-Efficient LLM Inference with Hardware-Aware Lossless Compression18 Mar 2026
- 2026Float8@2bits: Entropy Coding Enables Data-Free Model Compression30 Jan 2026
- 2026ZipMoE: Efficient On-Device MoE Serving via Lossless Compression and Cache-Affinity Scheduling29 Jan 2026
- 2025Scalable Parameter and Memory Efficient Pretraining for LLM: Recent Algorithmic Advances and Benchmarking28 May 2025
- 2025Scalable Parameter and Memory Efficient Pretraining for LLM: Recent Algorithmic Advances and Benchmarkingcited by 5
- 2026Float8@2bits: Entropy Coding Enables Data-Free Model Compressioncited by 2
- 2026UCCL-Zip: Lossless Compression Supercharged GPU Communicationcited by 2
- 2026ZipCCL: Efficient Lossless Data Compression of Communication Collectives for Accelerating LLM Trainingcited by 1
- 2026Lossless Tensor Compression as Program Synthesiscited by 1
- 2025Reducing Storage of Pretrained Neural Networks by Rate-Constrained Quantization and Entropy Codingcited by 0
- 2026The Thermodynamic Costs of Simple Linear Regressioncited by 0
- 2026Transforms for LLM Quantization: The Great Inversion and Format Co-Designcited by 0
- 2026EntroPack: Fast and Accurate Entropy-Coded Weight Compression at Arbitrary Bitratescited by 0
- 2026NeuralZip: Reusable Setup for Fast Lossless Compressioncited by 0
- 2026NeuralZip: Reusable Setup for Fast Lossless Compression7 Oct 2026
- 2026EntroPack: Fast and Accurate Entropy-Coded Weight Compression at Arbitrary Bitrates28 Sep 2026
- 2026Transforms for LLM Quantization: The Great Inversion and Format Co-Design25 Aug 2026
- 2026Lossless Tensor Compression as Program Synthesis3 Aug 2026
- 2026The Thermodynamic Costs of Simple Linear Regression18 May 2026
- 2026ZipCCL: Efficient Lossless Data Compression of Communication Collectives for Accelerating LLM Training30 Apr 2026
- 2026UCCL-Zip: Lossless Compression Supercharged GPU Communication19 Apr 2026
- 2026Float8@2bits: Entropy Coding Enables Data-Free Model Compression30 Jan 2026
- 2025Scalable Parameter and Memory Efficient Pretraining for LLM: Recent Algorithmic Advances and Benchmarking28 May 2025
- 2025Reducing Storage of Pretrained Neural Networks by Rate-Constrained Quantization and Entropy Coding24 May 2025
Similar, not cited
Close in wording, no citation either way
- 2025EntroLLM: Entropy Encoded Weight Compression for Efficient Large Language Model Inference on Edge Devices
- 2023Bandwidth-efficient Inference for Neural Image Compression
- 2026Kilobyte Models: Neural Networks as a Seed and a Quantized Latent
- 2023End-to-End Neural Network Compression via fracℓ1ℓ2 Regularized Latency Surrogates
- 2026FlashOptim: Optimizers for Memory-Efficient Training
- 2024Efficient Neural Compression with Inference-time Decoding
- 2021SmartDeal: Re-Modeling Deep Network Weights for Efficient Inference and Training
- 2024Dynamic Memory Compression: Retrofitting LLMs for Accelerated Inference