seq2seq

时间:2021-08-08 11:49:52

seq2seq:

seq2seq就是将输入序列经过encoder-decoder变成目标序列。

seq2seq

如图所示,输入序列是 [A, B, C, <EOS>],输出序列是  [W, X, Y, Z, <EOS>]

encoder-decoder:

主要过程就是用RNN对输入序列进行编码,然后再用RNN对上下文向量进行解码。

实现方式:

1、tf.nn.dynamic_rnn

    参考:https://github.com/ematvey/tensorflow-seq2seq-tutorials/blob/master/1-seq2seq.ipynb

流程:

输入序列: [A,B,C,EOS],其中A,B,C, EOS都要进行embedding,encoder部分的代码如下所示:

encoder_cell = tf.contrib.rnn.LSTMCell(encoder_hidden_units)

encoder_outputs, encoder_final_state = tf.nn.dynamic_rnn(
encoder_cell, encoder_inputs_embedded,
dtype=tf.float32, time_major=True,
)

 encoder_outputs是一个时间步的输出,这个在decoder中用不到。encoder_final_stata是最后一层的输出结果,encoder_final_state是一个二元组,(整体的记忆c,隐藏层状态h),然后用encoder_final_state来初始化decoder的状态,而decoder的输入序列为 [EOS, A, B, C],因为dynamic_rnn不能根据上一步的输出来作为当前的输入,所以对于输入来说是固定,而非动态变化的。

decoder_cell = tf.contrib.rnn.LSTMCell(decoder_hidden_units)

decoder_outputs, decoder_final_state = tf.nn.dynamic_rnn(
decoder_cell, decoder_inputs_embedded, initial_state=encoder_final_state, dtype=tf.float32, time_major=True, scope="plain_decoder",
)

2、tf.nn.raw_rnn

这种方法不像dynamic_rnn那样固定,它比较灵活,可以通过迭代函数改变每一个时间步的 输入状态、输入。

offical document

tf.nn.raw_rnn(
cell, //基础神经元
loop_fn, //迭代函数,每次的状态与输入都可以在这里定义
parallel_iterations=None,
swap_memory=False,
scope=None
)
输出:
(emit_ta, final_state, final_loop_state),其中emit_ta是TensorArray类型,其实就是每一个时间步输出的tensor的数组,final_state最后的状态,final_loop_state这个好像是None,不知道啥作用

实现步骤

整体:

decoder_outputs_ta, decoder_final_state, _ = tf.nn.raw_rnn(decoder_cell, loop_fn) //decoder_cell是基础神经单元,loop_fn是迭代函数

迭代函数:

//迭代函数包含time, previous_output, previous_state, previous_loop_state(这个相当于LSTM中那个全局的记忆)
def loop_fn(time, previous_output, previous_state, previous_loop_state):
if previous_state is None: # time == 0, 初始化
assert previous_output is None and previous_state is None
return loop_fn_initial()
else:
return loop_fn_transition(time, previous_output, previous_state, previous_loop_state) //在上一个时间步结束后,即将进入当前时间步时会执行该函数,目的就是确定要将哪些内容传给下一步作为状态输入和输入向量

初始化函数:

def loop_fn_initial():
initial_elements_finished = (0 >= decoder_lengths) # all False at the initial step
initial_input = eos_step_embedded #第一步的输入是EOS
initial_cell_state = encoder_final_state #状态输入就是encoder的最终输出状态,包括(c,h)
initial_cell_output = None
initial_loop_state = None # we don't need to pass any additional information
return (initial_elements_finished,
initial_input,
initial_cell_state,
initial_cell_output,
initial_loop_state)

迭代函数:

def loop_fn_transition(time, previous_output, previous_state, previous_loop_state):
#如何获取上一步的输出
yhat = softmax(previous_output * W + b)
然后概率最大的那个yhat即为上一步的输出结果,并对这个结果进行embedding,作为下一步的输入 def get_next_input():
output_logits = tf.add(tf.matmul(previous_output, W), b)
prediction = tf.argmax(output_logits, axis=)
next_input = tf.nn.embedding_lookup(embeddings, prediction)
return next_input

#判断是否停止,常数 >= tensor向量,tensor中每个位置都要和常数进行比较,结果是一个布尔型的tensor向量
elements_finished = (time >= decoder_lengths) # this operation produces boolean tensor of [batch_size]
# defining if corresponding sequence has ended
#因为这是一个batch块,所以该batch完成的标志是 所有的item都finish,所以需要reduce_all
finished = tf.reduce_all(elements_finished) # -> boolean scalar
#当前步的输入 = 上一步的输出(get_next_input)
#tf.cond(条件,True时调用的函数, False时调用的函数)
input = tf.cond(finished, lambda: pad_step_embedded, get_next_input)
state = previous_state #状态不用改变直接传过去
output = previous_output #previous_output也不用变,好像这个output是一个TensorArray吧?
loop_state = None return (elements_finished,
input,
state,
output,
loop_state)

调用过程:

decoder_outputs_ta, decoder_final_state, _ = tf.nn.raw_rnn(decoder_cell, loop_fn)

 这样就实现了将上一步decoder出来的结果作为下一步的输入,真正实现上图中的过程。

待补充Attention机制

参考:

https://github.com/ematvey/tensorflow-seq2seq-tutorials

https://hanxiao.github.io/2017/08/16/Why-I-use-raw-rnn-Instead-of-dynamic-rnn-in-Tensorflow-So-Should-You-0/