Dynamic Activation with Knowledge Distillation for Energy-Efficient Spiking NN Ensembles
Konstantaropoulos, Orestis, Mallios, Theodoris, Papadopouli, Maria
–arXiv.org Artificial Intelligence
--While foundation AI models excel at tasks like classification and decision-making, their high energy consumption makes them unsuitable for energy-constrained applications. Inspired by the brain's efficiency, spiking neural networks (SNNs) have emerged as a viable alternative due to their event-driven nature and compatibility with neuromorphic chips. This work introduces a novel system that combines knowledge distillation and ensemble learning to bridge the performance gap between artificial neural networks (ANNs) and SNNs. A foundation AI model acts as a teacher network, guiding smaller student SNNs organized into an ensemble, called Spiking Neural Ensemble (SNE). SNE enables the disentanglement of the teacher's knowledge, allowing each student to specialize in predicting a distinct aspect of it, while processing the same input. The core innovation of SNE is the adaptive activation of a subset of SNN models of an ensemble, leveraging knowledge-distillation, enhanced with an informed-partitioning (disentanglement) of the teacher's feature space. Moreover, SNE is significantly more efficient than the teacher network, reducing computational requirements by up to 20x with only a 2% drop in accuracy on the CIF AR-10 dataset. This disentanglement procedure achieves an accuracy improvement of up to 2.4% on the CIF AR-10 dataset compared to other partitioning schemes. Finally, we comparatively analyze SNE performance under noisy conditions, demonstrating enhanced robustness compared to its ANN teacher . In summary, SNE offers a promising new direction for energy-constrained applications. Foundation AI is repeatedly breaking ground in computer vision and machine learning [1], [2], with advancements at dramatic speed across various domains, including image and video classification, semantic segmentation, depth estimation, image captioning, and decision-making.
arXiv.org Artificial Intelligence
Feb-19-2025