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