Analysis of biologically plausible neuron models for regression with spiking neural networks
Mario De Florio, Adar Kahana, George Em Karniadakis
TL;DR
The paper addresses the challenge of using Spiking Neural Networks (SNNs) for regression in Scientific Machine Learning by comparing four biologically plausible membrane models—Leaky Integrate-and-Fire, FitzHugh-Nagumo, Izhikevich, and Hodgkin-Huxley—under two numerical solvers, Euler and Runge–Kutta 4th order. It demonstrates that more expressive membrane models (IZH, HH) yield lower regression error while producing fewer spikes, albeit with higher per-step computation. The study uses a shallow SNN with a single membrane layer and a trainable synapse, optimized by backpropagation on the output layer, and evaluates performance on three functions with varied smoothness, both with and without noise. The findings highlight a trade-off between accuracy and energy usage, suggesting that biologically faithful neurons can improve SciML SNN performance and reduce energy consumption on neuromorphic hardware, guiding model selection for regression tasks.
Abstract
This paper explores the impact of biologically plausible neuron models on the performance of Spiking Neural Networks (SNNs) for regression tasks. While SNNs are widely recognized for classification tasks, their application to Scientific Machine Learning and regression remains underexplored. We focus on the membrane component of SNNs, comparing four neuron models: Leaky Integrate-and-Fire, FitzHugh-Nagumo, Izhikevich, and Hodgkin-Huxley. We investigate their effect on SNN accuracy and efficiency for function regression tasks, by using Euler and Runge-Kutta 4th-order approximation schemes. We show how more biologically plausible neuron models improve the accuracy of SNNs while reducing the number of spikes in the system. The latter represents an energetic gain on actual neuromorphic chips since it directly reflects the amount of energy required for the computations.
