English

SQS: Bayesian DNN Compression through Sparse Quantized Sub-distributions

Machine Learning 2025-10-13 v1 Artificial Intelligence

Abstract

Compressing large-scale neural networks is essential for deploying models on resource-constrained devices. Most existing methods adopt weight pruning or low-bit quantization individually, often resulting in suboptimal compression rates to preserve acceptable performance drops. We introduce a unified framework for simultaneous pruning and low-bit quantization via Bayesian variational learning (SQS), which achieves higher compression rates than prior baselines while maintaining comparable performance. The key idea is to employ a spike-and-slab prior to inducing sparsity and model quantized weights using Gaussian Mixture Models (GMMs) to enable low-bit precision. In theory, we provide the consistent result of our proposed variational approach to a sparse and quantized deep neural network. Extensive experiments on compressing ResNet, BERT-base, Llama3, and Qwen2.5 models show that our method achieves higher compression rates than a line of existing methods with comparable performance drops.

Keywords

Cite

@article{arxiv.2510.08999,
  title  = {SQS: Bayesian DNN Compression through Sparse Quantized Sub-distributions},
  author = {Ziyi Wang and Nan Jiang and Guang Lin and Qifan Song},
  journal= {arXiv preprint arXiv:2510.08999},
  year   = {2025}
}
R2 v1 2026-07-01T06:28:39.694Z