训练准确性提高但验证准确性保持不变

Posted

技术标签:

【中文标题】训练准确性提高但验证准确性保持不变【英文标题】:Training accuracy improves but validation accuracy remains same 【发布时间】:2019-01-30 14:57:11 【问题描述】:

我正在使用 CNN 训练服装分类算法。我有大约 60000 张图像用于 10 个班级的训练(按 80:20 划分以进行训练和验证)。分离 10000 张图像进行测试。

训练准确度随着时间的推移而提高,但验证准确度保持不变。训练损失也减少了,但验证损失保持不变。

img_width, img_height = 28, 28
batch_size = 32
samples_per_epoch = 20000
validation_steps = 300
nb_filters1 = 32
nb_filters2 = 64
nb_filters3 = 128
conv1_size = 3
conv2_size = 2
pool_size = 2
classes_num = 10
epochs = 300

#learning_rate = 0.001
learning_rate = 0.01
decay_rate = learning_rate / epochs
momentum = 0.8
sgd = SGD(lr=learning_rate, momentum=momentum, decay=decay_rate, 
     nesterov=True)

model = Sequential()
model.add(
    Convolution2D(nb_filters1, conv1_size, conv1_size, border_mode="same", 
    input_shape=(img_width, img_height, 3)))
model.add(Activation("relu"))
model.add(MaxPooling2D(pool_size=(pool_size, pool_size)))

model.add(Convolution2D(nb_filters2, conv2_size, conv2_size, 
     border_mode="same"))
model.add(Activation("relu"))
model.add(MaxPooling2D(pool_size=(pool_size, pool_size), dim_ordering='th'))

model.add(Flatten())
model.add(Dense(256))
model.add(Activation("relu"))
model.add(Dropout(0.5))
model.add(Dense(classes_num, activation='softmax'))

model.compile(loss='categorical_crossentropy',
              optimizer=sgd,
              metrics=['accuracy'])

train_datagen = ImageDataGenerator(
rescale=1. / 255,
horizontal_flip=True
)

达到的训练准确率:约 96% 达到的验证准确率:约 92% 达到的测试准确率:约 87%

我的问题:我可以做些什么来提高验证准确性或最小化验证损失?可以做哪些改变来改进它?

【问题讨论】:

【参考方案1】:

你体验到的东西叫做Overfitting。 您可以添加更多正则化。最简单的方法是添加另一个 Dropout 层。

from keras.layers import Dropout
***

    model = Sequential()
    model.add(
        Convolution2D(nb_filters1, conv1_size, conv1_size, border_mode="same", 
        input_shape=(img_width, img_height, 3)))
    model.add(Activation("relu"))
    model.add(MaxPooling2D(pool_size=(pool_size, pool_size)))
    model.add(Dropout(0.3)  # <- THIS IS ADDED
    model.add(Flatten())
    ***

0.3 是将乘以 0 的神经元数量,因此它们的值不会包含在随后的计算中。您可以尝试添加其他 Dropout 层并更改它们的值。你也可以给你的层添加一些偏差,这里解释了https://keras.io/regularizers/。

【讨论】:

以上是关于训练准确性提高但验证准确性保持不变的主要内容,如果未能解决你的问题,请参考以下文章

带有keras的CNN,准确性保持不变并且没有提高

火车损失正在减少,但准确度保持不变

Keras DNN 预测模型准确率没有提高

Keras图像分类验证准确率更高

ConvNet验证与每个时期的准确性关系

使用 Keras 进行迁移学习,验证准确度从一开始就没有提高(超出原始基线),而训练准确度提高了