近年来,存内计算架构成为人工智能加速芯片的重要研究方向。存内计算架构能够降低数据访问开销,显著提升神经网络等智能算法的能效。目前,针对存内计算芯片的研究主要集中在定点存内计算领域,现有浮点存内计算的实现方案采用近存电路、指数—底数分离、浮点—定点转换等方式,往往面临较低并行度或较多浮点运算周期的挑战,难以实现高能效和高性能。考虑到更大规模网络模型和复杂任务的需求,以及浮点运算在神经网络训练过程中的必要性,浮点运算功能对于未来存内计算芯片非常重要。
针对高能效浮点存内计算的挑战,微电子所刘明院士团队与清华大学刘勇攀教授团队联合提出了“稠密存内+稀疏数字”的新型混合存内计算架构。团队发现神经网络各层的数据分布具有长尾特性,且其中大部分数据的指数位密集分布在一个较小的区间内。基于此,团队提出了将高能效存内计算核心用于执行密集指数位的数据运算,将高灵活度稀疏数字核心用于执行具有长尾特性的稀疏指数位的数据运算。高效的浮点—定点转换电路、灵活编码的稀疏数字核心以及节省高位计算的数字存内计算加法器树电路进一步提升了该设计方案的能效。该SRAM存内计算芯片在28nm工艺下流片,在4比特定点和16比特浮点情况下,存内计算核心的稠密网络峰值能效分别达到275TOPS/W和17.2TOPS/W,稀疏网络峰值能效分别达到1600TOPS/W和90TOPS/W。 该研究成果有助于推动SRAM存内计算芯片在高准确率浮点神经网络及神经网络训练等方面的应用。
基于该成果的文章“A 28nm 16.9-300TOPS/W Computing-in-Memory Processor Supporting Floating-Point NN Inference/Training with Intensive-CIM Sparse-Digital Architecture”入选固态电路领域顶级会议ISSCC 2023。微电子所助理研究员岳金山博士为论文第一作者,刘明院士为论文通讯作者。
图1:“稠密定点+稀疏浮点”存算一体混合架构:运算拆分原理
图2:28nm高能效浮点存内计算芯片
综合信息