Very Efficient Training of Convolutional Neural Networks using Fast Fourier Transform and Overlap-and-Add

Tyler Highlander and Andres Rodriguez

Abstract

Convolutional neural networks (CNNs) are currently state-of-the-art for various classification tasks, but are computationally expensive. Propagating through the convolutional layers is very slow, as each kernel in each layer must sequentially calculate many dot products for a single forward and backward propagation which equates to O((N^2)(n^2)) per kernel per layer where the inputs are N x N arrays and the kernels are n x n arrays. Convolution can be efficiently performed as a Hadamard product in the frequency domain. The bottleneck is the transformation which has a cost of O((N^2)log_2(N)) using the fast Fourier transform (FFT). However, the increase in efficiency is less significant when N >> n as is the case in CNNs. We mitigate this by using the ``overlap-and-add'' technique reducing the computational complexity to O((N^2)log_2(n)) per kernel. This method increases the algorithm's efficiency in both the forward and backward propagation, reducing the training and testing time for CNNs. Our empirical results show our method reduces computational time by a factor of up to 50.4 times the traditional convolution implementation.

Session

Poster 2

Files

Extended Abstract (PDF, 197K)
Paper (PDF, 1042K)

DOI

10.5244/C.29.160
https://dx.doi.org/10.5244/C.29.160

Citation

Bibtex

@inproceedings{BMVC2015_160,
	title={Very Efficient Training of Convolutional Neural Networks using Fast Fourier Transform and Overlap-and-Add},
	author={Tyler Highlander and Andres Rodriguez},
	year={2015},
	month={September},
	pages={160.1-160.9},
	articleno={160},
	numpages={9},
	booktitle={Proceedings of the British Machine Vision Conference (BMVC)},
	publisher={BMVA Press},
	editor={Xianghua Xie, Mark W. Jones, and Gary K. L. Tam},
	doi={10.5244/C.29.160},
	isbn={1-901725-53-7},
	url={https://dx.doi.org/10.5244/C.29.160}
}