模型优化方法及装置、电子设备、存储介质与流程

本公开涉及计算机领域，尤其涉及一种模型优化方法及装置、电子设备、存储介质。

背景技术：

1、终端中通常部署有用于实现各种功能的功能模型，例如，可以部署有用于提升画质的画质增强模型；再例如，可以部署有用于消除噪音的噪音消除模型。

2、大多功能模型在部署至终端之前，还需进行模型量化，以使其适配于该终端。

技术实现思路

1、本公开提供一种模型优化方法及装置、电子设备、存储介质，能够在功能模型中定位导致相应功能模型的输出结果中存在的缺陷的原因。

2、根据本公开的第一方面，提供一种模型优化方法，包括：

3、通过至少两种量化训练工具分别对待优化功能模型进行量化训练，得到与所述至少两种量化训练工具分别对应的至少两个待量化功能模型；

4、通过模型量化工具对所述至少两个待量化功能模型进行量化处理，得到至少两个量化后功能模型；

5、比对所述至少两个量化后功能模型，以基于比对结果对其中一个量化后功能模型进行模型优化，得到优化后功能模型。

6、根据本公开的第二方面，提供一种模型优化装置，包括：

7、训练单元，通过至少两种量化训练工具分别对待优化功能模型进行量化训练，得到与所述至少两种量化训练工具分别对应的至少两个待量化功能模型；

8、量化单元，通过模型量化工具对所述至少两个待量化功能模型进行量化处理，得到至少两个量化后功能模型；

9、优化单元，比对所述至少两个量化后功能模型，以基于比对结果对其中一个量化后功能模型进行模型优化，得到优化后功能模型。

10、根据本公开的第三方面，提供一种电子设备，包括：

11、处理器；

12、用于存储处理器可执行指令的存储器；

13、其中，所述处理器通过运行所述可执行指令以实现如第一方面所述的方法。

14、根据本公开的第四方面，提供一种计算机可读存储介质，其上存储有计算机指令，该指令被处理器执行时实现如第一方面所述方法的步骤。

15、在本公开的技术方案中，在进行模型量化之前，优先基于多种量化训练工具对待优化功能模型进行了量化训练，而在量化训练得到多个待量化功能模型后，再通过模型量化工具对这多个待量化功能模型进行量化处理，进而得到多个量化后功能模型。在此基础上，即可通过比对多个量化后功能模型的方式，对其中一个量化后功能模型进行模型优化，得到优化后功能模型。

16、应当理解的是，由于本公开采用了多种量化训练工具，使得本公开通过量化处理可以得到多个量化后功能模型。在此基础上，由于采用不同量化训练工具得到的各个量化后功能模型各有优劣，因此，基于各个量化后功能模型的比对结果对其中一个量化后功能模型进行优化，相当于将其他量化后功能模型作为参考模型，在该量化后功能模型中定位存在缺陷的部分，避免了相关技术中仅采用单个量化训练工具对模型进行量化训练，而无法定位量化后功能模型中存在的缺陷的问题。

技术特征：

1.一种模型优化方法，其特征在于，包括：

2.根据权利要求1所述的方法，其特征在于，所述通过至少两种量化训练工具分别对待优化功能模型进行量化训练，包括：

3.根据权利要求1所述的方法，其特征在于，所述待优化功能模型为：服务端训练的用于实现目标功能的服务端功能模型；

4.根据权利要求1所述的方法，其特征在于，所述基于比对结果对其中一个量化后功能模型进行模型优化，包括：

5.根据权利要求4所述的方法，其特征在于，所述从所述至少两个量化后功能模型中，确定出性能最佳的目标量化后功能模型，包括：

6.根据权利要求1所述的方法，其特征在于，所述比对所述至少两个量化后功能模型，以基于比对结果对其中一个量化后功能模型进行模型优化，包括：

7.根据权利要求6所述的方法，其特征在于，所述基于所述缺陷比对所述至少两个量化后功能模型，以根据比对结果对所述目标量化后功能模型优化，包括：

8.根据权利要求6所述的方法，其特征在于，所述基于所述缺陷比对所述至少两个量化后功能模型，以根据比对结果对所述目标量化后功能模型优化，包括：

9.根据权利要求6所述的方法，其特征在于，

10.一种模型优化装置，其特征在于，包括：

11.一种电子设备，其特征在于，包括：

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

技术总结
本公开是关于一种模型优化方法及装置、电子设备、存储介质。其中，该方法，包括：通过至少两种量化训练工具分别对待优化功能模型进行量化训练，得到与所述至少两种量化训练工具分别对应的至少两个待量化功能模型；通过模型量化工具对所述至少两个待量化功能模型进行量化处理，得到至少两个量化后功能模型；比对所述至少两个量化后功能模型，以基于比对结果对其中一个量化后功能模型进行模型优化，得到优化后功能模型。

技术研发人员：张红卫,王晓涛,雷磊
受保护的技术使用者：北京小米移动软件有限公司
技术研发日：
技术公布日：2024/7/4