卷积和降噪自编码器
1.导入各种包
from pathlib import Path
import pandas as pd
import numpy as np
from numpy.random import choice
import tensorflow as tf
from tensorflow.keras.layers import Input, Conv2D, MaxPooling2D, UpSampling2D
from tensorflow.keras.models import Model
from tensorflow.keras.callbacks import EarlyStopping, ModelCheckpoint
from tensorflow.keras.datasets import fashion_mnist
import matplotlib.pyplot as plt
import seaborn as sns
2.数据准备
(X_train, y_train), (X_test, y_test) = fashion_mnist.load_data()
X_train.shape, X_test.shape
class_dict = {0: 'T-shirt/top',
1: 'Trouser',
2: 'Pullover',
3: 'Dress',
4: 'Coat',
5: 'Sandal',
6: 'Shirt',
7: 'Sneaker',
8: 'Bag',
9: 'Ankle boot'}
classes = list(class_dict.keys())
#规范数据
image_size = 28
def data_prep_conv(x, size=image_size):
return x.reshape(-1, size, size, 1).astype('float32')/255
X_train_scaled = data_prep_conv(X_train)
X_test_scaled = data_prep_conv(X_test)
X_train_scaled.shape, X_test_scaled.shape
out:
#训练与函数结合
def train_autoencoder(path, model, x_train=X_train_scaled, x_test=X_test_scaled):
callbacks = [EarlyStopping(patience=5, restore_best_weights=True),
ModelCheckpoint(filepath=path, save_best_only=True, save_weights_only=True)]
model.fit(x=x_train, y=x_train, epochs=100, validation_split=.1, callbacks=callbacks)
model.load_weights(path)
mse = model.evaluate(x=x_test, y=x_test)
return model, mse
3.卷积Autoencoder
定义一个三层的编码器,它分别使用32、16和8个过滤器的2D卷积。第三层的编码大小是4 x 4 x 8 = 128,比之前的例子要大。
input_ = Input(shape=(28, 28, 1), name='Input_3D')
x = Conv2D(filters=32,
kernel_size=(3, 3),
activation='relu',
padding='same',
name='Encoding_Conv_1')(input_)
x = MaxPooling2D(pool_size=(2, 2), padding='same', name='Encoding_Max_1')(x)
x = Conv2D(filters=16,
kernel_size=(3, 3),
activation='relu',
padding='same',
name='Encoding_Conv_2')(x)
x = MaxPooling2D(pool_size=(2, 2), padding='same', name='Encoding_Max_2')(x)
x = Conv2D(filters=8,
kernel_size=(3, 3),
activation='relu',
padding='same',
name='Encoding_Conv_3')(x)
encoded_conv = MaxPooling2D(pool_size=(2, 2),
padding='same',
name='Encoding_Max_3')(x)
x = Conv2D(filters=8,
kernel_size=(3, 3),
activation='relu',
padding='same',
name='Decoding_Conv_1')(encoded_conv)
x = UpSampling2D(size=(2, 2), name='Decoding_Upsample_1')(x)
x = Conv2D(filters=16,
kernel_size=(3, 3),
activation='relu',
padding='same',
name='Decoding_Conv_2')(x)
x = UpSampling2D(size=(2, 2), name='Decoding_Upsample_2')(x)
x = Conv2D(filters=32,
kernel_size=(3, 3),
activation='relu',
name='Decoding_Conv_3')(x)
x = UpSampling2D(size=(2, 2), name='Decoding_Upsample_3')(x)
decoded_conv = Conv2D(filters=1,
kernel_size=(3, 3),
activation='sigmoid',
padding='same',
name='Decoding_Conv_4')(x)
autoencoder_conv = Model(input_, decoded_conv)
autoencoder_conv.compile(optimizer='adam', loss='mse')
autoencoder_conv.summary()
这里是定义了一个解码器,它限制了滤波器的数量,并使用2D采样代替最大池化来解决滤波器数量的减少的问题。out可见三层自动编码器有12785个参数,略高于前一种深度自动编码器容量的5%。
out:
path = (results_path / 'autencoder_conv.32.weights.hdf5').as_posix()
autoencoder_deep, mse = train_autoencoder(path,
autoencoder_conv,
x_train=X_train_scaled,
x_test=X_test_scaled)
out结果略
f'MSE: {mse:.4f} | RMSE {mse**.5:.4f}'
out:
训练在75个周期后停止,结果测试RMSE进一步减少9%,这是由于卷积滤波器从图像数据学习的能力更有效并且编码尺寸更大。
autoencoder_conv.load_weights(path)
reconstructed_images = autoencoder_deep.predict(X_test_scaled)
reconstructed_images.shape
fig, axes = plt.subplots(ncols=n_classes, nrows=2, figsize=(20, 4))
for i in range(n_classes):
axes[0, i].imshow(X_test_scaled[i].reshape(image_size, image_size), cmap='gray')
axes[0, i].axis('off')
axes[1, i].imshow(reconstructed_images[i].reshape(image_size, image_size) , cmap='gray')
axes[1, i].axis('off')
out:
4.去噪自编码器
自编码器在去噪任务中的应用只影响训练阶段。下面会在标准正态分布的Fashion MNIST数据中添加噪声,同时保持像素值在[0,1]范围内。
def add_noise(x, noise_factor=.3):
return np.clip(x + noise_factor * np.random.normal(size=x.shape), 0, 1)
X_train_noisy = add_noise(X_train_scaled)
X_test_noisy = add_noise(X_test_scaled)
fig, axes = plt.subplots(nrows=2, ncols=5, figsize=(20, 4))
axes = axes.flatten()
for i, ax in enumerate(axes):
ax.imshow(X_test_noisy[i].reshape(28, 28), cmap='gray')
ax.axis('off')
out:
x = Conv2D(filters=32, kernel_size=(3, 3), activation='relu', padding='same', name='Encoding_Conv_1')(input_)
x = MaxPooling2D(pool_size=(2, 2), padding='same', name='Encoding_Max_1')(x)
x = Conv2D(filters=32, kernel_size=(3, 3), activation='relu', padding='same', name='Encoding_Conv_2')(x)
encoded_conv = MaxPooling2D(pool_size=(2, 2), padding='same', name='Encoding_Max_3')(x)
x = Conv2D(filters=32, kernel_size=(3, 3), activation='relu', padding='same', name='Decoding_Conv_1')(encoded_conv)
x = UpSampling2D(size=(2, 2), name='Decoding_Upsample_1')(x)
x = Conv2D(filters=32, kernel_size=(3, 3), activation='relu', padding='same', name='Decoding_Conv_2')(x)
x = UpSampling2D(size=(2, 2), name='Decoding_Upsample_2')(x)
decoded_conv = Conv2D(filters=1, kernel_size=(3, 3), activation='sigmoid', padding='same', name='Decoding_Conv_4')(x)
autoencoder_denoise = Model(input_, decoded_conv)
autoencoder_denoise.compile(optimizer='adam', loss='mse')
path = (results_path / 'autencoder_denoise.32.weights.hdf5').as_posix()
callbacks = [EarlyStopping(patience=5,
restore_best_weights=True),
ModelCheckpoint(filepath=path,
save_best_only=True,
save_weights_only=True)]
#继续在有噪声的输入上训练卷积自编码器,目的是学习如何生成未损坏的原始数据:
autoencoder_denoise.fit(x=X_train_noisy,
y=X_train_scaled,
epochs=100,
batch_size=128,
shuffle=True,
validation_split=.1,
callbacks=callbacks)
out略
autoencoder_denoise.load_weights(path)
mse = autoencoder_denoise.evaluate(x=X_test_noisy, y=X_test_scaled)
f'MSE: {mse:.4f} | RMSE {mse**.5:.4f}'
out:
5.可视化
下图从上到下分别是原始图像和去噪后的图像。它说明了自编码器成功地从噪声图像中产生压缩编码,这些压缩编码与从原始图像中产生的非常相似。