基于半结构化剪枝的深度学习模型优化方法及系统

AITNT
正文
推荐专利
基于半结构化剪枝的深度学习模型优化方法及系统
申请号:CN202411833538
申请日期:2024-12-13
公开号:CN119397134A
公开日期:2025-02-07
类型:发明专利
摘要
本发明公开了基于半结构化剪枝的深度学习模型优化方法及系统,涉及模型优化技术领域,包括如下步骤:获取基于训练集训练好的yolov5s神经网络模型,标记为初始训练模型;基于初始训练模型父子层之间的依赖关系生成父子层分组列表;基于所有的剪枝卷积核方法剪枝初始训练模型获取初始剪枝模式,基于父子层分组列表与计算L2范数筛选初始剪枝模式获得最终剪枝模型;基于训练集对最终剪枝模型的非零权重部分进行训练调整;本发明用于解决现有的半结构化剪枝的深度学习模型优化技术中需要逐层分析和配置,导致了计算复杂度较高的问题。
技术关键词
深度学习模型优化 剪枝模型 核方法 模式 列表 神经网络模型 标记 剪枝方法 深度优先搜索算法 训练算法 模型优化技术 训练集 模型更新 预训练模型 邻域 模块 关系 组合方法 代表 复杂度
添加客服微信openai178,进AITNT官方交流群
驱动智慧未来:提供一站式AI转型解决方案
沪ICP备2023015588号