【问题标题】:The first layer's weights don't change after training when using Keras使用 Keras 时,第一层的权重在训练后不会改变
【发布时间】:2019-04-26 21:07:32
【问题描述】:

之前有人讨论过这个问题,但他们普遍收敛到梯度消失作为这个问题的根源。

但在我的模型中,只有两个隐藏层不太可能卡在梯度消失上,如下所示:

from __future__ import print_function

import keras
from keras.datasets import mnist
from keras.models import Sequential
from keras.layers import Dense, Dropout
from keras.optimizers import RMSprop

batch_size = 128
num_classes = 10
epochs = 20

(x_train, y_train), (x_test, y_test) = mnist.load_data()

x_train = x_train.reshape(60000, 784)
x_test = x_test.reshape(10000, 784)
x_train = x_train.astype('float32')
x_test = x_test.astype('float32')
x_train /= 255
x_test /= 255

y_train = keras.utils.to_categorical(y_train, num_classes)
y_test = keras.utils.to_categorical(y_test, num_classes)

model = Sequential()
model.add(Dense(512, activation='relu', kernel_initializer='random_uniform',input_shape=(784,)))
model.add(Dropout(0.2))
model.add(Dense(512, kernel_initializer='random_uniform',activation='relu'))
model.add(Dropout(0.2))
model.add(Dense(num_classes, kernel_initializer='random_uniform',activation='softmax'))


print (model.get_weights().__len__())
for i in range(6):
    print (str(i), "th layer shape: ", model.get_weights()[i].shape ,model.get_weights()[i].__len__(), "mean: ", np.mean(model.get_weights()[i]), "std dev: ", np.std(model.get_weights()[i]))
    print ("Before Training")
    print (model.get_weights()[i][0])


class LossHistory(keras.callbacks.Callback):
    def on_train_begin(self, logs={}):
        self.losses = []

    def on_batch_end(self, batch, logs={}):
        self.losses.append(logs.get('loss'))

batch_history = LossHistory()

model.summary()

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

history = model.fit(x_train, y_train,
                    batch_size=batch_size,
                    epochs=epochs,
                    verbose=1,
                    validation_data=(x_test, y_test),
                    callbacks = [batch_history])

for i in range(6):
    print (str(i), "th layer shape: ", model.get_weights()[i].shape ,model.get_weights()[i].__len__(), "mean: ", np.mean(model.get_weights()[i]), "std dev: ", np.std(model.get_weights()[i]))
    print ("After Training Training")
    print (model.get_weights()[i][0])

我截取了训练前后的权重截图。总之,第一层的权重在训练后没有变化,但第二层的权重确实发生了变化。 (由于参数较多,我只展示了权重矩阵第一行的一部分)

第一层: 训练前

第一层:训练后

第二层:训练前 第二层:训练后

【问题讨论】:

  • 不看其余代码,很难知道。
  • @StephaneBersier 刚刚更新了其余的代码。希望这能让你更清楚
  • 它仍然缺少一些代码,但现在我注意到您使用'random_uniform' 来初始化权重,这通常不是一个好的初始化器。您是否尝试过使用 He 正常初始化?
  • 实际上,在您的情况下,这并不重要。没关系。
  • 你能贴出训练前后输出权重的代码吗?另外,我想知道EPOCHS的价值

标签: tensorflow keras training-data


【解决方案1】:

经过一些调试,我意识到权重矩阵在训练后确实发生了变化,尽管 (784, 512) 矩阵的第一行(在屏幕截图中)看起来从未改变过。

原因是我使用的是经过预处理的 mnist 数据,这是一个手写数字图像数据集,只有那些有墨水的部分才会被编码为特定的 RGB 值。换句话说,那些边缘区域都是'0'。例如,图像二维矩阵的第一行总是'0'。因此,在第一个隐藏层的权重矩阵中,512 个权重向量中的每一个的第一个条目将始终使用 (dJ/da_1)*(da_1/dw_i1) 进行更新,而 (da_1/dw_i1) = x_1 即 '如上所述,在所有训练样本中为 0'。所以它永远不会更新。

【讨论】:

    猜你喜欢
    • 1970-01-01
    • 2017-10-07
    • 2021-12-09
    • 1970-01-01
    • 2019-12-25
    • 1970-01-01
    • 1970-01-01
    • 2020-01-12
    • 1970-01-01
    相关资源
    最近更新 更多