MTST: A Multi-Task Scheduling Transformer Accelerator for Edge Computing
October 29, 2024·,,
,·
0 min read
Zongcheng Yue
Dongwei Yan
Ran Wu
Dr Sean Longyu Ma
Chiu-Wing Sham
Abstract
Transformer is pivotal in Large Language Models (LLMs), enabling superior performance in language tasks. However, the abundance of parameters poses a challenge for deploying Transformer on resource-constrained edge devices for edge computing, such as Field-Programmable Gate Arrays (FPGAs). To overcome this limitation, we present an FPGA-based architecture that enables high-performance deployment of Transformer for edge computing, incorporating a unique multitask ping-pong scheme. To strike a balance between computing efficiency and logic resource consumption, we adopt a systolic array as a reusable processing engine (PE), employing meticulous space exploration techniques to determine the optimal quantity and dimensions of PEs within a given resource budget for our architecture. We implement the architecture on ZCU 102 FPGA platform, the experimental results illustrate that the proposed accelerator achieves state-of-the-art performance while consuming fewer resources.
Type
Publication
2024 IEEE 13th Global Conference on Consumer Electronics (GCCE)

Authors
Dr Sean Longyu Ma
(he/him)
Lecturer in Computer Science
Sean Longyu Ma is a Lecturer in the School of Computer Science at the
University of Auckland. His research focuses on FPGA-based computing,
RISC-V customisation, high-level synthesis, and heterogeneous computing.