基于深度学习的稀疏矩阵向量乘运算性能预测模型

摘要：

稀疏矩阵向量乘(SpMV)是求解稀疏线性方程组的计算核心,被广泛应用在经济学模型、信号处理等科学计算和工程应用中,对于SpMV及其调优技术的研究有助于提升解决相关领域问题的运算效率.传统SpMV自动调优方法基于硬件平台的体系结构参数设置来提升SpMV性能,但巨大的参数设置量导致搜索空间变大且自动调优耗时大幅增加.采用深度学习技术,基于卷积神经网络,构建由双通道稀疏矩阵特征融合以及稀疏矩阵特征与体系结构特征融合组成的SpMV运算性能预测模型,实现快速自动调优.为提高SpMV运算时间的预测精度,选取特征数据并利用箱形图统计SpMV时间信息,同时在佛罗里达稀疏矩阵数据集上进行实验设计与验证,结果表明,该模型的SpMV运算时间预测准确率达到80％以上,并且具有较强的泛化能力.

关键词：稀疏矩阵向量乘;自动调优;深度学习;卷积神经网络;特征融合

所属期刊栏目：48

分类号：TP332(计算技术、计算机技术)

资助基金：国家重点研发计划2017YFB0202302

在线出版日期：2022-02-24（万方平台首次上网日期，不代表论文的发表时间）

页数：共6页

页码：86-91