作者:尚大山等 来源:《先进智能系统》 发布时间:2022/6/14 13:43:24
选择字号:
科学家在忆阻器实现连续学习方面取得进展

 

近年来,以深度学习和神经网络为代表的人工智能技术,在人脸识别、自动驾驶、智慧城市和健康监测等领域迅速发展。然而,传统神经网络模型在处理多任务连续学习时面临灾难性遗忘问题,即神经网络在学习新任务时会遗忘已学到的知识,导致在执行先前任务时性能大幅下降。

中国科学院院士科院微电子研究所微电子器件与集成技术重点实验室研究员刘明团队研究员尚大山、香港大学博士王中锐与清华大学教授刘勇攀等,基于忆阻器存算一体芯片,开发了一款软-硬件协同设计的连续学习原型系统。在软件方面,受人类大脑启发,科研团队提出了一种基于突触元可塑性的混合精度连续学习模型(MPCL)。该模型采用前馈全连接神经网络架构,其中隐藏层权重分别由二值权重和浮点型权重两部分组成(图a)。模型进行多任务连续学习时采用非对称权重更新策略,通过限制与新任务相关性弱的权重更新,平衡权重的可塑性与稳定性,从而有效缓解灾难性遗忘(图b)。在硬件方面,科研人员将MPCL模型部署在一个由忆阻器(RRAM)存算一体(IMC)芯片和通用数字处理器组成的混合模拟数字硬件系统上(图c)。通过将二进制权重映射到忆阻器阵列,该系统采用IMC计算范式,利用基尔霍夫定律和欧姆定律加速向量矩阵乘法操作,减少了处理器和存储器之间数据传输的能量和时间开销。同时,为减轻忆阻器非理想因素对系统性能的影响,研究团队在软件模拟中重构了与硬件完全相同的计算流程,利用原位精调的方法优化了权重的映射过程,并使硬件系统在保持推理准确率的同时最大限度地提高计算并行度(图d)。该原型系统在MNIST和Fashion MNIST数据集上分别实现五个任务连续学习94.9%和95.3%的平均准确率(图e),同时,相较于传统数字系统,核心运算能耗得到大幅降低(图f)。这种基于忆阻器存算一体能力的连续学习实现方案,为未来构建具有自适应能力的低能耗人工智能体提供了参考。

a、混合精度连续学习模型程序框图;b、非对称更新策略限制权重更新;c、基于忆阻器芯片的数模混合硬件系统;d、并行度优化前后准确率对比;e、多任务连续学习的硬件准确率与软件基准对比;f、与数字系统MAC功耗对比。

近日,相关研究成果发表在《先进智能系统》(Advanced Intelligent Systems)上。Wiley学术出版集团科技新闻网站Advanced Science News以Continual learning, just like humans为题,对该成果进行了专题报道。研究工作得到科技部、国家自然科学基金委员会、科院香港大学的支持。(来源:中国科学院微电子研究所)

相关论文信息:https://doi.org/10.1002/aisy.202200026

 
特别声明:本文转载仅仅是出于传播信息的需要,并不意味着代表本网站观点或证实其内容的真实性;如其他媒体、网站或个人从本网站转载使用,须保留本网站注明的“来源”,并自负版权等法律责任;作者如果不希望被转载或者联系转载稿费等事宜,请与我们接洽。
 
 打印  发E-mail给: 
    
 
相关新闻 相关论文

图片新闻
银河系发现巨大黑洞 史上最亮伽马射线暴来自一颗坍缩的恒星
中国天眼揭秘宇宙“随机烟花” 导师:年年审毕业论文,总有这些问题!
>>更多
 
一周新闻排行
 
编辑部推荐博文