System.ArgumentOutOfRangeException: 索引和长度必须引用该字符串内的位置。 参数名: length 在 System.String.Substring(Int32 startIndex, Int32 length) 在 zhuanliShow.Bind()
【技术实现步骤摘要】
本专利技术属于人工智能,具体涉及一种模型剪枝方法、装置、计算机设备及存储介质。
技术介绍
1、目前,大多数语义分割模型是针对特定任务而设计,并在大规模数据集上进行训练得到的,因此常常无法扩展到多个不同任务进行使用。而对于语义分割的子领域医学图像分割,这种现象更为严重,造成这种现象的原因有两方面:一是医学图像具有高度的多样性,例如肿瘤分割,肿瘤的形状往往都是不规则的,而且不同器官组织的肿瘤往往也具有极大的差异。第二个原因是医学图像分割任务由于数据的标注更为困难以及病人隐私等问题,导致这些任务通常只有一个很小的训练数据集。这两个原因导致在一个医学分割任务数据集上训练得到的分割模型难以推广到其他任务,为此,需要对语义分割模型进行剪枝处理。
2、模型剪枝是一种减小深度学习模型规模的技术,通过去除模型中的冗余参数和连接,从而实现模型的稀疏化。现有的剪枝方法通常需要大量的计算资源和时间来完成剪枝过程。这是由于剪枝涉及对模型进行多次训练和评估,以确定哪些神经元可以被剪枝以优化网络结构,这对于资源有限的环境或需要快速部署的场景来说是一个挑战。其次,现有的剪枝方法通常不使用有关人工设计模型的先验知识,这意味着剪枝过程主要依赖于自动搜索和优化算法来确定最佳的网络结构。虽然这种自动化的方法可以发现一些新颖网络结构,但有时候缺乏人工设计的经验和知识可能导致生成的网络结构在某些任务上计算速度不够理想。
3、综上所述,现有的模型剪枝方法往往会消耗大量的资源并且忽视了目前现有性能优良模型的先验知识,导致其计算效率不高且模型性能受限。<
...【技术保护点】
1.一种模型剪枝方法,其特征在于,包括如下步骤:
2.根据权利要求1所述的模型剪枝方法,其特征在于,所述辅助指标函数为:
3.根据权利要求1所述的模型剪枝方法,其特征在于,使用滤波器重加权方法对C-UNet++神经网络中的滤波器进行激活,包括如下步骤:
4.根据权利要求3所述的模型剪枝方法,其特征在于,使用N2网络中与N1中处于同一位置的滤波器对N1网络中的冗余滤波器进行重加权,通过两个网络中相应权重的平均值公式确定,具体如下:
5.根据权利要求4所述的模型剪枝方法,其特征在于,所述系数γ的计算公式为:
6.根据权利要求5所述的模型剪枝方法,其特征在于,当时,γ等于0.5;当γ大于0.5。
7.根据权利要求1所述的模型剪枝方法,其特征在于,所述滤波器重加权方法为自适应加权策略。
8.一种模型剪枝装置,其特征在于,包括:
9.一种计算机设备,其特征在于,包括存储器和处理器;所述存储器存储有计算机程序,所述处理器用于运行所述存储器内的计算机程序,以执行权利要求1至7任一项所述的方法。
>10.一种计算机可读存储介质,其特征在于,所述计算机可读存储介质存储有计算机程,所述计算机程序适于处理器进行加载,以执行权利要求1至7任一项所述的方法。
...【技术特征摘要】
1.一种模型剪枝方法,其特征在于,包括如下步骤:
2.根据权利要求1所述的模型剪枝方法,其特征在于,所述辅助指标函数为:
3.根据权利要求1所述的模型剪枝方法,其特征在于,使用滤波器重加权方法对c-unet++神经网络中的滤波器进行激活,包括如下步骤:
4.根据权利要求3所述的模型剪枝方法,其特征在于,使用n2网络中与n1中处于同一位置的滤波器对n1网络中的冗余滤波器进行重加权,通过两个网络中相应权重的平均值公式确定,具体如下:
5.根据权利要求4所述的模型剪枝方法,其特征在于,所述系数γ的计算公式为:
6.根据...
还没有人留言评论。发表了对其他浏览者有用的留言会获得科技券。