🌙

脉冲神经网络中的有效迁移学习算法

Effective Transfer Learning Algorithm in Spiking Neural Networks

IEEE Transactions on Cybernetics · 2021
被引 30
ABS 3

中文导读

针对脉冲神经网络训练数据昂贵的问题,提出基于域不变表示的迁移学习框架,在多个数据集上验证了特征可迁移性和中心核对齐的有效性。

Abstract

As the third generation of neural networks, spiking neural networks (SNNs) have gained much attention recently because of their high energy efficiency on neuromorphic hardware. However, training deep SNNs requires many labeled data that are expensive to obtain in real-world applications, as traditional artificial neural networks (ANNs). In order to address this issue, transfer learning has been proposed and widely used in traditional ANNs, but it has limited use in SNNs. In this article, we propose an effective transfer learning framework for deep SNNs based on the domain in-variance representation. Specifically, we analyze the rationality of centered kernel alignment (CKA) as a domain distance measurement relative to maximum mean discrepancy (MMD) in deep SNNs. In addition, we study the feature transferability across different layers by testing on the Office-31, Office-Caltech-10, and PACS datasets. The experimental results demonstrate the transferability of SNNs and show the effectiveness of the proposed transfer learning framework by using CKA in SNNs.

脉冲神经网络迁移学习深度学习计算机视觉