Self-Supervised Any-Point Tracking by Contrastive Random Walks

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Shrivastava, Ayush, Owens, Andrew
Formato: Preprint
Publicado: 2024
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866914956332498944
author Shrivastava, Ayush
Owens, Andrew
author_facet Shrivastava, Ayush
Owens, Andrew
contents We present a simple, self-supervised approach to the Tracking Any Point (TAP) problem. We train a global matching transformer to find cycle consistent tracks through video via contrastive random walks, using the transformer's attention-based global matching to define the transition matrices for a random walk on a space-time graph. The ability to perform "all pairs" comparisons between points allows the model to obtain high spatial precision and to obtain a strong contrastive learning signal, while avoiding many of the complexities of recent approaches (such as coarse-to-fine matching). To do this, we propose a number of design decisions that allow global matching architectures to be trained through self-supervision using cycle consistency. For example, we identify that transformer-based methods are sensitive to shortcut solutions, and propose a data augmentation scheme to address them. Our method achieves strong performance on the TapVid benchmarks, outperforming previous self-supervised tracking methods, such as DIFT, and is competitive with several supervised methods.
format Preprint
id arxiv_https___arxiv_org_abs_2409_16288
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Self-Supervised Any-Point Tracking by Contrastive Random Walks
Shrivastava, Ayush
Owens, Andrew
Computer Vision and Pattern Recognition
We present a simple, self-supervised approach to the Tracking Any Point (TAP) problem. We train a global matching transformer to find cycle consistent tracks through video via contrastive random walks, using the transformer's attention-based global matching to define the transition matrices for a random walk on a space-time graph. The ability to perform "all pairs" comparisons between points allows the model to obtain high spatial precision and to obtain a strong contrastive learning signal, while avoiding many of the complexities of recent approaches (such as coarse-to-fine matching). To do this, we propose a number of design decisions that allow global matching architectures to be trained through self-supervision using cycle consistency. For example, we identify that transformer-based methods are sensitive to shortcut solutions, and propose a data augmentation scheme to address them. Our method achieves strong performance on the TapVid benchmarks, outperforming previous self-supervised tracking methods, such as DIFT, and is competitive with several supervised methods.
title Self-Supervised Any-Point Tracking by Contrastive Random Walks
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2409.16288