最大化互信息约束的R-NLN高频工件图像分类模型

作者:许知建; 欧阳; 李毅; 李柏林; 熊鹰
来源:机械制造与自动化, 2023, 52(06): 89-101.
DOI:10.19344/j.cnki.issn1671-5276.2023.06.022

摘要

高频元件精密加工工件需要定制,难以获取足量的图像样本以训练传统的CNN模型来完成自动化分类。提出基于元学习训练策略的关联非局部网络高频工件分类模型。在经典CNN模型的基础上引入NLN模块,通过关联样本的全局和局部特征提取自适应的任务特征;通过最大化互信息约束优化模型,提高模型的鲁棒性。实验结果表明:与多种主流的小样本分类模型相比,所提模型对miniImageNet和高频工件数据集分类的准确率有显著提高。

全文