Skip to main navigation Skip to search Skip to main content

Neural Network Compression via Learnable Wavelet Transforms

  • Moritz Wolter*
  • , Shaohui Lin
  • , Angela Yao
  • *Corresponding author for this work
  • University of Bonn
  • Fraunhofer Institute for Algorithms and Scientific Computing
  • National University of Singapore

Research output: Chapter in Book/Report/Conference proceedingConference contributionpeer-review

Abstract

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.

Original languageEnglish
Title of host publicationArtificial Neural Networks and Machine Learning – ICANN 2020 - 29th International Conference on Artificial Neural Networks, Proceedings
EditorsIgor Farkaš, Paolo Masulli, Stefan Wermter
PublisherSpringer Science and Business Media Deutschland GmbH
Pages39-51
Number of pages13
ISBN (Print)9783030616151
DOIs
StatePublished - 2020
Externally publishedYes
Event29th International Conference on Artificial Neural Networks, ICANN 2020 - Bratislava, Slovakia
Duration: 15 Sep 202018 Sep 2020

Publication series

NameLecture Notes in Computer Science (including subseries Lecture Notes in Artificial Intelligence and Lecture Notes in Bioinformatics)
Volume12397 LNCS
ISSN (Print)0302-9743
ISSN (Electronic)1611-3349

Conference

Conference29th International Conference on Artificial Neural Networks, ICANN 2020
Country/TerritorySlovakia
CityBratislava
Period15/09/2018/09/20

Keywords

  • Network compression
  • Wavelets

Fingerprint

Dive into the research topics of 'Neural Network Compression via Learnable Wavelet Transforms'. Together they form a unique fingerprint.

Cite this