Unrolling Ternary Neural Networks
Autor: | Martin Kumm, Peter Zipf, Philip H. W. Leong, David Boland, Martin Hardieck, Stephen Tridgell, Duncan J. M. Moss |
---|---|
Jazyk: | angličtina |
Rok vydání: | 2019 |
Předmět: |
Signal Processing (eess.SP)
FOS: Computer and information sciences Computer Science - Machine Learning General Computer Science Computational complexity theory Computer science 02 engineering and technology 01 natural sciences Machine Learning (cs.LG) 0103 physical sciences Datapath FOS: Electrical engineering electronic engineering information engineering 0202 electrical engineering electronic engineering information engineering Neural and Evolutionary Computing (cs.NE) Electrical Engineering and Systems Science - Signal Processing Field-programmable gate array 010302 applied physics Network architecture Artificial neural network Image and Video Processing (eess.IV) Reconfigurability Computer Science - Neural and Evolutionary Computing Electrical Engineering and Systems Science - Image and Video Processing Frame rate 020202 computer hardware & architecture Computer engineering Hardware acceleration |
Popis: | The computational complexity of neural networks for large scale or real-time applications necessitates hardware acceleration. Most approaches assume that the network architecture and parameters are unknown at design time, permitting usage in a large number of applications. This paper demonstrates, for the case where the neural network architecture and ternary weight values are known a priori, that extremely high throughput implementations of neural network inference can be made by customising the datapath and routing to remove unnecessary computations and data movement. This approach is ideally suited to FPGA implementations as a specialized implementation of a trained network improves efficiency while still retaining generality with the reconfigurability of an FPGA. A VGG style network with ternary weights and fixed point activations is implemented for the CIFAR10 dataset on Amazon's AWS F1 instance. This paper demonstrates how to remove 90% of the operations in convolutional layers by exploiting sparsity and compile-time optimizations. The implementation in hardware achieves 90.9 +/- 0.1% accuracy and 122 k frames per second, with a latency of only 29 us, which is the fastest CNN inference implementation reported so far on an FPGA. Accepted in TRETS |
Databáze: | OpenAIRE |
Externí odkaz: |