Prototypical Transformer As Unified Motion Learners
Cheng Han, Yawen Lu, Guohao Sun, James Chenhao Liang, Zhiwen Cao, Qifan Wang, Qiang Guan, Sohail Dianat
Abstract
In this work, we introduce the Prototypical Transformer (ProtoFormer), a general and unified framework that approaches various motion tasks from a prototype perspective. ProtoFormer seamlessly integrates prototype learning with Transformer by thoughtfully considering motion dynamics, introducing two innovative designs. First, Cross-Attention Prototyping discovers prototypes based on signature motion patterns, providing transparency in understanding motion scenes. Second, Latent Synchronization guides feature representation learning via prototypes, effectively mitigating the problem of motion uncertainty. Empirical results demonstrate that our approach achieves competitive performance on popular motion tasks such as optical flow and scene depth. Furthermore, it exhibits generality across various downstream tasks, including object tracking and video stabilization.
BibTeX
@inproceedings{
han2024prototypical,
title={Prototypical Transformer As Unified Motion Learners},
author={Cheng Han and Yawen Lu and Guohao Sun and James Chenhao Liang and Zhiwen Cao and Qifan Wang and Qiang Guan and Sohail Dianat and Raghuveer Rao and Tong Geng and ZHIQIANG TAO and Dongfang Liu},
booktitle={Forty-first International Conference on Machine Learning},
year={2024},
url={https://openreview.net/forum?id=JOrLz5d7OW}
}