CAE-ADMM: Implicit Bitrate Optimization via ADMM-based Pruning in Compressive Autoencoders
arXiv:1901.07196
Abstract
We introduce ADMM-pruned Compressive AutoEncoder (CAE-ADMM) that uses Alternative Direction Method of Multipliers (ADMM) to optimize the trade-off between distortion and efficiency of lossy image compression. Specifically, ADMM in our method is to promote sparsity to implicitly optimize the bitrate, different from entropy estimators used in the previous research. The experiments on public datasets show that our method outperforms the original CAE and some traditional codecs in terms of SSIM/MS-SSIM metrics, at reasonable inference speed.
5 pages, 4 figures, 2 tables, 1 algorithm
References in corpus (8)
- Adam: A Method for Stochastic Optimization
- Channel Pruning for Accelerating Very Deep Neural Networks
- Lossy Image Compression with Compressive Autoencoders
- Variable Rate Image Compression with Recurrent Neural Networks
- Real-Time Adaptive Image Compression
- Fast and Accurate Image Super-Resolution with Deep Laplacian Pyramid Networks
- Progressive Weight Pruning of Deep Neural Networks using ADMM
- Learning Convolutional Networks for Content-weighted Image Compression