Pulse neural network simulation method based on GPU

A technology of spiking neural network and simulation method, which is applied in the field of spiking neural network simulation based on GPU, can solve the problems of insufficient use of pulse sparsity, inability to expand acceleration, waste of computing resources, etc., to achieve strong generality and scalability, Avoid the effect of ineffective operation and low power consumption

Pending Publication Date: 2022-03-15
NANJING UNIV OF AERONAUTICS & ASTRONAUTICS
View PDF0 Cites 1 Cited by
  • Summary
  • Abstract
  • Description
  • Claims
  • Application Information

AI Technical Summary

Problems solved by technology

[0004] However, the existing GPU-based spiking neural network simulation technology still has the following disadvantages: most of the spiking neural network models are time-driven, and do not give full play to the advantages of spiking spa

Method used

the structure of the environmentally friendly knitted fabric provided by the present invention; figure 2 Flow chart of the yarn wrapping machine for environmentally friendly knitted fabrics and storage devices; image 3 Is the parameter map of the yarn covering machine
View more

Image

Smart Image Click on the blue labels to locate them in the text.
Viewing Examples
Smart Image
  • Pulse neural network simulation method based on GPU
  • Pulse neural network simulation method based on GPU
  • Pulse neural network simulation method based on GPU

Examples

Experimental program
Comparison scheme
Effect test

Embodiment 1

[0054] A kind of pulse neural network simulation method based on GPU, comprises the following steps:

[0055] S1. Initialize the neural network structure and network weights;

[0056] The concrete steps of described step S1 are:

[0057] S11. Select the appropriate neural network structure according to the type of data set, and build the corresponding neural network structure, specifically: for example, for the MNIST data set, you can choose to build a two-layer neural network structure, and use a fully connected method to connect the layers and neurons between layers; for the CiFar data set, a multi-layer neural network structure can be built, first perform some convolutional layer operations, and then output the results through the fully connected layer; for the Iris data set, a single neuron structure can be selected; the network weight The value uses random numbers to randomly generate weights, that is, the network weights use random numbers to randomly generate weights w...

the structure of the environmentally friendly knitted fabric provided by the present invention; figure 2 Flow chart of the yarn wrapping machine for environmentally friendly knitted fabrics and storage devices; image 3 Is the parameter map of the yarn covering machine
Login to view more

PUM

No PUM Login to view more

Abstract

The invention discloses a GPU-based spiking neural network simulation method. The method comprises the steps of initializing a neural network structure and a network weight; loading a data set and carrying out pulse coding by adopting a pulse coding module; calling a GPU calculation module to select a proper GPU to calculate the pulse neuron membrane voltage according to the data volume, the calculation amount and the priority of the calculation task, comparing whether the pulse neuron membrane voltage exceeds a threshold value or not, and issuing a pulse; creating a pulse queue, and adding neurons of a trigger pulse into the pulse queue; if the pulse queue is not empty, finding out a post-synaptic neuron corresponding to the next layer according to the network structure, and repeating the steps S3-S5 until an output layer is reached; and calculating a loss function according to the result of the output layer and the actual pulse result, and updating the neural network by adopting a gradient descent mode until the iteration is completed. According to the invention, the training speed of the spiking neural network is accelerated, the advantages of the spiking neural network in the aspects of low power consumption and low time delay are exerted, and the conditions of overlarge data set, insufficient video memory and incapability of training are avoided.

Description

technical field [0001] The invention belongs to the technical fields of pulse neural network, brain-inspired computing and high-performance computing, and in particular relates to a GPU-based pulse neural network simulation method. Background technique [0002] The spiking neural network is the third generation of artificial neural network, which uses spiking neurons as the basic unit and is connected to each other to form a neural network. The membrane potential of a spiking neuron accumulates in a non-linear manner, and when the threshold potential is reached, a pulse is released and enters the refractory period to cool down, while the pulse is transmitted to the next neuron. Large-scale spiking neural networks can not only obtain powerful computing power, but also better verify biodynamic properties. [0003] With the maturity of spiking neural network technology, spiking neural networks tend to be deeper and larger networks. The method of using hardware to accelerate s...

Claims

the structure of the environmentally friendly knitted fabric provided by the present invention; figure 2 Flow chart of the yarn wrapping machine for environmentally friendly knitted fabrics and storage devices; image 3 Is the parameter map of the yarn covering machine
Login to view more

Application Information

Patent Timeline
no application Login to view more
IPC IPC(8): G06N3/04G06N3/08G06F9/50
CPCG06N3/049G06N3/084G06F9/505Y02D10/00
Inventor 袁家斌夏涛李若玮
Owner NANJING UNIV OF AERONAUTICS & ASTRONAUTICS
Who we serve
  • R&D Engineer
  • R&D Manager
  • IP Professional
Why Eureka
  • Industry Leading Data Capabilities
  • Powerful AI technology
  • Patent DNA Extraction
Social media
Try Eureka
PatSnap group products