如何一步一步提高图像分类准确率?

一、问题描述

当我们在处理图像识别或者图像分类或者其他机器学习任务的时候,我们总是迷茫于做出哪些改进能够提升模型的性能(识别率、分类准确率)。。。或者说我们在漫长而苦恼的调参过程中到底调的是哪些参数。。。所以,我花了一部分时间在公开数据集 CIFAR-10 [1] 上进行探索,来总结出一套方法能够快速高效并且有目的性地进行网络训练和参数调整。

CIFAR-10 数据集有 60000 张图片,每张图片均为分辨率为 32*32 的彩色图片(分为 RGB3 个信道)。CIFAR-10 的分类任务是将每张图片分成青蛙、卡车、飞机等 10 个类别中的一个类别。本文主要使用基于卷积神经网络的方法(CNN)来设计模型,完成分类任务。

首先,为了能够在训练网络的同时能够检测网络的性能,我对数据集进行了训练集 / 验证集 / 测试集的划分。训练集主要用户进行模型训练,验证集主要进行参数调整,测试集主要进行模型性能的评估。因此,我将 60000 个样本的数据集分成了,45000 个样本作为训练集,5000 个样本作为验证集,10000 个样本作为测试集。接下来,我们一步步来分析,如果进行模型设计和改进。

二、搭建最简单版本的 CNN

对于任何的机器学习问题,我们一上来肯定是采用最简单的模型,一方面能够快速地 run 一个模型,以了解这个任务的难度,另一方面能够有一个 baseline 版本的模型,利于进行对比实验。所以,我按照以往经验和网友的推荐,设计了以下的模型。

模型的输入数据是网络的输入是一个 4 维 tensor,尺寸为 (128, 32, 32, 3),分别表示一批图片的个数 128、图片的宽的像素点个数 32、高的像素点个数 32 和信道个数 3。首先使用多个卷积神经网络层进行图像的特征提取,卷积神经网络层的计算过程如下步骤:

  1. 卷积层 1:卷积核大小 3*3,卷积核移动步长 1,卷积核个数 64,池化大小 2*2,池化步长 2,池化类型为最大池化,激活函数 ReLU。

  2. 卷积层 2:卷积核大小 3*3,卷积核移动步长 1,卷积核个数 128,池化大小 2*2,池化步长 2,池化类型为最大池化,激活函数 ReLU。

  3. 卷积层 3:卷积核大小 3*3,卷积核移动步长 1,卷积核个数 256,池化大小 2*2,池化步长 2,池化类型为最大池化,激活函数 ReLU。

  4. 全连接层:隐藏层单元数 1024,激活函数 ReLU。

  5. 分类层:隐藏层单元数 10,激活函数 softmax。

参数初始化,所有权重矩阵使用 random_normal(0.0, 0.001),所有偏置向量使用 constant(0.0)。使用 cross entropy 作为目标函数,使用 Adam 梯度下降法进行参数更新,学习率设为固定值 0.001。

该网络是一个有三层卷积层的神经网络,能够快速地完成图像地特征提取。全连接层用于将图像特征整合成分类特征,分类层用于分类。cross entropy 也是最常用的目标函数之一,分类任务使用 cross entropy 作为目标函数非常适合。Adam 梯度下降法也是现在非常流行的梯度下降法的改进方法之一,学习率过大会导致模型难以找到较优解,设置过小则会降低模型训练效率,因此选择适中的 0.001。这样,我们最基础版本的 CNN 模型就已经搭建好了,接下来进行训练和测试以观察结果。

训练 5000 轮,观察到 loss 变化曲线、训练集准确率变化曲线和验

  • 10
    点赞
  • 95
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值