卷积神经网络中的自适应加权池化

• 作者: 赵长乐 何利力
• 来源期刊: 软件导刊
• 学科: 工学
• 关键词: 卷积神经网络 自适应加权池化 图像分类 下采样
• 年，卷（期）: 2022,（3）
• 所属期刊栏目: 人工智能
• 页码范围: 83-88
• 页数: 6页
• 分类号: TP183
• 语种: 中文
• DOI: 10.11907/rjdk.211618

摘要

针对卷积神经网络池化过程中的空间信息丢失、对重要特征提取不准确等问题,提出自适应加权池化方法.根据池化窗口大小设置一组可训练的权重参数,将权重参数与每个池化窗口中排序后的特征值加权求和得到池化结果,在网络反向传播过程中通过梯度下降对权重参数进行迭代更新,得到最优权重参数.在Fashion-minist、Cifar10(分别使用浅层卷积神经网络和ResNet18网络结构)、Omniglot数据集上使用不同池化方法进行图像分类的验证实验,结果表明自适应加权池化将测试集的分类准确率分别提升了0.21％、0.43％、0.80％和0.66％.自适应加权池化能使神经网络根据任务类型不同自行选择最优池化策略,相较于常规池化方式在图像分类问题上取得了更高精度.