节点文献

基于改进VGG16图像分类方法研究

Research on Image Classification Method Based on VGG16

  • 推荐 CAJ下载
  • PDF下载
  • 不支持迅雷等下载工具,请取消加速工具后下载。

【作者】 伊卫国杨金玮

【Author】 YI Weiguo;YANG Jinwei;School of Computer and Communication Engineering, Dalian Jiaotong University;School of Software(College of Modern Information Industry), Dalian Jiaotong University;

【通讯作者】 杨金玮;

【机构】 大连交通大学计算机与通信工程学院大连交通大学软件学院(现代信息产业学院)

【摘要】 针对神经网络模型在训练过程中遇到的收敛速度慢和测试样本不平衡导致的准确率降低问题,提出了一种基于改进VGG16图像分类模型的LBF-VGG16(Leaky-Bactch-Focal-VGG16)。该模型将原Relu激活函数替换为Leaky Relu,并在卷积层与激活函数之间引入BN层,以优化收敛效果。在训练过程中,采用SGD优化器,并融入Focal Loss损失函数。试验结果表明,LBF-VGG16模型在分类效果和收敛速度方面较改进前均有显著提升

【Abstract】 Aiming at the image classification issue of the “dying neuron” problem associated with its ReLU activation function and the challenge of imbalanced test samples, a modified version of the VGG 16 model, named LBF-VGG 16(Leaky-Batch-Focal-VGG 16), is proposed. This enhanced model substitutes the traditional ReLU activation function with Leaky ReLU and integrates Batch Normalization layers between the linear and nonlinear components to foster better convergence. During the training phase, the Stochastic Gradient Descent optimizer is employed in conjunction with Focal Loss. Comparative experiments were conducted in two groups: one contrasting VGG 16 with LBF-VGG 16, and the other comparing VGG 13 + Local, VGG 19 + Focal Loss, and LBF-VGG 16. The outcomes demonstrate that the LBF-VGG 16 model outperforms the other models in terms of accuracy and convergence speed.

  • 【文献出处】 大连交通大学学报 ,Journal of Dalian Jiaotong University , 编辑部邮箱 ,2024年04期
  • 【分类号】TP391.41
  • 【下载频次】111
节点文献中: 

本文链接的文献网络图示:

本文的引文网络