首页|尺度因子正则化BN算法

尺度因子正则化BN算法

扫码查看
针对进一步提升深度神经网络训练的收敛速度问题,借鉴批规范化(Batch Normalization,BN)算法的特点,提出尺度因子正则化BN算法.通过对BN层中的可学习尺度因子γ施加L2正则化,使得γ得到衰减,进而参数的梯度上界降低,优化空间更加平滑.基于VGG16 Net与AlexNet,在cifar10、cifar100及裂缝图像数据集上进行该算法与BN算法的图像分类对比实验,结果表明该算法不仅提高了网络训练的收敛速度,而且在相同训练次数下提高了准确率.
BN ALGORITHM OF SCALE FACTOR REGULARIZATION
In order to further improve the convergence speed of deep neural network training,using the characteristics of batch normalization(BN)algorithm as reference,a BN algorithm of scale factor regularization is proposed.By applying L2 regularization to the learnable scale factor y in the BN layer,y was attenuated,the gradient upper bound of the parameter was reduced,and the optimization space was smoother.Based on VGG16 Net and AlexNet,the image classification comparison experiments between this algorithm and the BN algorithm were carried out on the cifar10,cifar100 and crack image datasets.The results show that the proposed algorithm not only improves the convergence speed of network training,but also improves the accuracy rate at the same training times.

Batch normalizationScale factorL2 regularizationImage classification

刘向阳、汪琦

展开 >

河海大学 江苏南京 211100

批规范化 尺度因子 L2正则化 图像分类

国家自然科学基金项目云南省重大科技专项计划项目

61001139202002AE090010

2024

计算机应用与软件
上海市计算技术研究所 上海计算机软件技术开发中心

计算机应用与软件

CSTPCD北大核心
影响因子:0.615
ISSN:1000-386X
年,卷(期):2024.41(6)