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
Локальные вентили временной шкалы для устойчивых к временной шкале непрерывно обучающихся импульсных нейронных сетей
Импульсные нейронные сети (SNNs) обладают преимуществами энергоэффективности на нейроморфном оборудовании, однако показывают низкую производительность в задачах, требующих быстрой адаптации и долгосрочной памяти, особенно в сценариях непрерывного обучения. В данной работе предлагается механизм локальных вентилей временной шкалы (LT-Gate) — модель нейрона, объединяющая динамику с двумя временными константами и адаптивный механизм вентилирования. Каждый импульсный нейрон параллельно отслеживает информацию быстрой и медленной временной шкалы, локально регулируя их влияние посредством обучаемых вентилей. Такая конструкция позволяет отдельному нейрону сохранять информацию медленного контекста при одновременном реагировании на быстрые сигналы, решая дилемму стабильности-пластичности. Исследование также вводит вариационную регуляризацию отслеживания, вдохновленную биологическим гомеостазом, для стабилизации активности разрядов. Экспериментальные результаты показывают, что LT-Gate достигает примерно 51% итоговой точности на эталонах временной классификации, что значительно превышает 46% недавнего базового метода непрерывного обучения на основе Хеббиана (HLOP).
Катастрофическое забывание при непрерывном обучении: SNNs, как и традиционные нейронные сети, забывают ранее полученные знания при обучении новым задачам
Дилемма стабильности-пластичности: необходимо достичь баланса между сохранением стабильности уже полученных знаний и пластичностью при обучении новым знаниям
Адаптивность к временной шкале: существующие методы испытывают трудности при обработке задач, требующих ответов на различных временных шкалах
Предложена модель нейрона LT-Gate: объединяет динамику с двумя временными шкалами и локальный механизм вентилирования, обеспечивая гибкое распределение временного кредита
Разработан механизм вариационной регуляризации: основан на теории биологического гомеостаза, поддерживает активацию нейронов в здоровом диапазоне, предотвращает нестабильность при переключении задач
Достигнуто значительное повышение производительности: значительное улучшение производительности непрерывного обучения на временных гетерогенных задачах без необходимости во внешней памяти или расширении архитектуры
Совместимость с нейроморфным оборудованием: эффективное отображение на микросхему Intel Loihi 2, поддержка локальной фильтрации и операций вентилирования для обучения на кристалле
Исследование сосредоточено на задачах инкрементального обучения во временной области: сеть должна осуществлять непрерывное обучение между входными областями с различными временными шкалами, например переход от высокочастотной (1000 Гц) входной области к низкочастотной (50 Гц) входной области, сохраняя при этом способность классификации для предыдущих областей.
Распределенный механизм вентилирования: в отличие от явных блоков вентилирования LSTM, механизм вентилирования LT-Gate локализован в состояние каждого импульсного нейрона
Двойной гомеостатический процесс: быстрая динамика реагирует на изменения входных данных, медленная динамика и вентилирование обеспечивают отрицательную обратную связь, поддерживая общую статистику частоты разрядов
Дизайн с учетом оборудования: все вычисления (затухание, накопление, пороговое значение, вентилирование) локальны для каждого нейрона или синапса, не требуют глобальной координации
Адаптивное распределение временных констант: посредством обучения γ сеть может распределять эффективный спектр временных констант между нейронами
После обучения только на 1000 Гц, LT-Gate достигает 48% точности на необученной медленной области (10 Гц), в то время как стандартная SNN достигает только 20%, демонстрируя мощную способность обобщения между областями.
Микросхема Intel Loihi поддерживает несколько синаптических следов с различными скоростями затухания, обеспечивая естественную поддержку оборудования для LT-Gate.
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.