北京时间10月14日,清华大学计算机科学与技术系(以下简称“计算机系”)张悠慧团队、精密仪器系(以下简称“精仪系”)施路平团队与合作者在《Nature》杂志发文,首次提出“类脑计算完备性”以及软硬件去耦合的类脑计算系统层次结构。这是清华大学计算机系以第一完成单位/通讯单位发表的首篇《Nature》论文,也完成了清华大学一年多以来在《Nature》正刊的类脑计算领域“三连发”。与通用计算机的“图灵完备性”概念与“冯诺依曼”体系结构相对应,本篇题为《一种类脑计算系统层次结构》(A system hierarchy for brain-inspired computing)的论文首次提出“类脑计算完备性”以及软硬件去耦合的类脑计算系统层次结构,通过理论论证与原型实验证明该类系统的硬件完备性与编译可行性,并扩展了类脑计算系统应用范围使之能支持通用计算。类脑计算处于起步阶段,国际上尚未形成公认的技术标准与方案,这一成果填补了完备性理论与相应系统层次结构方面的空白,利于自主掌握新型计算机系统核心技术。近年来,类脑计算研究受到了越来越多的关注。类脑计算,是借鉴生物神经系统信息处理模式和结构的计算理论、体系结构、芯片设计以及应用模型与算法的总称。其重要性正如欧盟人脑旗舰研究计划项目所指出的:“在未来10到20年内,谁要引领世界经济,谁就必须在这个领域领先。”
张悠慧说。 清华大学计算机系张悠慧研究员相对于通用计算机,这一定义放松了对系统计算过程和精度的约束。团队进一步提出相应的类脑计算机层次结构和确保类脑计算完备性的硬件原语(相当于通用处理器的机器指令)来充分利用这一新完备性带来的优势。该结构具有三个层次(下图):图灵完备的软件模型;类脑计算完备的硬件体系结构;位于两者之间的编译层;并设计构造性转化算法将任意图灵可计算函数转换为类脑计算完备硬件上的模型,进而带来以下优点:第一是计算通用性,即判断系统功能的边界。面向应用的软件模型是图灵完备的,为支持各种应用程序(不限于神经网络类型)的编程提供基础。第二是编译可行性,即降低软硬件开发的耦合程度。通过上述硬件原语以及构造性转化算法,确保“图灵完备”软件与“类脑计算完备”硬件原语序列间的“类脑计算完备性”等价转换(如同通用计算机在“图灵完备性”保证下的“程序编译”),实现了软硬件去耦合,从而增强应用系统的开发效率。第三是引入新的系统设计与优化维度——近似粒度。图:类脑计算机层次结构(左)与现有通用计算机(右)的对比在上述理论与算法基础上,构建支持通用图形处理器、类脑天机芯片和基于阻变存储器的神经形态芯片(仿真)这三类硬件的软件工具链示例,并以智能自行车(作为神经网络应用代表)、鸟群模拟、QR分解(后两者作为非神经网络的通用计算应用代表)为示范应用,证实了设计的可行性,同时测试显示这一设计能够大幅提升系统优化空间,进而显著提升系统效能。《Nature》杂志的一位审稿人认为“这是一个新颖的观点,并可能被证明是神经形态计算领域以及对人工智能的追求的重大发展(This is a novel perspective and could prove to be a significant development for the field of neuromorphic computing, and in the quest for artificial general intelligence)”。计算机体系结构专家、华中科技大学计算机学院教授金海评价此项研究“在类脑计算系统领域做出了基础性、原创性的贡献,有利于自主掌握新型计算系统软硬件核心技术。”论文的共同第一作者为清华大学计算机系研究员张悠慧、博士后渠鹏、博士生季宇、精仪系博士生张伟豪。共同通讯作者为张悠慧与施路平教授。其他合作者包括计算机系教授陈文光、计算机系教授郑纬民院士、原计算机系客座教授/美国特拉华大学电子与计算机工程系教授高光荣,以及来自精仪系、生物医学工程系、自动化系的师生。这项研究得到了清华大学、北京信息科学与技术国家研究中心、北京智源人工智能研究院、北京市“脑科学研究”科技专项计划、北京市未来芯片技术高精尖创新中心、科技部和国家自然基金委等单位的支持。值得一提的是,从今年春节前夕投稿,到8月上旬正式接收,论文的两轮修改由各团队在疫情期间通过线上紧密合作完成。成员们努力克服交流不便、实验数据处理困难等问题,针对文章实验设计回复了超过70页的反馈文件。与审稿人的交流也使得小组人员对类脑计算有了更为深入的思考,目前,进一步研究已经展开。文章的发表,意味着清华大学相关团队在一年多的时间内完成了类脑计算领域《自然》正刊三连发——从2019年8月1日的《Nature》封面文章“Towards Artificial General Intelligence with Hybrid Tianjic Chip Architecture”(精仪系施路平教授团队与合作者)开始,到2020年初的“Fully hardware-implemented memristor convolutional neural network”的文章(微纳电子系吴华强教授团队与合作者),直至此次“A system hierarchy for brain-inspired computing”一文。上述论文分别从“异构融合的新型类脑计算芯片与系统”、“基于忆阻器件的神经形态芯片”、以及“类脑计算完备性与系统层次结构”等角度完成了类脑计算领域的首次实现。“施路平教授团队的研究面向新型类脑计算芯片与系统,是在面向通用人工智能的应用方面开展工作,可以说是我们工作的牵引。而吴华强教授团队在新器件、新工艺层面的创新,则是我们工作的推动。对于我们研究的计算机系统结构而言,两者都是重要的支撑。”张悠慧说:“我们很荣幸能参与其中,进一步探索计算理论和计算系统架构的问题。”清华大学对学科交叉要求极高的类脑计算研究高度重视,于2014年9月创立由7家院系联合而成的类脑计算研究中心,精仪系施路平教授担任中心主任。中心融合了脑科学、电子、微电子、计算机、自动化、材料以及精密仪器等学科,进行全方位类脑智能研究。中心瞄准重大科学问题和国家重大需求,特别是未来卡脖子问题提前布局,给老师们以充分的信任和支持,倡导厚积薄发。近年来,类脑计算研究中心提出了符合脑科学基本规律的新型类脑计算架构——异构融合的天机类脑计算芯片架构,可同时支持计算机科学和神经科学的神经网络模型,发挥它们各自的优势。目前“天机芯”已发展到第三代。第一代天机芯于2015年06月成功流片,该芯片首次将人工神经网络(Artificial Neural Networks, ANNs)和脉冲神经网络(Spiking Neural Networks,SNNs)进行异构融合,同时兼顾技术成熟并被广泛应用的深度学习模型与未来具有巨大前景的计算神经科学模型。该工作于2015年12月受邀在电子器件领域顶级国际会议IEDM2015上做了特邀报告。还自主研发出第一代类脑计算软件工具链,可支持从深度学习框架到“天机芯”的自动映射和编译。第二代天机芯即为发表于去年8月1日《Nature》封面文章上的成果,具有高速度、高性能、低功耗的特点。相比于当时世界先进的IBM的TrueNorth芯片,其功能更全、灵活性和扩展性更好,密度提升20%,速度提高至少10倍,带宽提高至少100倍。同时,研究团队设计了一个自动驾驶自行车实验,来评估芯片整合多模态信息和做出迅速决策的能力。这款自动驾驶自行车,配备了“天机芯”和IMU传感器、摄像头,刹车电机、转向电机、驱动电机等致动器,以及控制平台、计算平台、天机板级系统等处理平台等。自行车的任务是执行实时物体检测、跟踪、语音命令识别、骑行减速等功能,还可实现避障过障、平衡控制和自主决策。这些任务中,部分运用了模拟大脑的模型,而其他则采用了机器学习算法模型。图:天机芯片驱动的自动驾驶自行车这一成果被两院院士选为2019年中国十大科技进展、科技部中国十大科学进展。《Nature》总编斯基珀博士在2019年接受新华社专访时指出:“清华天机芯片的论文将人工智能中的计算机科学研究与神经科学研究结合起来,是人工智能领域的重要里程碑”。目前,中心正在开发第三代天机芯以及新型类脑计算机。而基于现有天机芯片的天机类脑计算机已被2020年第一期人工智能杂志进行了报道——其解决了兼容传统计算架构的异构融合类脑计算系统集成技术等一系列问题,是一款能够满足类脑智能应用需求的普适类脑计算系统,主要包括系统架构、类脑处理器单元、软件工具链、输入输出子系统、类脑计算机加载测试环境等,一代样机可以对36路异步视频输入进行1000 帧/秒的实时数据处理。