买专利卖专利找龙图腾,真高效! 查专利查商标用IPTOP,全免费!专利年费监控用IP管家,真方便!
申请/专利权人:国微智芯(天津)科技有限公司
摘要:本公开提供了一种神经网络加速方法及装置、加速器、存储介质,该方法包括:将输入图片和预设的卷积核分别转换为第一矩阵和第二矩阵;将所述第一矩阵和所述第二矩阵输入至脉动阵列以实现卷积运算,得到第三矩阵;通过第一公式对第三矩阵进行归一化;所述归一化后的第三矩阵用于进行神经网络的自注意力运算。本公开提供的神经网络加速方法及装置、加速器、存储介质可以提高VisionTransformer神经网络的运算速度。
主权项:1.一种神经网络加速方法,其特征在于,包括:将输入图片和预设的卷积核分别转换为第一矩阵和第二矩阵;将所述第一矩阵和所述第二矩阵输入至脉动阵列以实现卷积运算,得到第三矩阵;通过第一公式对第三矩阵进行归一化;所述归一化后的第三矩阵用于进行神经网络的自注意力运算;其中,第一公式为: 其中,为归一化函数,为第三矩阵,为第三矩阵的量化矩阵,为预设的量化因子,为预设的比例参数,为预设的偏置参数,为第三矩阵的一行数据的累加和,为第三矩阵的一行数据的平方累加和,W为第三矩阵的列数;在进行卷积运算之前,还包括:通过第三公式对矩阵数据进行均匀量化;所述矩阵数据包括第一矩阵或第二矩阵;所述第三公式包括: , , 其中,表示矩阵数据,s为量化因子,zp为零点,()为函数,为取整操作,b为量化位宽;还包括:基于归一化后的第三矩阵构建自注意力运算中的查询向量Q,键向量K和值向量V;对查询向量Q,键向量K和值向量V依次进行自注意力运算,得到自注意力运算结果,T为转置运算,d表示键向量K的维度。
全文数据:
权利要求:
百度查询: 国微智芯(天津)科技有限公司 神经网络加速方法及装置、加速器、存储介质
免责声明
1、本报告根据公开、合法渠道获得相关数据和信息,力求客观、公正,但并不保证数据的最终完整性和准确性。
2、报告中的分析和结论仅反映本公司于发布本报告当日的职业理解,仅供参考使用,不能作为本公司承担任何法律责任的依据或者凭证。