一种剪枝可调的音频分离模型优化方法和装置


专利名称

一种剪枝可调的音频分离模型优化方法和装置

所属单位

智能科学与技术学院

通讯地址

北京市西城区展览路1号

联系人

科学技术发展研究院

电话

010-68322482

专利发明人

周若华

专利号

CN202410318898.0

专利类型

发明授权

专利状态

授权且有效

专利简介

本申请提供一种剪枝可调的音频分离模型优化方法和装置。包括:基于所述第一性能确定所述预训练的音频分离模型的参数的第一排序;根据优化目标对所述预训练的音频分离模型的参数进行聚类,获得多个参数簇;基于第一冗余阈值、所述第一排序和所述多个参数簇筛选第一剪枝范围;基于所述第一剪枝范围对预训练的音频分离模型进行剪枝和量化,都得到第二音频分离模型;检测所述第二音频分离模型的第二性能,基于所述第二性能调整聚类参数簇和所述第一冗余阈值,对所述预训练的音频分离模型重新剪枝和量化。本申请不断调整剪枝和量化的尺度,自适应实现剪枝、量化程度的调整,从而实现音频分离模型的最佳优化效果,保证音频分离效果,同时优化模型体积。
Baidu
map