期刊文献+
共找到1篇文章
< 1 >
每页显示 20 50 100
融合贝叶斯优化的轨面缺陷检测模型压缩方法
1
作者 井庆龙 闵永智 李成学 《兰州交通大学学报》 CAS 2024年第5期130-138,共9页
针对钢轨表面缺陷检测模型人工调参繁琐且模型容量大的问题,提出一种融合贝叶斯优化的轨面缺陷检测模型压缩方法。首先,利用贝叶斯优化算法得到平衡模型稀疏性与精度的最优超参数;其次,将所得超参数引入稀疏训练并在侧重压缩比和精度的... 针对钢轨表面缺陷检测模型人工调参繁琐且模型容量大的问题,提出一种融合贝叶斯优化的轨面缺陷检测模型压缩方法。首先,利用贝叶斯优化算法得到平衡模型稀疏性与精度的最优超参数;其次,将所得超参数引入稀疏训练并在侧重压缩比和精度的两种策略下对检测模型进行压缩,最后,通过知识蒸馏矫正压缩模型的参数从而补偿稀疏训练导致的精度损失。实验结果显示,该方法在两种稀疏训练策略下得到的轻量化轨面检测模型压缩率可达到96.35%和93.22%,且在硬件部署后的检测速度提升超过两倍,能够避免人工调参对压缩精度的负面影响。 展开更多
关键词 钢轨表面缺陷 缺陷检测模型 模型压缩 贝叶斯优化 稀疏训练
在线阅读 下载PDF
上一页 1 下一页 到第
使用帮助 返回顶部