English

Mixed-TD: Efficient Neural Network Accelerator with Layer-Specific Tensor Decomposition

Machine Learning 2023-06-23 v2 Hardware Architecture

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 customised 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.73x to 10.29x throughput per DSP to state-of-the-art CNNs. Our work is open-sourced: https://github.com/Yu-Zhewen/Mixed-TD

Keywords

Cite

@article{arxiv.2306.05021,
  title  = {Mixed-TD: Efficient Neural Network Accelerator with Layer-Specific Tensor Decomposition},
  author = {Zhewen Yu and Christos-Savvas Bouganis},
  journal= {arXiv preprint arXiv:2306.05021},
  year   = {2023}
}

Comments

accepted by FPL2023

R2 v1 2026-06-28T10:59:44.486Z