如何改进我的CNN?高且恒定的验证错误

Was*_*tay 4 python regression conv-neural-network tensorflow

我正在研究一个问题,根据奶牛的图像预测奶牛有多胖的分数。我应用了 CNN 来估计 0-5 之间的值(我拥有的数据集仅包含 2.25 和 4 之间的值)我使用了 4 个 CNN 层和 3 个隐藏层。

我实际上有两个问题:1/我得到了 0.05 的训练误差,但是在 3-5 个时期之后,验证误差仍然保持在 0.33 左右。2/ 我的神经网络预测的值在 2.9 到 3.3 之间,与数据集范围相比太窄了。正常吗?

我怎样才能改进我的模型?

model = tf.keras.models.Sequential([
    tf.keras.layers.Conv2D(16, (3,3), activation='relu', input_shape=(512, 424,1)),
    tf.keras.layers.MaxPooling2D(2, 2),
    tf.keras.layers.Conv2D(32, (3,3), activation='relu'),
    tf.keras.layers.MaxPooling2D(2, 2),
    tf.keras.layers.Conv2D(32, (3,3), activation='relu'),
    tf.keras.layers.MaxPooling2D(2, 2),
    tf.keras.layers.Conv2D(64, (3,3), activation='relu'),
    tf.keras.layers.MaxPooling2D(2,2),
    tf.keras.layers.Flatten(input_shape=(512, 424)),
    tf.keras.layers.Dense(256, activation=tf.nn.relu),
    tf.keras.layers.Dense(128, activation=tf.nn.relu),
    tf.keras.layers.Dense(64, activation=tf.nn.relu),
    tf.keras.layers.Dense(1, activation='linear')
])
Run Code Online (Sandbox Code Playgroud)

学习曲线:

学习曲线

预言:

预言

Ten*_*ort 9

这似乎是过度拟合的情况。你可以

  1. ShuffleData,用shuffle=Truecnn_model.fit。代码如下所示:

    history = cnn_model.fit(x = X_train_reshaped, y = y_train, batch_size = 512, epochs = epochs, callbacks=[callback], verbose = 1, validation_data = (X_test_reshaped, y_test), validation_steps = 10, steps_per_epoch=steps_per_epoch, shuffle = True)

  2. 使用Early Stopping. 代码如下所示

    callback = tf.keras.callbacks.EarlyStopping(monitor='val_loss', patience=15)

  3. 使用正则化。正则化代码如下所示(您也可以尝试 l1 正则化或 l1_l2 正则化):

from tensorflow.keras.regularizers import l2

Regularizer = l2(0.001)

cnn_model.add(Conv2D(64,3, 3, input_shape = (28,28,1), activation='relu', data_format='channels_last', activity_regularizer=Regularizer, kernel_regularizer=Regularizer))

cnn_model.add(Dense(units = 10, activation = 'sigmoid', activity_regularizer=Regularizer, kernel_regularizer=Regularizer))

  1. 您可以尝试使用BatchNormalization.

  2. 使用ImageDataGenerator. 有关更多信息,请参阅此链接

  3. 如果像素不是Normalized,将像素值除以255也有帮助。

  4. 最后,如果仍然没有变化,您可以尝试使用Pre-Trained ModelslikeResNetVGG Net等。