当前位置：首页 > news >正文

自编码器——数据降维及特征提取

news 来源：原创 2025/6/21 1:28:55

一.定义

二.原理

1. 编码器

2. 解码器

三.代码实现

1.步骤：

代码实现：

代码解析：

2.进一步优化：

四.应用场景

一.定义

自编码器（Autoencoder）是一种无监督学习的神经网络模型，主要用于数据的降维、特征提取和数据重构。自编码器通常由两部分组成：

编码器（Encoder）： 将输入数据压缩为一个低维的表示（也称为“编码”或“隐含特征”），这部分网络逐渐减少输入的维度，从而提取数据的关键特征。

解码器（Decoder）： 从编码器生成的低维表示中重构原始数据，解码器的目标是使得重构的输出尽量接近原始输入。

二.原理

自编码器的基本目标是通过神经网络学习一个压缩和解压缩的过程。具体来说，自编码器通过一个编码器将高维输入数据转换为一个低维的表示（称为"瓶颈层"或"隐含表示"），然后通过解码器将低维表示重构回原始的高维数据。

自编码器网络的目标是让输入数据 $x$ 和重构后的数据 $\hat{x}$ 尽可能接近，通常使用均方误差（MSE）作为损失函数：

$L(x, \hat{x}) = ||x - \hat{x}||^2$

1. 编码器

编码器的任务是将高维数据压缩为低维数据。编码器部分可以是一个由几层全连接神经网络（或卷积神经网络）组成的模型。编码器的输出即是数据的低维表示。

假设输入为 $x$ ，编码器为 $f$ ，则编码过程可以表示为：

$z = f(x)$

其中， $z$ 是编码器提取到的低维表示。

2. 解码器

解码器的任务是将编码器提取到的低维表示重构为原始输入的近似值。解码器也是一个神经网络，通过学习将 $z$ 解码为 $\hat{x}$ 。

解码过程可以表示为：

$\hat{x} = g(z)$

其中， $g$ 是解码器， $\hat{x}$ 是重构后的数据。

三.代码实现

下面我们用PyTorch实现一个简单的自编码器。我们将使用MNIST手写数字数据集作为例子，来展示自编码器如何进行数据降维和重构。

1.步骤：

导入所需库
定义编码器和解码器结构
训练自编码器
测试并可视化重构结果

代码实现：

import torch
import torch.nn as nn
import torch.optim as optim
import torch.utils.data as DataLoader
from torchvision import datasets, transforms
import matplotlib.pyplot as plt

# 1. 数据准备（以MNIST为例）
transform = transforms.Compose([transforms.ToTensor(), transforms.Normalize((0.5,), (0.5,))])
train_dataset = datasets.MNIST(root='./data', train=True, transform=transform, download=True)
train_loader = DataLoader.DataLoader(dataset=train_dataset, batch_size=64, shuffle=True)

# 2. 定义自编码器模型
class Autoencoder(nn.Module):
    def __init__(self):
        super(Autoencoder, self).__init__()
        # 编码器部分
        self.encoder = nn.Sequential(
            nn.Linear(28 * 28, 128),
            nn.ReLU(),
            nn.Linear(128, 64),
            nn.ReLU(),
            nn.Linear(64, 12)  # 压缩为12维特征
        )
        # 解码器部分
        self.decoder = nn.Sequential(
            nn.Linear(12, 64),
            nn.ReLU(),
            nn.Linear(64, 128),
            nn.ReLU(),
            nn.Linear(128, 28 * 28),  # 解码回28*28的图像尺寸
            nn.Tanh()  # 输出值范围在[-1, 1]，与输入图像的归一化范围匹配
        )

    def forward(self, x):
        x = self.encoder(x)
        x = self.decoder(x)
        return x

# 3. 初始化模型、损失函数和优化器
autoencoder = Autoencoder()
criterion = nn.MSELoss()  # 使用均方误差作为损失函数
optimizer = optim.Adam(autoencoder.parameters(), lr=0.001)

# 4. 训练自编码器
num_epochs = 5
for epoch in range(num_epochs):
    for data in train_loader:
        img, _ = data
        img = img.view(img.size(0), -1)  # 展平图像数据为1D
        output = autoencoder(img)
        loss = criterion(output, img)  # 输入和输出的损失
        optimizer.zero_grad()
        loss.backward()
        optimizer.step()
    print(f'Epoch [{epoch+1}/{num_epochs}], Loss: {loss.item():.4f}')

# 5. 可视化重构效果
test_dataset = datasets.MNIST(root='./data', train=False, transform=transform, download=True)
test_loader = DataLoader.DataLoader(dataset=test_dataset, batch_size=10, shuffle=False)

# 获取一组测试数据
test_data_iter = iter(test_loader)
test_img, _ = test_data_iter.next()
test_img = test_img.view(test_img.size(0), -1)
with torch.no_grad():
    reconstructed = autoencoder(test_img)

# 可视化原始图片和重构后的图片
f, axes = plt.subplots(1, 2)
# 原始图片
axes[0].imshow(test_img[0].view(28, 28), cmap='gray')
axes[0].set_title("Original Image")
# 重构图片
axes[1].imshow(reconstructed[0].view(28, 28), cmap='gray')
axes[1].set_title("Reconstructed Image")
plt.show()

代码解析：

数据预处理： 使用transforms对MNIST数据集进行归一化和转换为Tensor。
自编码器模型：
- 编码器部分将28x28的图像展平为784维，然后通过三层全连接网络逐步减少到12维的隐含表示。
- 解码器部分将12维的编码通过三层全连接网络重建回原始尺寸。
训练过程： 使用均方误差作为损失函数，Adam优化器进行梯度更新。模型尝试让输入图像和重构图像之间的误差最小。
可视化： 对测试数据进行重构，并对比原始图像和重构后的图像。