Language Modeling on a SpiNNaker2 Neuromorphic Chip

Research output: Contribution to book/conference proceedings/anthology/reportConference contributionContributedpeer-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 languageEnglish
Title of host publication2024 IEEE 6th International Conference on AI Circuits and Systems, AICAS 2024 - Proceedings
PublisherIEEE
Pages492-496
Number of pages5
ISBN (electronic)9798350383638
ISBN (print)979-8-3503-8364-5
Publication statusPublished - 25 Apr 2024
Peer-reviewedYes

Conference

Title6th IEEE International Conference on Artificial Intelligence Circuits and Systems
SubtitleCircuits and Systems for Neuro-Inspired, Cognitive, and Learning Abilities
Abbreviated titleIEEE AICAS 2024
Conference number6
Duration22 - 25 April 2024
Website
LocationKhalifa University
CityAbu Dhabi
CountryUnited Arab Emirates

External IDs

Scopus 85195072034
ORCID /0000-0001-8525-8702/work/168206512

Keywords

Sustainable Development Goals

Keywords

  • Energy efficient, Language model, Neuromorphic, Sparse activity, Sparse weights