arXiv · 2506.12000
An Efficient Compression of Deep Neural Network Checkpoints Based on Prediction and Context Modeling
Abstract
This paper is dedicated to an efficient compression of weights and optimizer states (called checkpoints) obtained at different stages during a neural network training process. First, we propose a prediction-based compression approach, where values from the previously saved checkpoint are used for context modeling in arithmetic coding. Second, in order to enhance the compression performance, we also propose to apply pruning and quantization of the checkpoint values. Experimental results show that our approach achieves substantial bit size reduction, while enabling near-lossless training recovery from restored checkpoints, preserving the model's performance and making it suitable for storage-limited environments.
Explore related subjects
Keep this discovery
Explore connections, maps & timelines
Yuriy Kim, Evgeny Belyaev. 2025-06-13. An Efficient Compression of Deep Neural Network Checkpoints Based on Prediction and Context Modeling. https://arxiv.org/abs/2506.12000
Cite the original work for its findings. Save a collection to share your selection of sources.