[发明专利]模型自动优化的方法及装置、设备、存储介质

权利要求书

1.一种模型自动优化的方法，其特征在于，所述方法包括：

根据所述模型的目标需求，确定所述模型的优化指标；

根据所述模型的优化指标，确定对应的目标约束条件；

根据所述模型的模型特征，生成所述模型的剪枝策略；

根据所述剪枝策略和所述目标约束条件，对所述模型进行裁剪，得到目标模型，以实现所述模型的自动优化。

2.根据权利要求1所述的方法，其特征在于，所述模型为多层结构，所述根据所述模型的模型特征，生成所述模型的剪枝策略，包括：

根据所述模型每一层的特征信息，确定所述模型每一层的剪枝比例决策；

根据所述模型每一层的剪枝比例决策，生成所述模型的剪枝策略。

3.根据权利要求2所述的方法，其特征在于，所述根据所述模型每一层的特征信息，确定所述模型每一层的剪枝比例决策，包括：

根据所述模型每一层的特征信息，确定所述模型每一层的特征向量；

根据所述模型每一层的特征向量，确定模型每一层对应的通道剪枝比例；

根据所述模型每一层的特征向量和所述模型每一层对应的通道剪枝比例，生成模型每一层的决策评分；

根据所述决策评分对所述通道剪枝比例进行反馈调节，确定所述模型每一层的剪枝比例决策。

4.根据权利要求1或2所述的方法，其特征在于，所述根据所述剪枝策略和所述目标约束条件，对所述模型进行裁剪，得到目标模型，包括：

根据所述剪枝策略对所述模型进行裁剪，得到中间模型；

根据所述目标约束条件对所述中间模型进行效果评估，得到评估结果；

若所述评估结果表明所述中间模型满足所述目标约束条件，对所述中间模型进行模型微调，得到所述目标模型。

5.根据所述权利要求4所述的方法，其特征在于，所述根据所述剪枝策略对所述模型进行裁剪，得到中间模型，包括：

根据所述剪枝策略确定各层的裁剪通道集合；

根据所述裁剪通道集合，通过应用程序接口API对所述模型的各层进行裁剪，得到所述中间模型。

6.根据权利要求4所述的方法，其特征在于，所述根据所述目标约束对所述中间模型进行效果评估，得到评估结果之后，所述方法还包括：

若所述评估结果表明所述中间模型不满足所述目标约束条件，对所述剪枝策略进行反馈调节，得到新的剪枝策略；

相应地，根据所述新的剪枝策略对所述中间模型进行裁剪，得到新的中间模型；

根据所述目标约束条件对所述新的中间模型进行效果评估，得到新的评估结果。

7.根据权利要求6所述的方法，其特征在于，若所述评估结果表明所述中间模型不满足所述目标约束，所述对所述剪枝策略进行反馈调节，得到新的剪枝策略，包括：

根据所述轻量化指标，确定反馈函数；

根据所述反馈函数，对所述剪枝策略进行反馈调节，得到新的剪枝策略。

8.一种模型自动优化的装置，其特征在于，所述装置包括第一确定模块、第二确定模块、生成模块和裁剪模块，其中：

所述第一确定模块，用于根据所述模型的目标需求，确定所述模型的优化指标；

所述第二确定模块，用于根据所述模型的优化指标，确定对应的目标约束条件；

所述生成模块，用于根据所述模型的模型特征，生成所述模型的剪枝策略；

所述裁剪模块，用于根据所述剪枝策略和所述目标约束条件，对所述模型进行裁剪，得到目标模型，以实现所述模型的自动优化。

9.一种模型自动优化的设备，包括存储器和处理器，所述存储器存储有可在处理器上运行的计算机程序，其特征在于，所述处理器执行所述程序时实现权利要求1至8任一项所述方法中的步骤。

10.一种计算机可读存储介质，其上存储有计算机程序，其特征在于，该计算机程序被处理器执行时实现权利要求1至7中任一项所述方法中的步骤。