A Novel GPU-Based Efficient Approach for Convolutional Neural Networks with Small Filters

Wenbin Jiang,Yiming Chen,Hai Jin,Ran Zheng,Ye Chi
DOI: https://doi.org/10.1007/s11265-016-1129-2
2016-01-01
Journal of Signal Processing Systems
Abstract:In recent years, convolutional neural networks (CNNs) as important parts of deep neural networks (DNNs) have achieved great successes in the field of computer vision. However, Convolution always takes much computation time in the DNNs. In order to improve the efficiency of CNNs, many solutions focusing on training algorithms and parallelism strategies have been proposed. In this paper, different from traditional GPU-based algorithms, a novel algorithm based on look-up table is proposed to speed up the CNNs with small filters by applying GPU. By transforming complex matrix multiplications operations in the convolution computation to some table-based simple summation operations, the overhead of convolution computation can be considerably reduced. The process of creating a table and looking up values in the table is very appropriate for parallelization on a GPU. The experimental results show that the proposed approach can improve the speed of convolution computation by 20–30 %, compared with existing state-of-the-art works with less accuracy loss.
What problem does this paper attempt to address?