Co-design of a TinyLLM using Programmable Logic and Software on an FPGA
Large language models (LLMs) provide unique scalable solutions to many natural language processing tasks such as translation, text generation, and content summarization. However, they require significant computing power to achieve high token throughput, which is costly. This work explores low-cost, low-power solutions for implementing LLMs in embedded hardware, describing the design space exploration (DSE) performed and the resulting system architecture. Based on the DSE, we demonstrate an adapted implementation of a quantized LLM using programmable logic (PL) and programmable software (PS) on a ZYNQ FPGA and ARM hybrid system. This TinyLLM retains the ability to produce cohesive text, generating 3.78 tokens/s with an observed 37x speedup over a CPU-based implementation on the same hardware while consuming just under 1.5W of power.
Authors
- Max Theisen (ORCID: https://orcid.org/0000-0001-7955-3801)
- Michael Müller (ORCID: https://orcid.org/0000-0001-7860-163X)
- Alexander Tyshka
- Darrin M. Hanna (ORCID: https://orcid.org/0000-0001-5580-8451)
Institutions
- Oakland University (US)
Publication Details
- Published
- 2024-08-11
- DOI
- https://doi.org/10.1109/mwscas60917.2024.10658754
- Citations
- 3
- Primary Topic
- Embedded Systems Design Techniques
- Type
- article
- Field-Weighted Citation Impact
- 1.47