1 背景
这篇文章[1]想要解决的问题是预测一个区域短时间内的降水变化,在它之前的工作(2015年之前)还很少有采用机器学习的方法来做相关预测。由于预测的输入是时序雷达图等具有空间和时间关系的数据,因此文中提出了convolutional LSTM (ConvLSTM)模型,用这个模型可以捕获数据的时空依赖,进而提高模型的预测结果。
2 方法
2.1 问题定义
假设从系统得到的雷达图是一张2D的
的图像,每一个像素点有
个测量值(可以理解为特征值),因此整个数据可以表达为
,其中
表示观察到数据的域。加上定期记录观测结果,我们可以得到一系列具有时空依赖的数据
。预测问题可以转变为根据前
个观测数据来预测后续
个数据,形式化表达如下,
2.2 LSTM
传统的LSTM模型可以形式化如下,
2.3 Convolutional LSTM
由于我们的输入是
,其具有空间信息,因此只需要将传统LSTM的公式稍作修改,把
的权重计算替换成
的卷积计算即可,其形式化如下,
2.4 模型
模型是堆砌ConvLSTM的encoder-decoder结构,但从文中给出的描述我觉得作者设计的模型并没有下采样和上采样的经典UNet结构(可能因为那时候都是同一年2015年的工作因此没有借鉴)。模型如下图所示,其通过一个卷积层提取图像特征之后,送入多个ConvLSTM层进行时空特征提取,最后把这些ConvLSTM的输出concat起来再通过1x1卷积进行最后的预测。
上述模型可以形式化为,
3 实验
3.1 Moving-MNIST数据集
通过MNIST中的数值图像组合而成,其结果如下图所示,
我复现了论文中基于这个数据集的实验(pytorch),详见我的Github[2]。
3.2 Radar Echo Dataset
该数据集作者没有公开。
4 结论
这篇文中提出了一种可以提取时空特征的结构ConvLSTM,并且通过实验证明了其有效性。该结构其实可以应用到其他具有时空关系的数据上。
[1] https://arxiv.org/abs/1506.04214v1
[2] https://github.com/czifan/ConvLSTM.pytorch