arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2004.09569cs.LGstat.ML

Neural network compression via learnable wavelet transforms

University of Bonn(波恩大学);Fraunhofer Center for Machine Learning and SCAI(弗劳恩霍夫机器学习与SCAI中心);National University of Singapore(新加坡国立大学)

Moritz Wolter, Shaohui Lin, Angela Yao

英文摘要:

Wavelets are well known for data compression, yet have rarely been applied to the compression of neural networks. This paper shows how the fast wavelet transform can be used to compress linear layers in neural networks. Linear layers still occupy a significant portion of the parameters in recurrent neural networks (RNNs). Through our method, we can learn both the wavelet bases and corresponding coefficients to efficiently represent the linear layers of RNNs. Our wavelet compressed RNNs have significantly fewer parameters yet still perform competitively with the state-of-the-art on synthetic and real-world RNN benchmarks. Wavelet optimization adds basis flexibility, without large numbers of extra weights. Source code is available at https://github.com/v0lta/Wavelet-network-compression.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑