[发明专利]神经网络构建方法及装置、设备、介质

说明书

本公开提供了一种神经网络构建方法及装置、设备、介质，属于人工智能芯片技术领域，用以减少跨众核芯片的传输。该方法包括：获取多个初始神经元簇并进行分组，分别获得多个迷你神经元簇；获取每个初始神经元簇内各迷你神经元簇的簇内连接权重矩阵；获取在不同初始神经元簇之间的簇间连接权重矩阵，簇间连接为稀疏连接；基于簇内连接权重矩阵和簇间连接权重矩阵对迷你神经元簇进行聚合，得到初始神经元簇的稀疏连接权重矩阵；基于初始神经元簇的稀疏连接权重矩阵构建神经形态网络。本公开提供的方法可应用于众核芯片，减少跨众核芯片的传输以及处理核资源的占用。

技术领域

本公开涉及人工智能芯片技术领域，特别涉及一种神经网络构建方法及装置、设备、介质。

背景技术

在生物神经网络中，前神经元胞体产生脉冲信号，经轴突末梢传递给突触，再经突触传递给后神经元树突的前端，由此实现脉冲信号的单向传递。受此启发，利用众核结构类脑计算芯片进行神经形态网络计算中，前神经元簇发放脉冲序列，脉冲序列经过突触权重计算得到突触后电流序列，突触后电流序列传递给后神经元簇。由此可见，在神经形态网络计算中，前神经元簇的脉冲序列需要经过一定的突触连接才能传递给后神经元簇。

发明内容

本公开提供一种神经网络构建方法及装置、设备、介质。

第一方面，本公开提供了一种神经网络构建方法，包括：

获取多个初始神经元簇；

对每个所述初始神经元簇进行分组，分别获得多个迷你神经元簇；

获取每个所述初始神经元簇内各所述迷你神经元簇的簇内连接权重矩阵；其中，所述簇内连接权重包括所述迷你神经元簇内突触自连对应的自连权重和所述迷你神经元簇之间突触互连对应的互连权重；

获取在不同所述初始神经元簇之间的簇间连接权重矩阵，其中，所述簇间连接为稀疏连接；

基于所述簇内连接权重矩阵和所述簇间连接权重矩阵对所述迷你神经元簇进行聚合，得到所述初始神经元簇的稀疏连接权重矩阵；

基于所述初始神经元簇的稀疏连接权重矩阵构建神经形态网络。

第二方面，本公开提供了一种网络仿真方法，包括：

通过本公开实施例提供的所述的神经网络构建方法构建的神经形态网络；

将所述神经形态网络中突触连接权重大于或等于预设的权重阈值的突触连接映射到同一或相邻的所述众核芯片；将所述神经形态网络中突触连接权重小于预设的权重阈值的突触连接映射到不同的所述众核芯片；其中，所述众核芯片包括多个处理核。

第三方面，本公开提供了一种神经网络构建装置，包括：

创建模块，用于获取多个初始神经元簇；

分组模块，用于对每个所述初始神经元簇进行分组，分别获得多个迷你神经元簇；

第一获取模块，用于获取每个所述初始神经元簇内各所述迷你神经元簇的簇内连接权重矩阵；其中，所述簇内连接权重包括所述迷你神经元簇内突触自连对应的自连权重和所述迷你神经元簇之间突触互连对应的互连权重；

第二获取模块，用于获取在不同所述初始神经元簇之间的簇间连接权重矩阵，其中，所述簇间连接为稀疏连接；

聚合模块，用于基于所述簇内连接权重矩阵和所述簇间连接权重矩阵对所述迷你神经元簇进行聚合，得到所述初始神经元簇的稀疏连接权重矩阵；

构建模块，用于基于所述初始神经元簇的稀疏连接权重矩阵构建神经形态网络。