Sequence-to-sequence basics in TensorFlow

Introduction

Sequence-to-sequence models help computers turn one sequence of things into another sequence. For example, changing a sentence in English into a sentence in French.

Translating sentences from one language to another.

Turning spoken words into text (speech recognition).

Summarizing long paragraphs into short summaries.

Generating responses in chatbots based on user messages.

Converting handwritten text images into typed text.

Syntax

TensorFlow

import tensorflow as tf
from tensorflow.keras.models import Model
from tensorflow.keras.layers import Input, LSTM, Dense

# Encoder
encoder_inputs = Input(shape=(None, num_features))
encoder_lstm = LSTM(units, return_state=True)
encoder_outputs, state_h, state_c = encoder_lstm(encoder_inputs)
encoder_states = [state_h, state_c]

# Decoder
decoder_inputs = Input(shape=(None, num_features))
decoder_lstm = LSTM(units, return_sequences=True, return_state=True)
decoder_outputs, _, _ = decoder_lstm(decoder_inputs, initial_state=encoder_states)
decoder_dense = Dense(num_classes, activation='softmax')
decoder_outputs = decoder_dense(decoder_outputs)

# Model
model = Model([encoder_inputs, decoder_inputs], decoder_outputs)

The encoder reads the input sequence and summarizes it into states.

The decoder uses these states to generate the output sequence step-by-step.

Examples

This creates an encoder that reads sequences with 50 features and outputs 64 hidden units.

TensorFlow

encoder_inputs = Input(shape=(None, 50))
encoder_lstm = LSTM(64, return_state=True)
encoder_outputs, state_h, state_c = encoder_lstm(encoder_inputs)
encoder_states = [state_h, state_c]

This sets up a decoder that produces sequences with 100 possible output classes.

TensorFlow

decoder_inputs = Input(shape=(None, 50))
decoder_lstm = LSTM(64, return_sequences=True, return_state=True)
decoder_outputs, _, _ = decoder_lstm(decoder_inputs, initial_state=encoder_states)
decoder_dense = Dense(100, activation='softmax')
decoder_outputs = decoder_dense(decoder_outputs)

Sample Model

This example builds a simple sequence-to-sequence model with LSTM layers. It trains on random data for 2 epochs and then predicts output for one sample.

TensorFlow

import numpy as np
import tensorflow as tf
from tensorflow.keras.models import Model
from tensorflow.keras.layers import Input, LSTM, Dense

# Parameters
num_encoder_tokens = 10
num_decoder_tokens = 15
latent_dim = 16

# Encoder
encoder_inputs = Input(shape=(None, num_encoder_tokens))
encoder_lstm = LSTM(latent_dim, return_state=True)
encoder_outputs, state_h, state_c = encoder_lstm(encoder_inputs)
encoder_states = [state_h, state_c]

# Decoder
decoder_inputs = Input(shape=(None, num_decoder_tokens))
decoder_lstm = LSTM(latent_dim, return_sequences=True, return_state=True)
decoder_outputs, _, _ = decoder_lstm(decoder_inputs, initial_state=encoder_states)
decoder_dense = Dense(num_decoder_tokens, activation='softmax')
decoder_outputs = decoder_dense(decoder_outputs)

# Model
model = Model([encoder_inputs, decoder_inputs], decoder_outputs)

# Compile
model.compile(optimizer='rmsprop', loss='categorical_crossentropy', metrics=['accuracy'])

# Dummy data
encoder_input_data = np.random.random((100, 5, num_encoder_tokens))
decoder_input_data = np.random.random((100, 6, num_decoder_tokens))
decoder_target_data = np.random.random((100, 6, num_decoder_tokens))

# Train
history = model.fit([encoder_input_data, decoder_input_data], decoder_target_data, batch_size=16, epochs=2)

# Predict on one sample
test_encoder_input = encoder_input_data[:1]
test_decoder_input = decoder_input_data[:1]
predictions = model.predict([test_encoder_input, test_decoder_input])

print(f"Predictions shape: {predictions.shape}")

Important Notes

Sequence lengths can vary, so we use None in input shapes.

Return states from the encoder to pass to the decoder for context.

Softmax activation in the decoder output layer helps pick the most likely next item.

Summary

Sequence-to-sequence models turn one sequence into another, like translating languages.

They use an encoder to understand the input and a decoder to create the output.

TensorFlow makes it easy to build these models with LSTM layers and functional API.

Practice

1. What is the main role of the encoder in a sequence-to-sequence model?

easy

A. To calculate the loss during training

B. To process the input sequence and create a summary representation

C. To generate the output sequence directly

D. To shuffle the input data randomly

Solution

1. Step 1: Understand the encoder's function

   The encoder reads the input sequence and compresses its information into a fixed-size context vector.

2. Step 2: Differentiate encoder from decoder

   The decoder uses this context vector to generate the output sequence, so the encoder does not generate output directly.

3. Final Answer:

   To process the input sequence and create a summary representation -> Option B

4. Quick Check:

   Encoder = input summary [OK]

Hint: Encoder summarizes input; decoder creates output [OK]

Common Mistakes:

• Confusing encoder with decoder
• Thinking encoder generates output sequence
• Assuming encoder shuffles data

2. Which TensorFlow layer is commonly used to build the encoder and decoder in sequence-to-sequence models?

easy

A. Dense

B. Dropout

C. LSTM

D. Conv2D

Solution

1. Step 1: Identify sequence processing layers

   LSTM layers are designed to handle sequences by remembering information over time steps.

2. Step 2: Exclude other layers

   Dense layers are fully connected but not specialized for sequences; Conv2D is for images; Dropout is for regularization.

3. Final Answer:

   LSTM -> Option C

4. Quick Check:

   LSTM = sequence layer [OK]

Hint: LSTM handles sequences best in TensorFlow [OK]

Common Mistakes:

• Choosing Dense for sequence memory
• Confusing Conv2D with sequence layers
• Using Dropout as main sequence layer

3. What will be the output shape of the decoder LSTM(128, return_sequences=True) layer if the input shape is (batch_size, 10, 64)?

medium

A. (batch_size, 10, 128)

B. (batch_size, 128)

C. (batch_size, 10, 64)

D. (batch_size, 64)

Solution

1. Step 1: Understand LSTM output with return_sequences=True

   When return_sequences=True, LSTM returns output for each time step, so output shape is (batch_size, time_steps, units).

2. Step 2: Match units and input shape

   The LSTM has 128 units and input time steps are 10, output shape is (batch_size, 10, 128).

3. Final Answer:

   (batch_size, 10, 128) -> Option A

4. Quick Check:

   return_sequences=True keeps time steps [OK]

Hint: return_sequences=True keeps time steps in output [OK]

Common Mistakes:

• Ignoring return_sequences effect
• Confusing units with input feature size
• Assuming output shape drops time steps

4. Identify the error in this TensorFlow sequence-to-sequence decoder snippet:

decoder_lstm = tf.keras.layers.LSTM(64, return_sequences=True, return_state=True)
outputs, state_h = decoder_lstm(decoder_inputs, initial_state=encoder_states)

medium

A. initial_state should be None

B. LSTM units should be 128, not 64

C. return_sequences should be False for decoder

D. Missing unpacking of the second state variable

Solution

1. Step 1: Check LSTM return values

   When return_state=True, LSTM returns three outputs: output, state_h, and state_c.

2. Step 2: Verify unpacking variables

   The code unpacks only two variables, missing the cell state (state_c), causing an error.

3. Final Answer:

   Missing unpacking of the second state variable -> Option D

4. Quick Check:

   return_state=True returns 3 outputs [OK]

Hint: return_state=True returns 3 outputs; unpack all [OK]

Common Mistakes:

• Unpacking only two outputs instead of three
• Changing units without reason
• Misusing return_sequences parameter

5. You want to build a sequence-to-sequence model in TensorFlow to translate sentences. Which approach correctly connects encoder and decoder?

hard

A. Use encoder LSTM to get states, then pass these states as initial_state to decoder LSTM

B. Feed decoder inputs directly to encoder without states

C. Use two separate LSTMs without sharing states or outputs

D. Only use Dense layers for both encoder and decoder

Solution

1. Step 1: Understand encoder-decoder connection

   The encoder processes input and produces states that summarize the input sequence.

2. Step 2: Use encoder states to initialize decoder

   Passing encoder states as initial_state to decoder LSTM allows the decoder to start generating output based on input context.

3. Final Answer:

   Use encoder LSTM to get states, then pass these states as initial_state to decoder LSTM -> Option A

4. Quick Check:

   Encoder states initialize decoder [OK]

Hint: Pass encoder states to decoder initial_state [OK]

Common Mistakes:

• Ignoring encoder states in decoder
• Feeding decoder inputs to encoder
• Using only Dense layers for sequences