跳到主要导航 跳到搜索 跳到主要内容

Neural Network Compression via Learnable Wavelet Transforms

  • 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 (Source code is available at https://github.com/v0lta/Wavelet-network-compression). Wavelet optimization adds basis flexibility, without large numbers of extra weights.

源语言英语
主期刊名Artificial Neural Networks and Machine Learning – ICANN 2020 - 29th International Conference on Artificial Neural Networks, Proceedings
编辑Igor Farkaš, Paolo Masulli, Stefan Wermter
出版商Springer Science and Business Media Deutschland GmbH
39-51
页数13
ISBN(印刷版)9783030616151
DOI
出版状态已出版 - 2020
已对外发布
活动29th International Conference on Artificial Neural Networks, ICANN 2020 - Bratislava, 斯洛伐克
期限: 15 9月 202018 9月 2020

出版系列

姓名Lecture Notes in Computer Science (including subseries Lecture Notes in Artificial Intelligence and Lecture Notes in Bioinformatics)
12397 LNCS
ISSN(印刷版)0302-9743
ISSN(电子版)1611-3349

会议

会议29th International Conference on Artificial Neural Networks, ICANN 2020
国家/地区斯洛伐克
Bratislava
时期15/09/2018/09/20

指纹

探究 'Neural Network Compression via Learnable Wavelet Transforms' 的科研主题。它们共同构成独一无二的指纹。

引用此