模型剪枝方法、装置、电子设备及存储介质

AITNT
正文
推荐专利
模型剪枝方法、装置、电子设备及存储介质
申请号:CN202411643677
申请日期:2024-11-15
公开号:CN119599082A
公开日期:2025-03-11
类型:发明专利
摘要
本申请提供了一种模型剪枝方法、装置、电子设备及存储介质;方法包括:获取待剪枝模型以及多个输入数据;通过目标子模型执行以下处理:对与目标子模型对应的输入数据进行预测,得到目标子模型的预测结果;定目标子模型对应的输入数据与其他输入数据之间的距离;基于每个子模型的预测结果,对目标子模型进行权重预测,得到目标子模型的预测权重,并融合距离以及预测权重,得到目标子模型的目标权重;基于每个子模型的目标权重,对待剪枝模型进行子模型剪枝,得到剪枝后的模型。通过本申请,能够提升子模型剪枝的准确性。
技术关键词
剪枝模型 模型剪枝方法 计算机可执行指令 样本 电子设备 强度 计算机程序产品 数据采集模块 处理器 可读存储介质 存储器 因子 参数 编码 数值
添加客服微信openai178,进AITNT官方交流群
驱动智慧未来:提供一站式AI转型解决方案
沪ICP备2023015588号