TKN: Transformer-based Keypoint Prediction Network For Real-time Video Prediction

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Li, Haoran, Li, XiaoLu, Lin, Yihang, Hao, Yanbin, Xie, Haiyong, Zhou, Pengyuan, Liao, Yong
Format: Preprint
Published: 2023
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866918338504949760
author Li, Haoran
Li, XiaoLu
Lin, Yihang
Hao, Yanbin
Xie, Haiyong
Zhou, Pengyuan
Liao, Yong
author_facet Li, Haoran
Li, XiaoLu
Lin, Yihang
Hao, Yanbin
Xie, Haiyong
Zhou, Pengyuan
Liao, Yong
contents Video prediction is a complex time-series forecasting task with great potential in many use cases. However, traditional methods prioritize accuracy and overlook slow prediction speeds due to complex model structures, redundant information, and excessive GPU memory consumption. These methods often predict frames sequentially, making acceleration difficult and limiting their applicability in real-time scenarios like danger prediction and warning.Therefore, we propose a transformer-based keypoint prediction neural network (TKN). TKN extracts dynamic content from video frames in an unsupervised manner, reducing redundant feature computation. And, TKN uses an acceleration matrix to reduce the computational cost of attention and employs a parallel computing structure for prediction acceleration. To the best of our knowledge, TKN is the first real-time video prediction solution that achieves a prediction rate of 1,176 fps, significantly reducing computation costs while maintaining other performance. Qualitative and quantitative experiments on multiple datasets have demonstrated the superiority of our method, suggesting that TKN has great application potential.
format Preprint
id arxiv_https___arxiv_org_abs_2303_09807
institution arXiv
publishDate 2023
record_format arxiv
spellingShingle TKN: Transformer-based Keypoint Prediction Network For Real-time Video Prediction
Li, Haoran
Li, XiaoLu
Lin, Yihang
Hao, Yanbin
Xie, Haiyong
Zhou, Pengyuan
Liao, Yong
Computer Vision and Pattern Recognition
Artificial Intelligence
Video prediction is a complex time-series forecasting task with great potential in many use cases. However, traditional methods prioritize accuracy and overlook slow prediction speeds due to complex model structures, redundant information, and excessive GPU memory consumption. These methods often predict frames sequentially, making acceleration difficult and limiting their applicability in real-time scenarios like danger prediction and warning.Therefore, we propose a transformer-based keypoint prediction neural network (TKN). TKN extracts dynamic content from video frames in an unsupervised manner, reducing redundant feature computation. And, TKN uses an acceleration matrix to reduce the computational cost of attention and employs a parallel computing structure for prediction acceleration. To the best of our knowledge, TKN is the first real-time video prediction solution that achieves a prediction rate of 1,176 fps, significantly reducing computation costs while maintaining other performance. Qualitative and quantitative experiments on multiple datasets have demonstrated the superiority of our method, suggesting that TKN has great application potential.
title TKN: Transformer-based Keypoint Prediction Network For Real-time Video Prediction
topic Computer Vision and Pattern Recognition
Artificial Intelligence
url https://arxiv.org/abs/2303.09807