关键词:
阻变存储器
自适应自反馈神经元
脉冲神经网络
权重量化
手写数字识别
摘要:
脑科学与人工智能研究正在相互促进中迎来飞速发展,具有交叉学科特点的神经网络也因此受到人们日益广泛的关注。当前以深度卷积神经网络(Deep Convolutional Neural Network,DCNN)为代表的人工神经网络(Artificial Neural Network,ANN)得到研究与发展,虽已在人工智能领域取得了一些成果,但仍存在一些局限性:由于网络的连接密集度、复杂度较高,导致计算和存储资源需求极高;同时,由于时间上的关联性较弱,传统网络模型难以处理事件驱动的时空信息,且训练通常需要大量样本。为解决这些问题,研究人员提出了模仿生物大脑神经元运行机制的第三代神经网络,即脉冲神经网络(Spiking Neural Network,SNN)。由于SNN在信息传递过程中的稀疏连接特性,使其具备更高的计算效率和资源利用率;同时,SNN能够更好地处理实时事件和动态环境中的信息,因此更加适用于事件驱动的计算,展示出较强的生物性,且在训练过程中样本需求相对较少。但基于传统硬件电路平台模拟SNN仍存在一些问题,主要包括如何实现突触权重的无条件约束写入、如何提高神经元硬件电路的仿生程度、如何设计网络外围控制电路等。针对这些问题,本文围绕网络结构、神经元、网络外围控制电路等方面展开系统研究,设计了基于自适应自反馈神经元的脉冲神经网络硬件平台,为实现低能耗神经形态芯片设计提供可行方案和参考,具体内容如下:
(1)为实现SNN硬件化,本文搭建了基础网络模型,并针对神经元、突触等硬件单元进行设计与验证。首先基于RRAM设计具有高可塑性的忆阻突触电路,单元采用1T1R结构,不仅可以实现精准控制突触电路中每个单元,且解决了漏电流问题,进一步降低电路能耗。其次设计LIF神经元电路,基于CMOS搭建电流镜电路、积分电路、不应期电路等,实现脉冲发放功能,且由于不应期电路的存在,神经元不会对连续刺激信号做出连续不断地反映,提高神经元电路生物性。此外,针对突触电路与神经元级联问题,设计了一个基于Verilog-AMS语言的信号同步采集器,该采集器能够高效率、无失真地将突触电路输出的电流信号同步输入到神经元电路中,增强网络稳定性,并为后续神经元功能优化提供基础。
(2)为提高硬件电路的仿生程度,对LIF神经元电路做进一步优化。在生物系统中,神经元可以根据输入信息自适应地调整膜电位强度,进而调整脉冲发放频率,根据此特性,设计自适应电路模块,使LIF神经元电路也可以自适应调整积分电路中阈值电压大小,从而模拟神经元自适应功能。除此之外,神经元之间竞争机制也非常重要,其通过自反馈增强兴奋神经元间的连接强度,同时抑制其他神经元,因此,设计自反馈电路使LIF神经元电路通过竞争机制以提高识别准确率。最后,为使LIF神经元电路能够最大程度模拟生物神经元,把自适应电路和自反馈电路融合在一起形成自适应自反馈电路模块,使LIF神经元不仅可以根据输入信号和输出信号自适应调整自身膜电位强度,还可以通过神经元之间的竞争机制增强兴奋神经元间的连接强度,提高网络稳定性、准确性和学习能力。
(3)根据前两部分的设计与优化,搭建网络外围控制电路,并提出新的权重量化方法,这种方法在不改变电路结构前提下,实现正负权重共同写入,极大地提高网络生物性和稳定性;其次优化网络最终输出模块,该模块可以对网络识别结果进行快速分析,提高数据分析效率;最后,本文基于MNIST数据集对设计的硬件仿真平台进行手写数字识别验证,不仅实现单张图片准确识别,还支持多张图片连续识别,最终得到90.8%的识别准确率,与相同网络结构的软件平台92%的准确率相比,仅有1.2%的差距。以上数据证明所设计SNN硬件平台的可行性和应用价值,为后续SNN相关研究提供参考。