Xóa layer khỏi model Keras Sequential Python

Mô tả

Keras Sequential model cho phép xóa layer cuối cùng bằng phương thức pop().

Ví dụ: Xóa layer

import tensorflow as tf
from tensorflow import keras
from tensorflow.keras import layers

# Tạo model 3 layers
model = keras.Sequential([
    layers.Dense(64, activation='relu', input_shape=(10,), name='layer1'),
    layers.Dense(32, activation='relu', name='layer2'),
    layers.Dense(1, name='layer3')
])

print(f"Số layer ban đầu: {len(model.layers)}")  # 3
for i, layer in enumerate(model.layers):
    print(f"  Layer {i}: {layer.name}, output_shape={layer.output_shape}")

# Xóa layer cuối (layer3)
model.pop()
print(f"\nSau pop(): {len(model.layers)} layers")
for i, layer in enumerate(model.layers):
    print(f"  Layer {i}: {layer.name}, output_shape={layer.output_shape}")

# Xóa tiếp layer2
model.pop()
print(f"\nSau pop() lần 2: {len(model.layers)} layers")

Đầu ra

Số layer ban đầu: 3
  Layer 0: layer1, output_shape=(None, 64)
  Layer 1: layer2, output_shape=(None, 32)
  Layer 2: layer3, output_shape=(None, 1)

Sau pop(): 2 layers
  Layer 0: layer1, output_shape=(None, 64)
  Layer 1: layer2, output_shape=(None, 32)

Sau pop() lần 2: 1 layers
  Layer 0: layer1, output_shape=(None, 64)

Lưu ý quan trọng

• pop() chỉ xóa layer cuối cùng (LIFO - stack behavior).
• Không thể xóa layer giữa hoặc layer đầu tiên trực tiếp.
• Sau pop(), model cần được recompile nếu đã compile trước đó.
• Weights của layer bị xóa sẽ bị mất (trừ khi đã save riêng).

Cách xóa layer bất kỳ (Workaround)

def remove_layer(model, layer_index):
    """Xóa layer tại index bất kỳ bằng cách rebuild model."""
    if layer_index < 0 or layer_index >= len(model.layers):
        raise IndexError("Layer index out of range")
    
    # Lấy config của các layer còn lại
    new_layers = [layer for i, layer in enumerate(model.layers) if i != layer_index]
    
    # Tạo model mới
    new_model = keras.Sequential(new_layers)
    
    # Build lại (cần input_shape cho layer đầu)
    if new_model.layers:
        new_model.build(model.input_shape)
    
    return new_model

# Sử dụng
model = remove_layer(model, 1)  # Xóa layer ở giữa

Khi nào dùng pop()?

• Transfer learning: Bỏ classification head, thêm head mới.
• Thử nghiệm architecture: Thêm/bớt layer trong quá trình thiết kế.
• Feature extraction: Lấy output của layer trung gian.