基于知识蒸馏的多阶段脉冲神经网络训练方法及装置

By mapping spiking neural networks and artificial neural networks to a unified logits space in stages, calculating the joint loss, and updating the parameters of spiking neural networks, the problems of low training efficiency and poor accuracy of spiking neural networks are solved, achieving efficient and low-cost training results.

CN119129702BActive Publication Date: 2026-07-17ZHEJIANG UNIV

Patent Information

Authority / Receiving Office
CN · China
Patent Type
Patents(China)
Current Assignee / Owner
ZHEJIANG UNIV
Filing Date
2024-09-12
Publication Date
2026-07-17

AI Technical Summary

Technical Problem

Existing spiking neural network training methods are inefficient in deep structures and complex tasks, and their accuracy is significantly lower than that of artificial neural networks, while also suffering from high power consumption.

Method used

A multi-stage spiking neural network training method based on knowledge distillation is adopted, which divides the spiking neural network and artificial neural network into multiple stages, maps them to a unified logits space through fully connected layers, calculates the joint loss based on confidence and features, and uses the backpropagation method to update the network parameters.

Benefits of technology

It improves the training efficiency and accuracy of spiking neural networks, narrows the accuracy gap with artificial neural networks, and reduces the power consumption during the training process.

✦ Generated by Eureka AI based on patent content.

Smart Images

  • Figure CN119129702B_ABST
    Figure CN119129702B_ABST
Patent Text Reader

Abstract

本发明公开了一种基于知识蒸馏的多阶段脉冲神经网络训练方法及装置,该方法通过分阶段将脉冲神经网络的中间输出映射到logits空间中,与和人工神经网络的最终结果分别计算基于置信度的损失和基于特征的样本间损失,最后把各阶段的基于置信度的损失和基于特征的样本间损失整合计算得到总损失函数,从而训练更新脉冲神经网络的网络参数,用于执行图像分类任务。本发明所述的训练方法将人工神经网络和脉冲神经网络的输出映射到统一的logits空间,并且考虑了基于置信度和特征的样本间的联合损失,制定了一个能够让脉冲神经网络更高效从人工神经网络中提取知识的方案,从而进一步缩小了人工神经网络和脉冲神经网络之间的精度的差距。
Need to check novelty before this filing date? Find Prior Art