MTST: A Multi-Task Scheduling Transformer Accelerator for Edge Computing

October 29, 2024·
Zongcheng Yue
,
Dongwei Yan
,
Ran Wu
Dr Sean Longyu Ma
Dr Sean Longyu Ma
,
Chiu-Wing Sham
· 0 min read
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)
publications
Dr Sean Longyu Ma
Authors
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.