Language Modeling on a SpiNNaker2 Neuromorphic Chip
Research output: Contribution to book/conference proceedings/anthology/report › Conference contribution › Contributed › peer-review
Contributors
Abstract
As large language models continue to scale in size rapidly, so too does the computational power required to run them. Event-based networks on neuromorphic devices offer a potential way to reduce energy consumption for inference significantly. However, to date, most event-based networks that can run on neuromorphic hardware, including spiking neural networks (SNNs), have not achieved task performance even on par with LSTM models for language modeling. As a result, language modeling on neuromorphic devices has seemed a distant prospect. In this work, we demonstrate the first-ever implementation of a language model on a neuromorphic device – specifically the SpiNNaker2 chip – based on a recently published event-based architecture called the EGRU. SpiNNaker2 is a many-core neuromorphic chip designed for large-scale asynchronous processing, and the EGRU is architected to leverage such hardware efficiently while maintaining competitive task performance. This implementation marks the first time a neuromorphic language model matches LSTMs, setting the stage for taking task performance to the level of large language models. We also demonstrate results on a gesture recognition task based on inputs from a DVS camera. Overall, our results showcase the feasibility of this neuro-inspired neural network in hardware, highlighting significant gains versus conventional hardware in energy efficiency for the common use case of single batch inference.
Details
Original language | English |
---|---|
Title of host publication | 2024 IEEE 6th International Conference on AI Circuits and Systems, AICAS 2024 - Proceedings |
Publisher | IEEE |
Pages | 492-496 |
Number of pages | 5 |
ISBN (electronic) | 9798350383638 |
ISBN (print) | 979-8-3503-8364-5 |
Publication status | Published - 25 Apr 2024 |
Peer-reviewed | Yes |
Conference
Title | 6th IEEE International Conference on Artificial Intelligence Circuits and Systems |
---|---|
Subtitle | Circuits and Systems for Neuro-Inspired, Cognitive, and Learning Abilities |
Abbreviated title | IEEE AICAS 2024 |
Conference number | 6 |
Duration | 22 - 25 April 2024 |
Website | |
Location | Khalifa University |
City | Abu Dhabi |
Country | United Arab Emirates |
External IDs
Scopus | 85195072034 |
---|---|
ORCID | /0000-0001-8525-8702/work/168206512 |
Keywords
Sustainable Development Goals
ASJC Scopus subject areas
Keywords
- Energy efficient, Language model, Neuromorphic, Sparse activity, Sparse weights