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 | Institute of Electrical and Electronics Engineers (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