摘要
在复杂的陆战环境中,图像分类技术是快速区分装甲车辆目标的一种重要手段。针对现有基于卷积神经网络(CNN)的主流分类算法对于训练样本的数量及质量有较高要求,在新装备装甲车辆图像分类任务中精度不足的问题,提出一种集成了两个基于不同学习策略的CNN的迁移学习方法。一个CNN在图像样本较易获取、数量充足的老式装甲车辆图像数据集上进行预训练,学习局部细节特征;另一个CNN在图像质量较低的新装备装甲车辆的虚拟图像数据集上进行预训练,学习全局特征。对预训练好的CNN均利用数量有限的新装备装甲车辆真实样本按照不同策略微调,提升表征能力。设计基于Optuna超参数优化框架的自学习模型集成机制,可对两个CNN的输出进行自主加权优化,进一步提高算法的分类准确率。实验结果表明,与随机初始化训练的同一模型相比,所提方法在新装备装甲车辆图像分类任务中准确率提高7%,有效缓解了训练样本偏少的问题。
In complicated land warfare environments,image classification techniques is an important tool to quickly distinguish armored vehicle targets.To address the problem that the existing mainstream classification algorithms based on Convolutional Neural Network(CNN)have high requirements for the number and quality of training samples and perform with insufficient accuracy in the image classification task of newly equipped armored vehicles,a transfer learning method that integrates two CNNs based on different learning strategies is proposed.Specifically,one CNN is pre-trained on an old-fashioned armored vehicle image dataset whose samples can be easily obtained and have sufficient quantity to learn local detail features.The other CNN is pre-trained on the dataset of virtual images of the newly equipped armored vehicles with a low image quality to learn the global features.The pre-trained CNNs are all fine-tuned according to different strategies using a limited number of real samples of newly equipped armored vehicles to improve the characterization capability.A self-learning model integration mechanism based on the Optuna hyperparametric optimization framework is designed,which can autonomously weight the outputs of the two CNNs for optimization and further improve the classification accuracy of the algorithm.The experimental results show that the accuracy of the proposed algorithm is improved by 7%in the image classification task of newly equipped armored vehicles compared with the same model trained from scratch,which effectively alleviates the problem of insufficient training samples.
作者
刘懿
任济寰
吴祥
薄煜明
LIU Yi;REN Jihuan;WU Xiang;BO Yuming(School of Automation,Nanjing University of Science and Technology,Nanjing 210094,Jiangsu,China)
出处
《兵工学报》
EI
CAS
CSCD
北大核心
2023年第8期2319-2328,共10页
Acta Armamentarii
基金
国家自然科学基金项目(62103192)
中国博士后科学基金项目(2021M691597)。
关键词
迁移学习
卷积神经网络
装甲车辆分类
特征提取
模型集成机制
transfer learning
convolutional neural network
armored vehicles
feature extraction
model integration mechanism
