Keras中的序列标记

时间:2016-01-23 17:36:28

标签: machine-learning deep-learning keras lstm recurrent-neural-network

我正在处理句子标签问题。我已经完成了自己的嵌入和填充,我的输入看起来像:

X_i = [[0,1,1,0,2,3...], [0,1,1,0,2,3...], ..., [0,0,0,0,0...],  [0,0,0,0,0...], ....]

对于句子中的每个单词,我想预测四个类中的一个,所以我想要的输出应该如下:

Y_i = [[1,0,0,0], [0,0,1,0], [0,1,0,0], ...]

我简单的网络架构是:

model = Sequential()

model.add(LSTM(input_shape = (emb,),input_dim=emb, output_dim=hidden, return_sequences=True))
model.add(TimeDistributedDense(output_dim=4))
model.add(Activation('softmax'))
    model.compile(loss='binary_crossentropy', optimizer='adam')

model.fit(X_train, Y_train, batch_size=32, nb_epoch=3, validation_data=(X_test, Y_test), verbose=1, show_accuracy=True)

它在训练时显示大约95%,但是当我尝试用训练模型预测新句子时,结果非常糟糕。看起来模型只是为第一个单词学习了一些类并且每次都显示它。我认为问题可能是:

  1. 自己编写填充(句子末尾的零向量),会不会使学习变得更糟?

  2. 我应该尝试学习不同长度的句子,不填充(如果是的话,你能帮我在Keras训练这种模型吗?)

  3. 错误的学习目标,但我尝试了均方误差,二元交叉熵等,它没有改变。

  4. 我认为TimeDistributedDensesoftmax的某些内容,我已经知道它是如何运作的,但仍然不是100%肯定。

  5. 我很高兴看到有关此问题的任何暗示或帮助,谢谢!

1 个答案:

答案 0 :(得分:1)

我个人认为你误解了“序列标签”的含义。

你的意思是:

  1. X是一个句子列表,每个元素X[i]是一个任意长度的单词序列?
  2. Y[i]X[i]的类别,Y[i]的一种热门形式是类似[0, 1, 0, 0]的数组?
  3. 如果是,那么它不是序列标签问题,这是一个分类问题。

    请勿使用TimeDistributedDense,如果是多类别分类问题,即len(Y[i]) > 2,请使用“categorical_crossentropy”代替“binary_crossentropy