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

分享到

【获取途径】超星期刊网
【作者】龚赛君，曹红，董志诚
【作者单位】西藏大学信息科学技术学院
【年份】2022
【卷号】第11期
【ISSN】1009-3044
【关键词】神经网络剪枝遗传算法特征映射相似度小波分解
【摘要】随着深度神经网络在人工智能领域的广泛应用，其模型参数也越来越庞大，神经网络剪枝就是用于在资源有限设备上部署深度神经网络。结合神经网络通道级编码和自动剪枝的概念，提出了一种基于遗传算法的通道搜索模型。其主要思想是通过小波分解计算剪枝网络与原始网络特征映射的相似性，以相似度为适应度函数进行通道的搜索。在实验中，选择ResNet系列网络在ImageNet数据集上的剪枝性能进行评估，与现有的神经网络剪枝算法相比，该模型不仅具有更高的剪枝率，而且具有更好的精度。
【基金】国家自然科学基金
【文献类型】期刊