清华团队雷达回波时空预测模型MIM复现

关于MIM的模型不进行过多介绍，感兴趣的可以去看原论文，或者看其他作者的解读，比如AI蜗牛车【时空序列预测第五篇】时空序列预测模型之Memory In Memory（学习高阶非平稳特征信息）

这个模型是之前打雷达回波外推比赛的时候复现的代码，一直没来得及整理。在复现过程中发现了一些比较有意思的地方，原作者发布了tf的代码，复现的时候对照论文和代码看了一下，tf代码有些地方和论文描述不是很一致（和其他人也交流过这篇文章的复现问题，也都表示存在不一致的地方），然后就按照论文中给出的公式和框架结构进行了复现。

模型代码为Pytorch，这里仅给出了模型的源代码（源代码已发布在GitHub）。下图是利用雷达回波数据的测试结果：

以下是源代码链接，感兴趣的可以测试一下，如有什么疑问欢迎提出一起交流学习。GitHub源: https://github.com/bugsuse/MIM_PyTorch

—END—