Mixed-TD: efficient neural network accelerator with layer-specific tensor decomposition
OA Location
Author(s)
Yu, Zhewen
Bouganis, Christos-Savvas
Type
Conference Paper
Abstract
Neural Network designs are quite diverse, from VGG-style to ResNet-style, and from Convolutional Neural Networks to Transformers. Towards the design of efficient accelerators, many works have adopted a dataflow-based, inter-layer pipelined architecture, with a customized hardware towards each layer, achieving ultra high throughput and low latency. The deployment of neural networks to such dataflow architecture accelerators is usually hindered by the available on-chip memory as it is desirable to preload the weights of neural networks on-chip to maximise the system performance. To address this, networks are usually compressed before the deployment through methods such as pruning, quantization and tensor decomposition. In this paper, a framework for mapping CNNs onto FPGAs based on a novel tensor decomposition method called Mixed-TD is proposed. The proposed method applies layer-specific Singular Value Decomposition (SVD) and Canonical Polyadic Decomposition (CPD) in a mixed manner, achieving 1.73× to 10.29× throughput per DSP to state-of-the-art CNNs. Our work is open-sourced: https://github.com/Yu-Zhewen/Mixed-TD.
Date Issued
2023-11-02
Date Acceptance
2023-09-04
Citation
2023 33rd International Conference on Field-Programmable Logic and Applications (FPL), 2023
ISBN
979-8-3503-4151-5
ISSN
1946-1488
Publisher
IEEE
Journal / Book Title
2023 33rd International Conference on Field-Programmable Logic and Applications (FPL)
Sponsor
Engineering & Physical Science Research Council (EPSRC)
Identifier
http://dx.doi.org/10.1109/fpl60245.2023.00036
Grant Number
EP/S030069/1
Source
2023 33rd International Conference on Field-Programmable Logic and Applications (FPL)
Publication Status
Published
Start Date
2023-09-04
Finish Date
2023-09-08
Coverage Spatial
Gothenburg, Sweden
Date Publish Online
2023-11-02