arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2212.11642cs.CVcs.AI

基于预测编码的带编解码LSTM的多尺度网络用于视频预测

Predictive Coding Based Multiscale Network with Encoder-Decoder LSTM for Video Prediction

South China University of Technology(华南理工大学);The Hong Kong Polytechnic University(香港理工大学)

Chaofan Ling, Junpei Zhong, Weihua Li

AI总结:

针对传统预测编码模型无法预测未来的问题,提出嵌入编解码LSTM的多尺度预测编码视频预测模型,改进训练策略,在多个数据集上取得良好性能。

摘要:

我们提出了一种用于未来视频帧预测的多尺度预测编码模型。受认知科学中“预测编码”理论的启发,该模型通过自底向上和自顶向下的信息流组合进行更新,能够增强不同网络层级之间的交互。然而,传统预测编码模型仅分层级预测当前发生的情况,而非预测未来。为解决该问题,我们的模型采用多尺度方法(从粗到细):高层神经元生成更粗的预测(更低分辨率),低层神经元生成更精细的预测(更高分辨率)。\n在网络架构方面,我们将编解码网络直接嵌入LSTM模块内部,并在不同网络层级间共享最终编码的高层语义信息。与传统的Encoder-LSTM-Decoder架构相比,这实现了当前输入与LSTM历史状态之间的全面交互,从而学习到更可靠的时空依赖关系。此外,为应对对抗训练中的不稳定性,并缓解长期预测中预测误差的累积,我们对训练策略提出了若干改进。我们的方法在KTH、Moving MNIST和Caltech Pedestrian等数据集上取得了良好性能。代码可在https://github.com/Ling-CF/MSPN获取。

英文摘要:

We present a multi-scale predictive coding model for future video frames prediction. Drawing inspiration on the ``Predictive Coding" theories in cognitive science, it is updated by a combination of bottom-up and top-down information flows, which can enhance the interaction between different network levels. However, traditional predictive coding models only predict what is happening hierarchically rather than predicting the future. To address the problem, our model employs a multi-scale approach (Coarse to Fine), where the higher level neurons generate coarser predictions (lower resolution), while the lower level generate finer predictions (higher resolution). In terms of network architecture, we directly incorporate the encoder-decoder network within the LSTM module and share the final encoded high-level semantic information across different network levels. This enables comprehensive interaction between the current input and the historical states of LSTM compared with the traditional Encoder-LSTM-Decoder architecture, thus learning more believable temporal and spatial dependencies. Furthermore, to tackle the instability in adversarial training and mitigate the accumulation of prediction errors in long-term prediction, we propose several improvements to the training strategy. Our approach achieves good performance on datasets such as KTH, Moving MNIST and Caltech Pedestrian. Code is available at https://github.com/Ling-CF/MSPN.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑