跳到正文
arXiv Artificial Intelligence· Augusta van Haren, Hanna Hoogen, Luca Pattavina, Quincy Rommen·· 18 小时前AI 评分22

ST-GRU PredRNN-V2:用 GRU 重构 PredRNN-V2 做视频预测

ST-GRU PredRNN-V2: A GRU-Based Reformulation of PredRNN-V2 for Video Prediction

AI 导读

研究者提出 ST-GRU 单元,用 GRU 式的更新门和重置门替代 ST-LSTM 每个记忆状态的三个门,并将时间记忆状态并入隐藏状态,参数量减少约 5%。将其嵌入 PredRNN-V2 后在 Moving MNIST 基准上按原实验协议评测,十帧平均 MSE 55.1、SSIM 0.883、LPIPS 0.065、PSNR 19.53dB。

来源:arXiv Artificial Intelligence · arxiv.org