基于贝叶斯优化的结构化剪枝方法、装置、终端及介质

本申请提供基于贝叶斯优化的结构化剪枝方法、装置、终端及介质，本发明与其他手动剪枝方法相比，无需对每个卷积层手动设置剪枝阈值来识别候选的滤波器参数；与随机剪枝方法相比，使用更系统的方法来寻找候选剪枝参数，避免了模型重要参数被剪枝的可能性；与其他基于群体的方法相比，并不会在每次迭代中都生成一组解决方案，所提出的方法受益于贝叶斯优化的探索策略，因此所耗的计算时间更少；本发明所提出的方法具有一个新颖的目标函数，在找到模型的最佳参数时考虑了资源约束和模型性能。
上海交通大学
陈子渊 | 宋卓然 | 梁晓峣