AIEarth人工智能创新挑战赛是阿里达摩院联合南京信息工程大学、国家气候中心、国家海洋环境预报中心、阿里云天池平台等机构共同举办的大赛。“AI Earth”人工智能创新挑战赛-AI赋能厄尔尼诺预测
目前比赛已经完美收官,并且已经发布了决赛现场的路演视频录屏|AI Earth人工智能创新挑战赛。目前,决赛第二名吴先生的队伍已在GitHub发布了源代码。吴先生的队伍在B榜排名第一,最终决赛路演取得了第二名。
雷锋网在比赛相关队伍模型的介绍中指出,第二名队伍在SA-ConvLSTM 和ConvLSTM的基础上增加了Self-attention memory module,即一个对空间维度采取注意力机制的模块,和一个可记忆非局部时空特征的memory unit M。通过传统的ConvLSTM计算得到的H会再通过Self-attention memory module得到最终SA-ConvLSTM的输出和下一个时刻的M。
图注:自我注意力记忆模块
图注:自我注意力 ConvLSTM (SA-ConvLSTM) 模块
该模型充分发现了自注意力机制在较长时间尺度的气候演变时空预报中的作用。
创新之处在于,考虑了空间非局部的、时间更连续的时空结构演变特征,整个西太平洋海温时空结构演变的信息,用时空预报作为时间序列预报的进一步扩充,并利用自注意力机制充分提取时空结构信息来做预报。
第二名队伍发布了比赛中所使用的三个模型源码,包括Conv-TT-lstm、SA-ConvLSTM、TimeSformer,其中SA-ConvLSTM为最佳模型,同时给出了完整的训练流程,感兴趣的可以前往GitHub下载学习。
此次比赛应该算是国内气象领域又一次高规格的AI大赛,能有Top3队伍开源代码还是很难得的。对于入门选手以及想要学习Top队伍比赛技巧的同学而言也是很好的机会。希望以后类似的比赛能有更多的源码开源~
可直接通过参考链接3下载,或在后台回复“AIEarth“获取完整代码。
参考链接:
1. https://www.leiphone.com/category/industrynews/i7QN2yBmHGBULaUu.html?viewType=weixin
2. https://ojs.aaai.org/index.php/AAAI/article/view/6819
3. https://github.com/jerrywn121/TianChi_AIEarth