摘要

针对卷积神经网络中的传统池化算法不能很好地考虑到池化域内每个元素与该池化域所含特征之间关联性的问题,提出一种基于高斯函数的池化算法。该算法首先根据池化域内各元素的值和所有元素的最大值计算高斯函数的三个参数值,然后运用高斯函数计算池化域内所有元素的权重,最后根据该权重对池化域内所有元素值计算加权平均值,并以此作为池化结果。选择LeNet5、VGG16、ResNet18和MobileNet v3作为实验模型,在公开数据集CIFAR-10、Fer2013和德国交通标志识别基准(GTSRB)上进行实验,并与最大池化、平均池化、随机池化、混合池化、模糊池化、融合随机池化和soft池化七种池化算法进行对比。实验结果表明,该算法在三个数据集上相较其它算法在精度方面均有0.5到6个百分点的提升,且该算法在运行效率方面优于上述除最大池化和平均池化两种池化算法外的其它池化算法,从而验证该算法有效且具适合应用于对运算时间要求不高但对精度要求较高的情况。