计算机与数字工程2024,Vol.52Issue(7) :2183-2188.DOI:10.3969/j.issn.1672-9722.2024.07.047

基于卷积神经网络的图像数据增强优化策略研究

Research on Optimization Strategy of Image Data Enhancement Based on Convolutional Neural Network

佟国香 刘洪俊 田飞翔
计算机与数字工程2024,Vol.52Issue(7) :2183-2188.DOI:10.3969/j.issn.1672-9722.2024.07.047
✕

基于卷积神经网络的图像数据增强优化策略研究

Research on Optimization Strategy of Image Data Enhancement Based on Convolutional Neural Network

佟国香 1刘洪俊 1田飞翔1
扫码查看

作者信息

  • 1. 上海理工大学光电信息与计算机工程学院 上海 200093
  • 折叠

摘要

论文基于卷积神经网络模型,提出一种改进的自动化图像数据增强策略.针对原有策略搜索空间的离散化及模型训练过程中超参数优化不稳定的问题,通过降低搜索空间策略的复杂度、优化子网络模型的训练过程、选取更有效率的增强随机搜索算法实现超参数优化等方法对原有策略进行改进.并针对不同类型的数据集进行了验证,实验结果表明,论文提出的数据增强策略在CIFAR-10、CIFAR-100、ImageNet数据集上提升了图像分类的准确性,取得了先进的实验效果.

Abstract

Based on convolution neural network model,this paper proposes an improved automatic image data enhancement strategy.Focusing on the discretization of the original strategy search space and the instability of super parameter optimization in model training process,the original strategy is improved by reducing the complexity of search space strategy,optimizing the training process of sub-network model,and selecting more efficient enhanced random search algorithm to realize super parameter optimiza-tion.And the different types of datasets are used to verify.The results show that the proposed enhancement strategy improves the ac-curacy of image classification on CIFAR-10,CIFAR-100 and ImageNet datasets,it achieves advanced experimental results.

关键词

卷积神经网络/图像数据增强/超参数优化/增强随机搜索/图像分类

Key words

convolution neural network/image data enhancement/super parameter optimization/enhanced random search/image classification

引用本文复制引用

基金项目

国家重点研发计划项目(2018YFB1700902)

出版年

2024
计算机与数字工程
中国船舶重工集团公司第七0九研究所

计算机与数字工程

CSTPCD
影响因子:0.355
ISSN:1672-9722
段落导航相关论文