使用卷积LSTM时如何忽略某些数据点？

Question

我希望使用动能和海面温度来预测洋流涡度。我的数据包括在湾流区域的卫星动能读数和表面温度读数。我计划使用将递归体系结构（LSTM）与卷积网络模型相结合的混合神经网络。

我的数据集包括1996年至2018年的每日动能和温度读数，总共8036x80x120网格。例如，给定kinetic energy和temperature，我希望NN预测vorticity。

我的问题是，我该如何影响我的神经网络以忽略/打折陆地地形而仅关注海洋数据。陆地地形数据以NaN值的形式存储在我的图像数组中。

我正在使用PyTorch。

Answer 1

All you need is attention将为您寻找所需的解决方案。该机制将帮助您专注于输入中的某些功能/块。 medium post提供了有关如何将此机制应用于图像的教程。