基于遗传搜索的卷积神经网络结构化剪枝

摘要：

随着深度神经网络在人工智能领域的广泛应用,其模型参数也越来越庞大,神经网络剪枝就是用于在资源有限设备上部署深度神经网络.结合神经网络通道级编码和自动剪枝的概念,提出了一种基于遗传算法的通道搜索模型.其主要思想是通过小波分解计算剪枝网络与原始网络特征映射的相似性,以相似度为适应度函数进行通道的搜索.在实验中,选择ResNet系列网络在ImageNet数据集上的剪枝性能进行评估,与现有的神经网络剪枝算法相比,该模型不仅具有更高的剪枝率,而且具有更好的精度.

关键词：神经网络剪枝、遗传算法、特征映射、相似度、小波分解

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

资助基金：国家自然科学基金;中央支持地方改革发展项目;西藏大学中央财政支持地方高校改革发展专项资金项目;中央财政支持西藏大学改革发展地方高校;中央财政支持西藏大学改革发展地方高校

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

页数：共3页

页码：4-6