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

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

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

convolution neural networkimage data enhancementsuper parameter optimizationenhanced random searchimage classification

佟国香、刘洪俊、田飞翔

展开 >

上海理工大学光电信息与计算机工程学院 上海 200093

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

国家重点研发计划项目

2018YFB1700902

2024

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

计算机与数字工程

CSTPCD
影响因子:0.355
ISSN:1672-9722
年,卷(期):2024.52(7)