Language Modeling on a SpiNNaker2 Neuromorphic Chip

Publikation: Beitrag in Buch/Konferenzbericht/Sammelband/GutachtenBeitrag in KonferenzbandBeigetragenBegutachtung

Beitragende

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

OriginalspracheEnglisch
Titel2024 IEEE 6th International Conference on AI Circuits and Systems, AICAS 2024 - Proceedings
Herausgeber (Verlag)IEEE
Seiten492-496
Seitenumfang5
ISBN (elektronisch)9798350383638
ISBN (Print)979-8-3503-8364-5
PublikationsstatusVeröffentlicht - 25 Apr. 2024
Peer-Review-StatusJa

Konferenz

Titel6th IEEE International Conference on Artificial Intelligence Circuits and Systems
UntertitelCircuits and Systems for Neuro-Inspired, Cognitive, and Learning Abilities
KurztitelIEEE AICAS 2024
Veranstaltungsnummer6
Dauer22 - 25 April 2024
Webseite
OrtKhalifa University
StadtAbu Dhabi
LandVereinigte Arabische Emirate

Externe IDs

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

Schlagworte

Ziele für nachhaltige Entwicklung

Schlagwörter

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