Local Timescale Gates for Timescale-Robust Continual Spiking Neural Networks
Tiwari, Chauhan
Spiking neural networks (SNNs) promise energy-efficient artificial intelligence on neuromorphic hardware but struggle with tasks requiring both fast adaptation and long-term memory, especially in continual learning. We propose Local Timescale Gating (LT-Gate), a neuron model that combines dual time-constant dynamics with an adaptive gating mechanism. Each spiking neuron tracks information on a fast and a slow timescale in parallel, and a learned gate locally adjusts their influence. This design enables individual neurons to preserve slow contextual information while responding to fast signals, addressing the stability-plasticity dilemma. We further introduce a variance-tracking regularization that stabilizes firing activity, inspired by biological homeostasis. Empirically, LT-Gate yields significantly improved accuracy and retention in sequential learning tasks: on a challenging temporal classification benchmark it achieves about 51 percent final accuracy, compared to about 46 percent for a recent Hebbian continual-learning baseline and lower for prior SNN methods. Unlike approaches that require external replay or expensive orthogonalizations, LT-Gate operates with local updates and is fully compatible with neuromorphic hardware. In particular, it leverages features of Intel's Loihi chip (multiple synaptic traces with different decay rates) for on-chip learning. Our results demonstrate that multi-timescale gating can substantially enhance continual learning in SNNs, narrowing the gap between spiking and conventional deep networks on lifelong-learning tasks.
academic
Portes d'Échelle Temporelle Locale pour les Réseaux de Neurones Impulsionnels Robustes aux Échelles Temporelles en Apprentissage Continu
Les réseaux de neurones impulsionnels (SNNs) offrent des avantages énergétiques sur le matériel neuromorphe, mais présentent des performances insuffisantes dans les tâches nécessitant une adaptation rapide et une mémoire à long terme, particulièrement dans les scénarios d'apprentissage continu. Cet article propose les portes d'échelle temporelle locale (LT-Gate), un modèle de neurone combinant des dynamiques à double constante de temps et un mécanisme de porte adaptatif. Chaque neurone impulsionnel suit en parallèle les informations d'échelle temporelle rapide et lente, en régulant localement leur influence par le biais d'une porte apprise. Cette conception permet à chaque neurone de maintenir les informations de contexte lent tout en réagissant aux signaux rapides, résolvant ainsi le dilemme stabilité-plasticité. L'étude introduit également une régularisation de suivi de variance inspirée par l'homéostasie biologique pour stabiliser l'activité de décharge. Les expériences montrent que LT-Gate atteint environ 51% de précision finale sur les repères de classification temporelle, soit une amélioration significative par rapport à la ligne de base récente d'apprentissage continu Hebbian (HLOP) de 46%.
Oubli Catastrophique en Apprentissage Continu: Les SNNs, comme les réseaux de neurones traditionnels, oublient les connaissances précédemment apprises lors de l'apprentissage de nouvelles tâches
Dilemme Stabilité-Plasticité: Nécessité d'équilibrer la stabilité des connaissances acquises et la plasticité pour apprendre de nouvelles connaissances
Adaptabilité aux Échelles Temporelles: Les méthodes existantes ont du mal à traiter les tâches nécessitant des réponses à différentes échelles temporelles
L'informatique neuromorphe suscite un intérêt considérable en raison de ses avantages énergétiques pilotés par les événements
L'apprentissage continu est une technologie clé pour réaliser des systèmes d'IA d'apprentissage tout au long de la vie
Les systèmes nerveux biologiques résolvent ce problème par le biais de multiples processus adaptatifs et d'échelles temporelles intrinsèques différentes
Proposition du modèle de neurone LT-Gate: Combinaison de dynamiques à double échelle temporelle et d'un mécanisme de porte local, permettant une attribution flexible du crédit temporel
Conception d'un mécanisme de régularisation de variance: Basé sur la théorie de l'homéostasie biologique, maintenant l'activation des neurones dans une plage saine et prévenant l'instabilité lors des changements de tâche
Amélioration Significative des Performances: Amélioration considérable des performances d'apprentissage continu sur les tâches hétérogènes temporelles, sans mémoire externe ni croissance architecturale
Compatibilité avec le Matériel Neuromorphe: Mappage efficace sur la puce Intel Loihi 2, supportant le filtrage local et les opérations de porte pour l'apprentissage sur puce
La recherche se concentre sur les tâches d'apprentissage incrémental dans le domaine temporel: le réseau doit effectuer un apprentissage continu entre des domaines d'entrée à différentes échelles temporelles, comme passer d'un domaine d'entrée haute fréquence (1000Hz) à un domaine basse fréquence (50Hz), tout en maintenant la capacité de classification pour les domaines précédents.
Mécanisme de Porte Distribué: Contrairement aux unités de porte explicites des LSTM, le mécanisme de porte de LT-Gate est localisé à chaque état de neurone impulsionnel
Processus de Double Homéostasie: La dynamique rapide répond aux changements d'entrée, tandis que la dynamique lente et la porte fournissent une rétroaction négative, maintenant les statistiques globales du taux de décharge
Conception Consciente du Matériel: Tous les calculs (décroissance, accumulation, seuil, porte) sont locaux à chaque neurone ou synapse, sans nécessiter de coordination globale
Allocation Adaptative de Constantes de Temps: Par l'apprentissage de γ, le réseau peut distribuer un spectre efficace de constantes de temps entre les neurones
Après entraînement uniquement à 1000Hz, LT-Gate atteint 48% de précision sur le domaine lent non entraîné (10Hz), tandis que le SNN standard n'atteint que 20%, démontrant une capacité de généralisation transversale robuste.
1 Bellec et al. "Long short-term memory and learning-to-learn in networks of spiking neurons." NeurIPS 2018.
2 Cannon & Miller. "Stable control of firing rate mean and variance by dual homeostatic mechanisms." J Math Neurosci 2017.
3 Davies et al. "Loihi: A neuromorphic manycore processor with on-chip learning." IEEE Micro 2018.
7 Xiao et al. "Hebbian learning based orthogonal projection for continual learning of snns." ICLR 2024.