BasicRNNCell的偏置未被训练

我的网络在训练时卡在了大约70%的准确率,验证准确率仅略高于50%(这是一个二分类问题)。我在调试中,希望排除以下原因。

网络中包括一个BasicRNNCell

rnn_cell = tensorflow.contrib.rnn.BasicRNNCell(100)

tensorflow为这个RNN层创建了权重和偏置,我可以使用tf.trainable_variables()找到它们:

<tf.Variable 'network/rnn/basic_rnn_cell/kernel:0' shape=(200, 100) dtype=float32_ref>,<tf.Variable 'network/rnn/basic_rnn_cell/bias:0' shape=(100,) dtype=float32_ref>

在训练过程中,kernel:0的值如我所期望的每一步都在变化,而bias的值则全部为零且不变化。我使用的是AdagradOptimizer:

optimizer = tf.train.AdagradOptimizer(learning_rate=learning_rate).minimize(cost)

据我所知,偏置应该像权重一样在每次训练步骤中更新。

这是预期的行为吗?这是否可能是我学习停滞的原因?


回答:

根据我的经验,我不会关注这样的细节。从你提供的信息来看,很难判断是否应该有非零偏置。在使用RNN时,采用梯度裁剪是一种很好的做法。我看到使用这种方法在训练中有了很大的改进

    optimizer = tf.train.AdamOptimizer(learning_rate)    # 计算并裁剪梯度    params = tf.trainable_variables()    gradients = tf.gradients(rnn.loss, params)    clipped_gradients, _ = tf.clip_by_global_norm(        gradients, FLAGS.max_gradient_norm)    train_op = optimizer.apply_gradients(zip(clipped_gradients,params), global_step=global_step)

通常,最大梯度范数在1到5之间效果最好。我也更喜欢使用AdamOptimizer。希望这对你有帮助!

Related Posts

使用LSTM在Python中预测未来值

这段代码可以预测指定股票的当前日期之前的值,但不能预测…

如何在gensim的word2vec模型中查找双词组的相似性

我有一个word2vec模型,假设我使用的是googl…

dask_xgboost.predict 可以工作但无法显示 – 数据必须是一维的

我试图使用 XGBoost 创建模型。 看起来我成功地…

ML Tuning – Cross Validation in Spark

我在https://spark.apache.org/…

如何在React JS中使用fetch从REST API获取预测

我正在开发一个应用程序,其中Flask REST AP…

如何分析ML.NET中多类分类预测得分数组?

我在ML.NET中创建了一个多类分类项目。该项目可以对…

发表回复

您的邮箱地址不会被公开。 必填项已用 * 标注