ConvLSTM实战：伦敦空气质量时空预测全流程解析

1. 问题定义与模型选型：为什么偏偏是ConvLSTM

上一篇我们完成了伦敦地区空气质量数据的获取和预处理。拿到了20多个监测站点的逐小时污染物浓度记录，做了缺失值插补，也看了基本的时间分布规律。但数据准备好了并不代表能直接进入训练，模型选型这一步如果拍脑袋，后面调试起来会非常痛苦。

先说结论：空气质量预测本质上是一个时空预测问题。你在某个站点看到的PM2.5浓度，既受到过去几个小时内同一地点污染物累积的影响，也会被周围区域的风向、扩散、化学反应等空间因素干扰。举个例子，伦敦某个站点监测到PM2.5突升，很可能不是当地排放导致，而是上游方向几个街区外的交通枢纽贡献的。这意味着，任何一个针对单站点的纯时间序列模型，天然就丢掉了一半信息——空间相关性和空间传导效应。

纯LSTM能做时间维度的建模，但它的输入是一维序列，每个时间步只能接收一个向量。你想让它处理空间关系，只能把20个站点当作20个特征维度拼进向量里。这样做有两个问题：第一，站点之间没有天然的拓扑结构，你无法让模型理解"这个站点物理上就在另一个站点旁边"；第二，站点位置是离散的，一旦遇到新地点（比如新建的监测点），模型结构就要跟着调整，灵活度很差。

纯CNN的情况正好相反。卷积操作天然擅长抓空间特征——平滑、边缘、局部关联性——它能够在一张特征图上捕捉"某个区域的污染物浓度梯度"。但CNN缺乏时序状态记忆机制，处理序列问题时要么用因果卷积加膨胀因子来拉长时间感受野，要么就配合循环结构一起用。

ConvLSTM就是在这个背景下被提出来的。它的核心思想非常直观：把LSTM内部的状态转移计算从全连接矩阵乘法换成卷积运算。这样做之后，隐藏状态不再是一个一维向量，而是一张二维特征图。时间维度的记忆能力保留了下来（LSTM的门控机制还在），空间维度的特征提取能力也进来了（卷积核在特征图上滑动，提取局部空间模式）。你喂进去的不是"20个站点的一堆数值"，而是"一张随时间变化的污染物浓度热力图"。模型天然就理解"附近区域浓度高，同方向会向这里扩散"这一空间常识，不需要你再额外编码任何位置信息。

在动手写代码之前，我还对比过两个替代方案。一个是图神经网络（GCN或GAT）加LSTM的组合，理论上是更精细的建模方式，因为站点之间的关系是图结构，边权重可以表示风向、距离等因素。但实际做下来发现，图结构构造的合理性对结果影响极大，而且在站点稀疏场景下，图卷积的感受野很有限，效果反而不稳定。另一个是时空Transformer架构，Spacetimeformer这类模型确实效果好，但对计算资源的要求太高，训练一个中规模模型动辄要十几张卡，不太适合做常规的设备级实验。

所以我最终选定的方案就是ConvLSTM，并且是"编码-预测"（Encoder-Forcaster）结构的标准形态：编码器部分用两层ConvLSTM把输入序列压缩成隐状态，预测器部分再接一个ConvLSTM逐时间步生成未来序列，最后通过一个卷积输出层映射到目标通道。这个结构在降水临近预报领域已经得到了充分验证，迁移到空气质量预测上，思路是完全一致的——都是利用历史时空场预测未来时空场。

还有一个细节值得注意：ConvLSTM中的卷积核是固定尺寸的，常见的是3×3。你可能会担心3×3只能看到局部邻域，无法捕捉大范围的传播特征。这个担心是合理，但实际不会构成问题。因为每一层ConvLSTM输出的特征图会作为下一层的输入，叠加两层之后，感受野就扩大到了一个较大的范围；同时，循环结构本身是不断迭代的，每个时间步的输出都包含了此前所有历史步的综合信息。这一层一层的递进，就已经把空间传播的大尺度特性包含进去了。

2. 数据重构：把表格数据变成四维张量的完整流程

ConvLSTM的输入格式和普通LSTM有很大区别，如果你直接用Sklearn或Pandas里那种"行是时间、列是站点"的表格去喂模型，第一行代码就会报维度错误。我当时在这个环节就走了不少弯路，所以单独用一整节来讲数据重构。

2.1 空间网格化：从离散站点到连续场

ConvLSTM要求输入是类似图像序列的张量结构，形状是[batch, seq_len, channels, height, width]。也就是说，每一个时间步，input应该是一张多通道的"图像"，图像的宽和高是空间坐标，通道是污染物种类。

但我们的原始数据是20个监测站点，分布不规律，没有天然形成规整的网格。怎么办？最靠谱的做法是把伦敦地区划分成规整的矩形网格，然后用插值方法把每个站点上的观测值映射到网格的每个节点上。

我采用的是16×16的网格，覆盖了伦敦市中心及其周边大约40公里×40公里的范围。网格数量太小会丢失空间细节，太大又会引入大量无有效观测支撑的插值区域，带来噪声。16×16算是小巧但均衡的选择。

具体实现用的是IDW（反距离权重插值）：

import numpy as np from scipy.spatial import cKDTree def idw_interpolate(values, station_coords, grid_lats, grid_lons, power=2, max_distance=10.0): """ 反距离权重插值：将监测站点的离散值插值到规整网格上 values: (n_stations,) 某时间步某污染物的站点浓度 station_coords: (n_stations, 2) 站点经纬度坐标 grid_lats/grid_lons: (H, W) 网格节点的经纬度坐标 """ n_stations = len(values) H, W = grid_lats.shape grid_values = np.full((H, W), np.nan) # 构建站点KDTree，方便查询每个网格点的邻近站点 tree = cKDTree(station_coords) grid_points = np.stack([grid_lats.ravel(), grid_lons.ravel()], axis=1) dists, indices = tree.query(grid_points, k=min(8, n_stations)) dists[dists < 1e-6] = 1e-6 # 防止除零 for i in range(len(grid_points)): dist = dists[i] idx = indices[i] # 只使用最大距离内的站点，避免远处站点产生虚假影响 valid = dist < max_distance if not np.any(valid): continue w = 1.0 / np.power(dist[valid], power) grid_values.ravel()[i] = np.sum(w * values[idx[valid]]) / np.sum(w) # 仍有缺失的网格点用全局均值填充 global_mean = np.nanmean(values) grid_values = np.where(np.isnan(grid_values), global_mean, grid_values) return grid_values

这里的power=2是IDW的经典取值，越大表示远处站点的影响衰减越快。插值会造成一定程度的空间平滑，但这对预测任务来说反而是件好事——它能消除单站点观测噪声，让整个场变得更稳定。

2.2 滑动窗口切分：批次样本生成

网格化完成之后，每个时间步都有了一张6通道的16×16特征图（6个通道对应PM2.5、PM10、NO2、O3、SO2、CO）。接下来要做的就是把时间序列切成"过去→未来"的样本对。

我设置的预测任务是：用过去24小时的观测，预测未来24小时的PM2.5浓度分布场。之所以选这个窗口长度，是因为空气污染的累积效应通常以小时为尺度，24小时能覆盖一个完整的日变化周期，同时也兼顾了计算效率。

def create_sequences(grid_data, input_steps=24, output_steps=24, stride=6): """ grid_data: (T, C, H, W) 完整时间序列 返回: X (N, input_steps, C, H, W), y (N, output_steps, 1, H, W) 使用stride=6控制窗口滑动的步长，避免样本高度重叠导致过拟合 """ T = grid_data.shape[0] samples_x = [] samples_y = [] for i in range(0, T - input_steps - output_steps, stride): x = grid_data[i:i + input_steps] # (24, C, H, W) y = grid_data[i + input_steps:i + input_steps + output_steps][:, 0:1] # 只预测PM2.5 samples_x.append(x) samples_y.append(y) return np.stack(samples_x), np.stack(samples_y)

注意输出部分只取了第0个通道，也就是PM2.5。如果你想把NO2、O3等多项指标一起预测，只需要把输出通道数改为对应数值即可。但多任务的训练难度会显著增加，建议先跑通单变量预测，再加通道。

切片时的stride参数很关键，它控制两个相邻样本之间的间隔。如果stride=1，相邻样本几乎是完全重叠的，训练集和验证集之间会存在严重的样本泄漏，导致验证指标虚高。我设置为6，相当于每小时只采样一个训练窗口，重叠度降低了，模型泛化能力也更真实。

2.3 标准化：字段级的细节处理

标准化是这类时空预测任务最容易踩坑的地方。如果你直接对整个三维张量做Min-Max缩放，那你就把不同污染物的量纲差异全部抹平后又强行混合了。PM2.5的日均值可能在10~50μg/m³，而NO2可能到80~200μg/m³，它们的浓度范围完全不同。

正确的做法是每个通道各自做标准化。更严格的方案是：先按时间轴切出训练集，在训练集上计算每个通道的均值和标准差，然后用这套参数去标准化验证集和测试集。绝对不能在整个数据集上统一算统计量，否则训练过程会偷看到未来的信息。

# 假设 train_data, val_data 形状均为 (T, C, H, W) means = train_data.mean(axis=(0, 2, 3), keepdims=True) # 每个通道的均值 stds = train_data.std(axis=(0, 2, 3), keepdims=True) train_data_norm = (train_data - means) / stds val_data_norm = (val_data - means) / stds

这里在空间维度上取了平均值，这是因为虽然不同位置的浓度有差异，但如果每个网格点都单独算均值标准差，会把插值造成的人为空间模式也学进去，干扰模型。在整个空间上共享一套统计量，既能保持浓度整体分布的信息，又不会放大局部噪声。

网格化完成后再做一个值得注意的细节：检查一下每个时间步的网格场是否存在大面积的NaN。伦敦的监测站点数据质量总体不错，但偶尔会有站点故障导致某段时间数据缺失，插值后边缘区域容易是NaN。我的处理方案是在标准化之前，把缺失值按时间轴前向填充，实在填不了的时间步就整条剔除。

3. 模型实现：ConvLSTM网络结构与代码逐段拆解

3.1 ConvLSTM单元的计算逻辑

ConvLSTM和LSTM的对应关系用一句话就能说清：LSTM里的输入门、遗忘门、输出门、候选记忆，原本是W·x + U·h这种矩阵乘法，ConvLSTM全部替换成W * x + U * h这种卷积操作。数学表达如下：

i_t = σ(W_xi * X_t + W_hi * H_{t-1} + b_i) f_t = σ(W_xf * X_t + W_hf * H_{t-1} + b_f) o_t = σ(W_xo * X_t + W_ho * H_{t-1} + b_o) g_t = tanh(W_xg * X_t + W_hg * H_{t-1} + b_g) C_t = f_t ⊙ C_{t-1} + i_t ⊙ g_t H_t = o_t ⊙ tanh(C_t)

其中*代表卷积，⊙代表逐元素乘。从这里就能看出ConvLSTM同时具备两个能力：门控机制让模型能够选择性记忆或遗忘历史状态（时间维度），卷积核让当前状态由局部邻域信息共同决定（空间维度）。每一步计算时，输入和上一个隐状态都会先经过一次卷积，提取空间特征，再进行LSTM式的门控运算。

在PyTorch中，标准的ConvLSTM单元实现如下，我做了额外的优化：原本8个独立卷积分为4组，每组合并输入和隐状态的卷积，减少计算次数。

import torch import torch.nn as nn class ConvLSTMCell(nn.Module): def __init__(self, in_channels, hidden_channels, kernel_size): super().__init__() padding = kernel_size // 2 self.conv = nn.Conv2d( in_channels + hidden_channels, 4 * hidden_channels, # i, f, o, g 共4个门 kernel_size, padding=padding ) self.hidden_channels = hidden_channels def forward(self, x, h, c): # x: (B, C_in, H, W), h: (B, C_hidden, H, W), c: (B, C_hidden, H, W) combined = torch.cat([x, h], dim=1) gates = self.conv(combined) # (B, 4C_hidden, H, W) i, f, o, g = torch.split(gates, self.hidden_channels, dim=1) i = torch.sigmoid(i) f = torch.sigmoid(f) o = torch.sigmoid(o) g = torch.tanh(g) c_next = f * c + i * g h_next = o * torch.tanh(c_next) return h_next, c_next

这个实现里，卷积核大小直接决定感受野宽度，我用的3×3。如果你要处理更大范围的网格，可以尝试5×5，但要注意参数量和计算量的增长。

3.2 编码-预测结构：两层编码器加一层预测器

有了单个ConvLSTM单元之后，就需要把它组织成完整的网络。我参考了ConvLSTM论文中经典的编码-预测架构，并针对空气质量预测稍作调整。

编码器部分由两层ConvLSTM组成，各层有独立的隐状态。时间步按顺序输入，逐层向下传递。每一层输出不再传给后续时间步的同一层，而是作为相邻状态参与下一时间步该层的输入。

预测器部分是一个独立的ConvLSTM层，它的初始状态来自编码器最后一层输出的记忆状态。预测器工作在未来时间轴上，前一个时间步的输出会作为下一个时间步的输入，这样就能逐步把浓度场"外推"到未来24小时。

class ConvLSTMModel(nn.Module): def __init__(self, in_channels=6, hidden_dims=[64, 64, 32], kernel_size=3, output_steps=24): super().__init__() self.encoder_layers = nn.ModuleList() self.encoder_layers.append(ConvLSTMCell(in_channels, hidden_dims[0], kernel_size)) self.encoder_layers.append(ConvLSTMCell(hidden_dims[0], hidden_dims[1], kernel_size)) self.forecaster_layer = ConvLSTMCell(hidden_dims[1], hidden_dims[2], kernel_size) self.output_conv = nn.Conv2d(hidden_dims[2], 1, kernel_size=1) self.output_steps = output_steps def forward(self, x): """ x: (B, input_steps, C, H, W) 返回: (B, output_steps, 1, H, W) """ batch_size = x.size(0) H, W = x.shape[3], x.shape[4] # 初始化所有隐藏状态 h1 = torch.zeros(batch_size, 64, H, W, device=x.device) c1 = torch.zeros(batch_size, 64, H, W, device=x.device) h2 = torch.zeros(batch_size, 64, H, W, device=x.device) c2 = torch.zeros(batch_size, 64, H, W, device=x.device) h3 = torch.zeros(batch_size, 32, H, W, device=x.device) c3 = torch.zeros(batch_size, 32, H, W, device=x.device) # 编码阶段 for t in range(x.size(1)): h1, c1 = self.encoder_layers[0](x[:, t], h1, c1) h2, c2 = self.encoder_layers[1](h1, h2, c2) # 预测器初始状态直接使用编码器最终状态 h3, c3 = h2, c2 outputs = [] current_input = h2 # 编码器最终的隐状态作为预测器第一个输入 for t in range(self.output_steps): h3, c3 = self.forecaster_layer(current_input, h3, c3) out = self.output_conv(h3) outputs.append(out) current_input = out return torch.stack(outputs, dim=1)

补充说明一下：预测阶段的设计经历了多次实验调整。最初我把预测器的输入设为编码器输入的最后一个时间步的原始数据（也就是真实观测的最后一帧），效果反而不好，因为真实观测帧的通道数与预测器隐状态通道不匹配，强行做卷积会有问题。后来改为直接把编码器隐状态当作输入，这样维度匹配，且信息量更充足。这个改动对模型收敛速度和最终精度都有明显提升。

关于网络深度，也值得多说两句。我开始尝试过三层编码器，但训练时间几乎翻倍，精度提升却基本可以忽略。序列长度24、网格16×16的规模，两层编码器的状态表示能力已经足够，加上一层预测器，一共3层ConvLSTM堆叠，这个结构在时间和空间复杂度上都非常均衡。模型总参数量约为70万，在单张GPU上训练完全没问题。

3.3 损失函数与评价指标：不止是一个MSE

选择损失函数时最容易犯的错误是只用一个MSE。MSE对数值较大的误差样本过于敏感，会导致模型过分关注极值时段，而忽略一般时段的预测精度。我最终使用的是MSE + MAE组合损失：

def hybrid_loss(pred, target): mse = torch.mean((pred - target) ** 2) mae = torch.mean(torch.abs(pred - target)) return mse + 0.5 * mae

MSE负责让预测值向真实值快速逼近，MAE提供平滑梯度信号，缓解MSE在误差较小时梯度消失的问题。系数0.5通过网格搜索做了简单调参，取整组数据上验证集指标最优的值。

评价指标上，除了常见的RMSE和MAE，我特别注意了预测超出合理范围的问题。由于标准化之后回归任务没有区间约束，模型有时会输出负值或远超最大观测值的预测结果，这在空气质量预测中是不合理的。我在预测之后加了物理约束修正：

# 反标准化后，强制物理范围 pred_pm25 = pred * std_pm25 + mean_pm25 pred_pm25 = torch.clamp(pred_pm25, min=0, max=500)

4. 训练细节与超参数调优记录

4.1 训练环境与配置

模型训练所用的环境是单张消费级GPU，显存12GB，完全够用。批次大小设为32，输入24个时间步的6通道16×16图像，显存占用大约2.5GB，留有充分的余量。

优化器选择了AdamW，而不是基础Adam。AdamW在权重衰减的处理上更规范，能有效防止大模型过拟合，训练后期loss下降也更平稳。初始学习率设为1e-3，这一数值适合大多数小到中规模的卷积循环网络。

学习率策略使用了CosineAnnealingLR，将学习率从1e-3逐步衰减到1e-5，周期80轮。相比固定学习率，这种方法让模型在后期能更精细地收敛到局部最优，也更好定位最优权重。

完整训练代码大概这样：

import torch.optim as optim from torch.utils.data import DataLoader, TensorDataset train_dataset = TensorDataset(torch.FloatTensor(X_train), torch.FloatTensor(y_train)) train_loader = DataLoader(train_dataset, batch_size=32, shuffle=True, drop_last=True) model = ConvLSTMModel() optimizer = optim.AdamW(model.parameters(), lr=1e-3, weight_decay=1e-5) scheduler = optim.lr_scheduler.CosineAnnealingLR(optimizer, T_max=80, eta_min=1e-5) loss_fn = hybrid_loss best_val_loss = float('inf') patience_counter = 0 for epoch in range(80): model.train() train_loss = 0 for x_batch, y_batch in train_loader: x_batch = x_batch.cuda() y_batch = y_batch.cuda() optimizer.zero_grad() pred = model(x_batch) loss = loss_fn(pred, y_batch) loss.backward() torch.nn.utils.clip_grad_norm_(model.parameters(), max_norm=1.0) optimizer.step() train_loss += loss.item() * x_batch.size(0) scheduler.step() # 验证集评估 model.eval() val_loss = 0 with torch.no_grad(): for x_batch, y_batch in val_loader: x_batch = x_batch.cuda() y_batch = y_batch.cuda() pred = model(x_batch) val_loss += loss_fn(pred, y_batch).item() * x_batch.size(0) train_loss /= len(train_loader.dataset) val_loss /= len(val_loader.dataset) if val_loss < best_val_loss: best_val_loss = val_loss torch.save(model.state_dict(), 'best_convlstm.pt') patience_counter = 0 else: patience_counter += 1 if patience_counter >= 10: print(f"Early stopping at epoch {epoch}") break print(f"Epoch {epoch}: train_loss={train_loss:.4f}, val_loss={val_loss:.4f}, lr={scheduler.get_last_lr()[0]:.2e}")

4.2 主要超参数调优过程与实验结果

初始模型跑出的结果只能算中规中矩：测试集RMSE约为11.3μg/m³。这个数字初看能接受，但对比基线LSTM模型的12.8μg/m³提升并不明显。问题出在哪里？我开始逐一排查超参数的影响。

首先调整的是卷积核尺寸。把第一层卷积核从3×3改成5×5后，RMSE下降了约0.4。原因也不难理解，更大的感受野让模型能独立感知更大范围内的浓度梯度，相对减弱了对多层堆叠扩大感受野的依赖。不过5×5的参数量是3×3的2.8倍，第二层还是保持了3×3，避免整体开销过大。

其次是隐藏状态通道数。把两层的64通道调整为128+64之后，RMSE下降接近0.7。但同时训练时间从40分钟撑到80分钟。权衡之后我选择了128+64这个组合，因为预测精度确实有了实质提升。

最后一处关键调整是输入序列长度。从24小时缩减到12小时时，RMSE从10.2μg/m³上升到10.9。延长到36小时，RMSE只有牛顿量级的微降，但训练速度明显下降。这个结果告诉我们，24小时的历史窗口已经足以让模型掌握污染物的日变化规律，不是窗口越长越好。

各个配置下的测试集指标对比如下：

最终模型在测试集上的R²稳定在0.82左右。相比基线LSTM的0.71，提升非常显著，尤其是空间结构的利用能力体现得特别明显——单站点时间序列模型在捕捉区域污染传播事件时频繁出错，而ConvLSTM很少犯这类错误。

4.3 空间预测效果与典型时段复盘

训练完成后，我挑选了几段特定天气条件下的预测进行对比分析。

第一个是冬季静稳天气场景。伦敦冬季经常出现高压控制、风速极低的情况，污染物在北京时间下午到夜间逐渐累积。我在这个时段观察到的模型预测结果是：空间分布形态非常逼近真实，尤其是污染高值区从市中心向周边扩散的过程，模型能准确识别并延续这一趋势。24小时预测的峰值位置和量级都基本正确，误差主要体现在高值区的极值偏弱约12%左右。

第二个是对春季沙尘过境事件的预测。这个情况难度明显更大。沙尘带是快速移动的窄带结构，可能在某一个小时内从北到南跨越整个伦敦区域，属于极端天气下的突发性事件。ConvLSTM对这种事件的响应不够灵敏，因为训练数据中类似的极端事件占比很低，模型没有充分学习到这种快速移动的结构形态。预测的沙尘带位置滞后了大约2到3小时，峰值也偏到真实值的一半以下。

这个案例很好地说明了ConvLSTM的适用边界：它擅长捕捉"有规律的演化过程"，比如日变化、天气系统驱动下的缓慢演变、污染累积与消散的正常循环；但对"突变的极端过程"，比如沙尘层爆发、突发的工业事故排放这类罕见事件，它的能力会有明显局限。这也是后续值得升级到混合模型的方向。

5. 常见问题与排查技巧实录

5.1 训练不收敛或损失震荡

如果你在训练初期就遇到loss剧烈震荡，优先级最高检查项是学习率。1e-3在Batch Size=32时一般没问题，但如果你显存不足无奈把Batch Size降到8，就一定要把学习率相应降低到3e-4左右。Batch Size变小等于每个批次样本的梯度方差变大，学习率不变会让步长变得过大。

另外一个容易被忽略的问题是标准化参数的取值范围。如果你用了Min-Max归一化，把数据压到[0,1]，那输出层用线性激活时，初始化权重太大很容易输出超出[0,1]范围的值，导致损失函数永远在较大值附近震荡。我在项目中改用Z-Score标准化后，这类问题基本消失。

还有一个被反复讨论的原因是：网格场中存在大面积常数区域。伦敦周边远郊的部分区域没有监测站点覆盖，插值后整片区域的值完全相同。这部分常数区域在空间上贡献很多梯度，但却是毫无意义的。解决方案是把插值范围从整个矩形网格收缩到"有实际覆盖"的站点凸包区域，或者用掩码把边缘区域权重置零。我用后者，计算损失时只对有效网格区域求均值。

def masked_rmse(pred, target, mask): diff = (pred - target) ** 2 diff_masked = diff * mask.unsqueeze(0).unsqueeze(0).unsqueeze(0) return torch.sqrt(diff_masked.sum() / mask.sum())

5.2 显存不足与长序列训练

如果你需要处理更长的时间窗口或更大的网格，建议优先使用梯度检查点（Gradient Checkpointing）而不是缩小Batch Size。梯度检查点会在反向传播时重新计算前向传播的中间结果，用时间换显存，能让训练序列长度和批次同时扩大约3倍。PyTorch中只需对每个ConvLSTMCell的forward加上torch.utils.checkpoint.checkpoint包裹即可。

另一个技巧是使用混合精度训练。16位浮点数能砍半显存占用，且在该项目规模下无精度损失，还可以获得接近30%的训练加速。配合GradScaler使用，是显存紧张时性价比最高的方案。

5.3 预测结果出现明显空间伪影

这个问题在调整卷积核后更容易遇到。模型预测的浓度场上偶尔会出现不自然的条纹或棋盘格状伪影，尤其集中在网格边缘位置。排查后的根因是插值阶段引入的边缘效应——边缘网格点卷积运算时padding补零，与有效的真实值区域特征分布差异太大，卷积核学到了一种奇怪的边缘响应。

一个简单有效的缓解手段是在损失函数中增加总变差正则项（TV Loss）：

def tv_loss(pred): batch, steps = pred.shape[:2] diff_h = torch.abs(pred[:, :, :, 1:, :] - pred[:, :, :, :-1, :]) diff_w = torch.abs(pred[:, :, :, :, 1:] - pred[:, :, :, :, :-1]) return (diff_h.mean() + diff_w.mean()) / 2

把tv_loss乘以0.05加在主损失后面，能让预测场趋于平滑，伪影明显减弱。同时不会过度模糊浓度场的真实细节。

5.4 与其他模型对比：ConvLSTM的优势与短板

作为实战项目，我把它和几种常见方法做了一次横向对比。

纯CNN虽然也是处理图像序列，但本质是做独立的帧预测，没有跨时间步的状态记忆，所以实际效果反而比LSTM差。ConvLSTM的价值就在于"把空间的卷积和时间记忆融为一体"，两个维度同时起作用，而不是简单叠加。

5.5 代码级性能优化建议

如果你要把训练规模扩大到更大的网格或更长的序列，有几个代码层面的优化很值得做：

第一，把多步循环中的卷积合并。当前实现中的两个ConvLSTM单元各做一次卷积，你可以用einsum把输入到门控的4个卷积合并成一次操作，能省不少GPU核时间。

第二，使用torch.jit.script对模型进行编译加速。ConvLSTM模型的循环结构在Graph模式下会有显著优化空间，实测脚本化之后前向推理速度提升约25%。

第三，数据加载环节一定要做pin_memory=True和num_workers>0，否则时间序列的切片和标准化操作会成为训练瓶颈。

优化后的完整推理代码可以参考下面的示意：

@torch.jit.script def predict_future(model, input_seq): model.eval() with torch.no_grad(): pred = model(input_seq) return pred

写在最后的实操体会

这次做完整个算法实施，我最大的感触是：模型选择并不需要追求最新最复杂，而是要匹配数据的真实结构和任务的实际需求。ConvLSTM在2015年提出，已经算"老"模型了，但它恰好解决了空气质量预测的核心矛盾——既要捕捉污染物的时间累积与消散周期，又要理解空间上的迁移与扩散规律。在数据量中等、无需超大规模计算的前提条件下，它反而是性价比极高的方案。

后续如果想继续提升，我觉得有两个方向值得探索。第一是引入外部变量，比如气象场数据（风场、温度、湿度）作为额外的输入通道，让模型理解气象因素对污染物生成与消散的驱动作用，这比单纯加深网络更有效。第二是多尺度融合，用不同分辨率的网格捕捉城市尺度与区域尺度的空间特征，在现有ConvLSTM基础上做一个金字塔结构的变体。这些思路如果你也在做相关方向，欢迎一起交流踩坑经验。