内蒙古科技大学学报2024,Vol.43Issue(1) :77-81.DOI:10.16559/j.cnki.2095-2295.2024.01.015

基于多模型融合的细粒度图像分类算法

Fine-grained mage classification algorithm based on multi-model fusion

王宁 李宝山
内蒙古科技大学学报2024,Vol.43Issue(1) :77-81.DOI:10.16559/j.cnki.2095-2295.2024.01.015

基于多模型融合的细粒度图像分类算法

Fine-grained mage classification algorithm based on multi-model fusion

王宁 1李宝山1
扫码查看

作者信息

  • 1. 内蒙古科技大学 信息工程学院,内蒙古 包头 014010
  • 折叠

摘要

针对细粒度图像分类中单模型方法泛化能力不足问题,提出了一种用于细粒度图像分类的动态权重多模型融合方法.该方法使用基于注意力机制的网络模型作为参与融合的子模型,同时在模型训练过程中,提出了权重自适应调整算法.该算法能够根据子模型在每次训练中的实际表现和自适应的调整其权重值,保证模型整体达到最优状态.实验结果表明:相较于传统的单模型方法,此方法在提升分类效果的同时模型性能也更加稳定,而且在复杂背景分类任务中表现优异,现实意义更强.

Abstract

Aiming at the problem of insufficient generalization ability of single model method in fine-grained image classification, a dy-namic weighted multi-model fusion method for fine-grained image classification is proposed in this paper. This method uses the network model based on the attention mechanism as the sub-model participating in the fusion, and at the same time, during the model training process, a weight adaptive adjustment algorithm is proposed. According to the actual performance of the sub model in each training, the algorithm can adaptively adjust its weight value to ensure that the whole model reaches the optimal state. The experimental results show that, compared with the traditional single-model method, the method in this paper improves the classification effect while the mod-el performance is more stable, and it performs well in complex background classification tasks, with stronger practical significance.

关键词

图像分类/细粒度/多模型融合/动态权重/注意力机制

Key words

image classification/fine-grained/multi-model fusion/dynamic weight/attention mechanism

引用本文复制引用

基金项目

内蒙古自治区自然科学基金(2019MS06021)

内蒙古自治区自然科学基金(2021MS06007)

内蒙古自治区科技重大专项(2019ZD025)

内蒙古自治区研究生教育教学改革研究与实践项目(YJG20191012710)

内蒙古科技大学创新基金(2019QDL-S09)

内蒙古科技大学创新基金(2019QDL-S10)

内蒙古科技成果转化专项(2020.1-2021.12)

出版年

2024
内蒙古科技大学学报
内蒙古科技大学

内蒙古科技大学学报

影响因子:0.247
ISSN:2095-2295
参考文献量13
段落导航相关论文