TFLearn - 大型数据集导致NaN损失

时间:2016-12-06 21:37:32

标签: python machine-learning tensorflow deep-learning tflearn

我不知道你是否可以在这里帮助我,但我遇到了一个我无法弄清楚的问题。我有一个大的(对我来说)大约450,000个条目的数据集。每个条目都是大约700个整数的列表,格式如下:

[217088.0, 212992.0, 696.0, 191891.0, 524.0, 320.0, 0.0, 496.0, 0, 0, 364.0, 20.0, 0, 1.0, 0, 0.0, 0, 4.0, 22.0, 0, 672.0, 46.0, 16.0, 0.0, 0.0, 106496.0, 8.0, 0, 4.0, 2.0, 26.0, 640.0, 0.0, 1073741888.0, 624.0, 516.0, 4.0, 3.0, 0, 4319139.0, 0.0, 0, 0.0, 36.0, 8.0, 217088.0, 0.0, 0, 0, 0, 4.0, 5.0, 0, 20.0, 255624.0, 65535.0, 5.10153058443, 396.0, 4319140.0, 552.0, 144.0, 28.0, 5.0, 1048576.0, 217088.0, 350.0, 0.0, 0, 7.0, 1048576.0, 260.0, 0, 116.0, 0, 322.0, 0.0, 0, 4319141.0, 0.0, 10.0, 0.0, 9.0, 4.0, 0, 0, 0, 6.36484131641, 0.0, 0, 11.0, 72.0, 372.0, 45995.0, 217088.0, 0, 4096.0, 12.0, 80.0, 592.0, 264.0, 0, 0, 4096.0, 0.0, 256.0, 0.0, 49152.0, 700.0, 0, 4096.0, 0, 0, 0.0, 336.0, 8.0, 0, 0.0, 0, 4319142.0, 0.0, 60.0, 308.0, 4319143.0, 0, 0, 0, 0, 0, 0.742746270768, 316.0, 420.0, 276.0, 1073741888.0, 0.0, 332.0, 284.0, 0, 1107296320.0, 0.0, 4.0, 13.0, 18.0, 0.0, 632.0, 424.0, 261200.0, 0.0, 299008.0, 0.0, 4096.0, 0, 0.0, 299008.0, 0, 658.0, 0, 4319144.0, 4319145.0, 12.0, 50.0, 292.0, 688.0, 484.0, 70.0, 20.0, 4319146.0, 16.0, 17.0, 0, 0, 0, 0.0, 18.0, 4.0, 330.0, 0.0, 0, 0.0, 42.0, 303104.0, 19.0, 8.0, 20.0, 0.0, 0.0, 544.0, 340.0, 0, 14.0, 0, 209078.0, 0.0, 0.0, 22.0, 0, 209078.0, 0.0, 0.0, 18932.0, 4319147.0, 4.58031739078, 0.0, 376.0, 0.0, 0, 632.0, 4.0, 0, 0, 0, 428.0, 0, 0, 323584.0, 0.0, 24.0, 4.0, 368.0, 12.0, 40.0, 0, 720.0, 4.0, 348.0, 267.0, 20468.0, 32.0, 45995.0, 303104.0, 0.0, 0.0, 0, 0, 224.0, 16.0, 4.0, 44.0, 0.0, 0.0, 444.0, 720.0, 0, 1180.0, 0.0, 16.0, 412.0, 0.0, 4.0, 8462.0, 600.0, 568.0, 16.0, 0, 2.0, 36.0, 0.0, 6.0, 0, 21.0, 0.0, 24.0, 0, 4.0, 652.0, 4319148.0, 92.0, 8.0, 2.0, 0, 0.0, 0, 16.0, 0, 0, 324.0, 4.0, 300.0, 0, 278.0, 400.0, 0, 0.0, 0, 352.0, 0, 0.0, 209078.0, 8.0, 4096.0, 8.0, 36.0, 0.0, 256.0, 268435456.0, 0.0, 48.0, 4319149.0, 6.0, 4319150.0, 0, 416.0, 0, 0, 283.0, 4.0, 0, 0, 0, 8.0, 592.0, 0, 0, 25.0, 0.0, 0, 0, 0.0, 332.0, 212992.0, 540.0, 512.0, 0, 532.0, 20.0, 26.0, 0.0, 0, 52.0, 440.0, 7.0, 488.0, 8.0, 12.0, 0.0, 60.0, 14.0, 3221225536.0, 7.0, 56.0, 432.0, 4.0, 0, 12.0, 0.0, 40.0, 680.0, 16.0, 504.0, 344.0, 576.0, 0.0, 452.0, 266240.0, 290816.0, 578.0, 0, 552.0, 34.0, 0.0, 636.0, 88.0, 698.0, 282.0, 328.0, 38.0, 8.0, 480.0, 64.0, 4319151.0, 0.0, 0.0, 34.0, 460.0, 64.0, 0, 612.0, 0.0, 4319152.0, 0, 604.0, 0, 436.0, 0, 0, 20.0, 0, 4.0, 0, 0, 0, 0, 40.0, 356.0, 584.0, 0, 84.0, 0.0, 0, 0, 0, 294912.0, 7.0, 29.0, 20.0, 0, 60.0, 0.0, 268.0, 536.0, 4319153.0, 0.0, 106.0, 456.0, 24.0, 404.0, 0, 31.0, 0, 380.0, 24.0, 648.0, 0.0, 0, 0, 0.0, 0, 0, 0, 0.0, 0, 0, 0.0, 0.0, 1883.0, 5.85655736551, 34.0, 17744.0, 28680.0, 38.0, 36.0, 0.0, 24576.0, 596.0, 107.0, 33.0, 4.0, 5.0, 0, 0, 45995.0, 384.0, 8.0, 0, 0, 500.0, 20468.0, 34.0, 312.0, 8.0, 660.0, 0.0, 35.0, 608.0, 0, 684.0, 8.0, 68.0, 0.0, 32.0, 34.0, 23117.0, 3.0, 520.0, 0, 4319154.0, 0, 0, 512.0, 8.0, 28.0, 4096.0, 0, 538.0, 0.0, 572.0, 0.0, 2.0, 36.0, 0.0, 0.0, 32.0, 32.0, 4.0, 28.0, 0, 4.0, 38.0, 68.0, 9.0, 0.0, 0, 0.0, 36.0, 39.0, 618.0, 0, 8.0, 266240.0, 4.0, 5.0, 34.0, 304.0, 0, 0.0, 20.0, 40.0, 0.0, 0.0, 0, 580.0, 556.0, 4.0, 8.0, 262.0, 0, 12.0, 32.0, 0, 76.0, 12.0, 184.0, 720.0, 4.0, 16.0, 644.0, 16.0, 28680.0, 4319155.0, 720.0, 0.0, 564.0, 392.0, 672.0, 0.0, 24.0, 492.0, 0, 0.0, 676.0, 0, 0, 0, 12.0, 592.0, 360.0, 8.0, 692.0, 552.0, 4.0, 36.0, 512.0, 7198.0, 42.0, 44.0, 45.0, 4319156.0, 20.0, 388.0, 476.0, 5.0, 36.0, 20480.0, 47.0, 16.0, 326.0, 0.0, 12.0, 0.0, 0.0, 7.0, 272.0, 280.0, 0.0, 0, 288.0, 48.0, 4319157.0, 10.0, 448.0, 4.0, 4.0, 0, 20468.0, 408.0, 2.0, 50.0, 560.0, 0, 1610612768.0, 8.0, 0, 620.0, 656.0, 4.0, 4096.0, 51.0, 0, 0, 0.0, 28.0, 0, 616.0, 0, 296.0, 2.0, 632.0, 468.0, 28.0, 32.0, 52.0, 0, 528.0, 0, 28.0, 0.0, 0, 24.0, 18.0, 4096.0, 0, 8.0, 180.0, 664.0, 4319158.0, 26.0, 0.0, 6.0, 0, 4096.0, 472.0, 0, 28.0, 72.0, 464.0, 672.0, 0, 24.0, 4.0, 0, 28680.0, 0, 0, 18.0, 0, 0, 4319159.0, 24.0, 28.0, 16.0]

我正在使用Tflearn尝试根据此数据创建分类模型,例如每个条目都有0或1个标签,我正在尝试训练模型来预测未知条目是0还是1.以下是我的代码摘要:

def main():
    ## Options ##
    num_tf_layers = 10           # Number of fully connected layers, ex. softmax layer
    num_tf_layer_nodes = 32     # Number of nodes in the fully connected layers
    print_test_scores = 1       # Bool to print test set and predictions
    use_validation_set = 0      # Bool to use testing set when fitting
    num_tf_epochs = 10
    tf_batch_size = 1
    tf_learn_rate = 0.001

    ## Opening files

    print("Preparing labels...")
    trainY = tflearn.data_utils.to_categorical(temp_train_Y, nb_classes=2)
    if use_validation_set:
        testY = tflearn.data_utils.to_categorical(temp_test_Y, nb_classes=2)
    print('Forming input data...')
    net = tflearn.input_data(shape=[None, len(trainX[0])])
    print('Creating fully connected layers...')
    for i in range(num_tf_layers):
        net = tflearn.fully_connected(net, num_tf_layer_nodes)
    print('Creating softmax layer...')
    net = tflearn.fully_connected(net, 2, activation='softmax')
    print('Preparing regression...')
    net = tflearn.regression(net, learning_rate=tf_learn_rate)
    print('Preparing DNN...')
    model = tflearn.DNN(net)
    print('Fitting...')
    if use_validation_set:
        model.fit(trainX, trainY, n_epoch=num_tf_epochs, batch_size=tf_batch_size, validation_set=(testX, testY), show_metric=True)
    else:
        model.fit(trainX, trainY, n_epoch=num_tf_epochs, batch_size=tf_batch_size, show_metric=True)
    print('Complete...')

我的基础是以下TFlearn example。这个程序可以很好地处理一小部分数据,250 0和250 1。我的准确率高达80%,我认为增加更多的数据有助于提高准确率。但是,在添加大量数据后,损失很快就会流向NaN。甚至没有快速完成450,000次迭代。经过一些研究后,我发现我可能有太高的学习率,因为我已将其保留为默认值。我把它设置在0.1和0.000001之间,没有任何东西阻止了去NaN的损失。我也尝试将批量大小更改为1到1024之间,并将层数更改为3到20之间。没有任何帮助。有没有人对改变什么或如何以不同的方式解决这个问题有任何想法?

谢谢!

1 个答案:

答案 0 :(得分:2)

我猜你的网络正在遭遇vanishing gradient problem。这不是神经网络的基本问题 - 它是由某些激活函数引起的基于梯度的学习方法的问题。让我们试着直观地理解问题及其背后的原因。

<强>问题

基于梯度的方法通过了解参数值的微小变化将如何影响网络的输出来学习参数的值。如果参数值的变化导致网络输出的变化非常小 - 网络就无法有效地学习参数,这是一个问题。

这正是在消失的梯度问题中发生的事情 - 网络输出相对于早期层中的参数的梯度变得非常小。这是一种奇特的说法,即使早期层的参数值发生很大变化也不会对输出产生很大影响。让我们试着了解这个问题发生的时间和原因。

<强>原因

消失梯度问题取决于激活函数的选择。许多常见的激活功能(例如sigmoid或tanh)&#39; squash&#39;它们以非常非线性的方式输入到非常小的输出范围。例如,sigmoid将实数行映射到&#34; small&#34;范围[0, 1]。结果,输入空间的大区域被映射到极小的范围。在输入空间的这些区域中,即使输入的大的变化也会在输出中产生小的变化 - 因此梯度很小。

当我们将多个这样的非线性层堆叠在彼此之上时,这变得更加糟糕。例如,第一层将大输入区域映射到较小的输出区域,该较小的输出区域将被第二层映射到甚至更小的区域,第二层将被第三层映射到更小的区域,依此类推。因此,即使第一层参数发生较大变化也不会对输出产生太大影响。

我们可以通过使用不具备“挤压”功能的激活功能来避免此问题。输入空间进入一个小区域。一种流行的选择是整流线性单位,它将x映射到max(0,x)

在Quora上从post采用的答案。

  

更新:爆炸梯度问题

有时,渐变在早期图层中变得更大,并且它被称为爆炸梯度问题。例如,如果您为权重矩阵选择较大的值并以渐变变大的方式设置偏差值,那么神经网络将受到爆炸梯度问题的影响。另一个原因可能是,如果您的数据点本身很大,即使学习率较低,也会在梯度下降期间导致非常大的步长。因此,您可以在训练之前按列对数据点进行标准化,以避免出现梯度爆炸问题。

此外,较大的学习率可能是爆发梯度问题的另一个潜在原因。我鼓励你仔细阅读article,讨论有关消失和爆炸梯度问题及其解决方案的基本思路。

感谢@timleathart的深刻见解。