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
Compuertas de Escala de Tiempo Local para Redes Neuronales de Picos Continuas Robustas a Escala de Tiempo
Las redes neuronales de picos (SNNs) poseen ventajas de eficiencia energética en hardware neuromórfico, pero tienen un desempeño deficiente en tareas que requieren adaptación rápida y memoria a largo plazo, particularmente en escenarios de aprendizaje continuo. Este artículo propone Compuertas de Escala de Tiempo Local (LT-Gate), un modelo neuronal que combina dinámicas de doble constante de tiempo y mecanismos de compuerta adaptativa. Cada neurona de pico rastrea en paralelo información de escala de tiempo rápida y lenta, regulando localmente su influencia mediante mecanismos de compuerta aprendidos. Este diseño permite que las neuronas individuales mantengan información de contexto lento mientras responden a señales rápidas, abordando el dilema estabilidad-plasticidad. El estudio también introduce regularización de seguimiento de varianza inspirada en homeostasis biológica para estabilizar la actividad de disparo. Los experimentos demuestran que LT-Gate alcanza aproximadamente 51% de precisión final en puntos de referencia de clasificación temporal, una mejora significativa respecto al 46% de la línea base de aprendizaje continuo Hebbiano reciente (HLOP).
Olvido Catastrófico en Aprendizaje Continuo: Las SNNs, como las redes neuronales tradicionales, olvidan el conocimiento previamente adquirido al aprender nuevas tareas
Dilema Estabilidad-Plasticidad: Necesidad de equilibrio entre mantener la estabilidad del conocimiento adquirido y la plasticidad para aprender nuevo conocimiento
Adaptabilidad de Escala de Tiempo: Los métodos existentes tienen dificultades para manejar tareas que requieren respuestas en diferentes escalas de tiempo
Propuesta del Modelo Neuronal LT-Gate: Combina dinámicas de doble escala de tiempo y mecanismos de compuerta local, implementando asignación flexible de crédito temporal
Diseño de Mecanismo de Regularización de Varianza: Basado en teoría de homeostasis biológica, mantiene activaciones neuronales en rango saludable, previniendo inestabilidad durante cambios de tarea
Logro de Mejora Significativa de Desempeño: Mejora sustancial del desempeño de aprendizaje continuo en tareas heterogéneas temporales, sin requerir memoria externa o crecimiento arquitectónico
Compatibilidad con Hardware Neuromórfico: Mapeo eficiente al chip Intel Loihi 2, soportando operaciones locales de filtrado y compuerta para aprendizaje en chip
La investigación se enfoca en tareas de aprendizaje incremental en dominio temporal: la red debe realizar aprendizaje continuo entre dominios de entrada en diferentes escalas de tiempo, como cambiar de dominio de entrada de alta frecuencia (1000Hz) a baja frecuencia (50Hz), mientras mantiene capacidad de clasificación en dominios anteriores.
Mecanismo de Compuerta Distribuida: A diferencia de unidades de compuerta explícitas en LSTM, el mecanismo de compuerta LT-Gate se localiza en cada estado neuronal de pico
Proceso Dual de Homeostasis: Dinámicas rápidas responden cambios de entrada, dinámicas lentas y compuerta proporcionan retroalimentación negativa, manteniendo estadísticas de tasa de disparo general
Diseño Consciente del Hardware: Todos los cálculos (decaimiento, acumulación, umbral, compuerta) son locales a cada neurona o sinapsis, sin requerir coordinación global
Asignación Adaptativa de Constante de Tiempo: Mediante aprendizaje de γ, la red puede distribuir espectro efectivo de constante de tiempo entre neuronas
Después de entrenamiento solo en 1000Hz, LT-Gate alcanza 48% de precisión en dominio lento no entrenado (10Hz), mientras que SNN estándar solo alcanza 20%, demostrando fuerte capacidad de generalización entre dominios.
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.