English

Building Efficient Deep Neural Networks with Unitary Group Convolutions

Machine Learning 2019-04-10 v2 Machine Learning

Abstract

We propose unitary group convolutions (UGConvs), a building block for CNNs which compose a group convolution with unitary transforms in feature space to learn a richer set of representations than group convolution alone. UGConvs generalize two disparate ideas in CNN architecture, channel shuffling (i.e. ShuffleNet) and block-circulant networks (i.e. CirCNN), and provide unifying insights that lead to a deeper understanding of each technique. We experimentally demonstrate that dense unitary transforms can outperform channel shuffling in DNN accuracy. On the other hand, different dense transforms exhibit comparable accuracy performance. Based on these observations we propose HadaNet, a UGConv network using Hadamard transforms. HadaNets achieve similar accuracy to circulant networks with lower computation complexity, and better accuracy than ShuffleNets with the same number of parameters and floating-point multiplies.

Keywords

Cite

@article{arxiv.1811.07755,
  title  = {Building Efficient Deep Neural Networks with Unitary Group Convolutions},
  author = {Ritchie Zhao and Yuwei Hu and Jordan Dotzel and Christopher De Sa and Zhiru Zhang},
  journal= {arXiv preprint arXiv:1811.07755},
  year   = {2019}
}

Comments

8 pages, 2 figures