International Conference on Computer and Knowledge Engineering

Home / 12th International Conference on Computer and Knowledge Engineering

Low-Cost and Hardware Efficient Implementation of Pooling Layers for Stochastic CNN Accelerators

Authors :

Mobin Vaziri¹ Hadi Jahanirad²

1- University of Kurdistan 2- University of Kurdistan

Keywords :

Convolutional neural networks،Stochastic computing،Pooling layers،High-level synthesis tools

Abstract :

With the astonishing achievements of Convolutional Neural Network (CNN) accelerators in real-time applications, the deployment of CNNs on hardware has become an attractive matter. Pooling layers in CNNs are employed for reducing the computation of convolutional layers. Nevertheless, their hardware implementation strategy can impact the accuracy and performance of accelerators. This paper presents a novel parallel Stochastic Computing (SC) based architecture of pooling modules in hardware for stochastic CNN accelerators. With this approach, the SC-based average pooling is reconfigurable with 1.28 times lower power consumption, and the max pooling layer achieves area reduction with the ratio of 4.36. Increasing the accuracy and extending the application of stochastic CNN accelerators in different classification problems is also achieved by implementing AAD pooling with the proposed method.