Feng Linghao, Zhao Dongcheng, Zeng Yi
Brain-inspired Cognitive Intelligence Lab, Institute of Automation, Chinese Academy of Sciences, Beijing, China; School of Future Technology, University of Chinese Academy of Sciences, China.
Brain-inspired Cognitive Intelligence Lab, Institute of Automation, Chinese Academy of Sciences, Beijing, China; Center for Long-term Artificial Intelligence, China.
Neural Netw. 2024 Oct;178:106423. doi: 10.1016/j.neunet.2024.106423. Epub 2024 Jun 1.
Generative models based on neural networks present a substantial challenge within deep learning. As it stands, such models are primarily limited to the domain of artificial neural networks. Spiking neural networks, as the third generation of neural networks, offer a closer approximation to brain-like processing due to their rich spatiotemporal dynamics. However, generative models based on spiking neural networks are not well studied. Particularly, previous works on generative adversarial networks based on spiking neural networks are conducted on simple datasets and do not perform well. In this work, we pioneer constructing a spiking generative adversarial network capable of handling complex images and having higher performance. Our first task is to identify the problems of out-of-domain inconsistency and temporal inconsistency inherent in spiking generative adversarial networks. We address these issues by incorporating the Earth-Mover distance and an attention-based weighted decoding method, significantly enhancing the performance of our algorithm across several datasets. Experimental results reveal that our approach outperforms existing methods on the MNIST, FashionMNIST, CIFAR10, and CelebA. In addition to our examination of static datasets, this study marks our inaugural investigation into event-based data, through which we achieved noteworthy results. Moreover, compared with hybrid spiking generative adversarial networks, where the discriminator is an artificial analog neural network, our methodology demonstrates closer alignment with the information processing patterns observed in the mouse. Our code can be found at https://github.com/Brain-Cog-Lab/sgad.
基于神经网络的生成模型在深度学习中面临着重大挑战。目前,此类模型主要局限于人工神经网络领域。脉冲神经网络作为第三代神经网络,由于其丰富的时空动态特性,更接近大脑的处理方式。然而,基于脉冲神经网络的生成模型尚未得到充分研究。特别是,以往基于脉冲神经网络的生成对抗网络的工作是在简单数据集上进行的,效果不佳。在这项工作中,我们率先构建了一个能够处理复杂图像且性能更高的脉冲生成对抗网络。我们的首要任务是识别脉冲生成对抗网络中固有的域外不一致和时间不一致问题。我们通过引入推土机距离和基于注意力的加权解码方法来解决这些问题,显著提高了我们的算法在多个数据集上的性能。实验结果表明,我们的方法在MNIST、FashionMNIST、CIFAR10和CelebA数据集上优于现有方法。除了对静态数据集的研究外,本研究还首次对基于事件的数据进行了调查,并取得了显著成果。此外,与鉴别器为人工模拟神经网络的混合脉冲生成对抗网络相比,我们的方法更符合在小鼠中观察到的信息处理模式。我们的代码可在https://github.com/Brain-Cog-Lab/sgad上获取。