Test-Time Training on Video Streams

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Wang, Renhao, Sun, Yu, Tandon, Arnuv, Gandelsman, Yossi, Chen, Xinlei, Efros, Alexei A., Wang, Xiaolong
Formato: Preprint
Publicado: 2023
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866913634827894784
author Wang, Renhao
Sun, Yu
Tandon, Arnuv
Gandelsman, Yossi
Chen, Xinlei
Efros, Alexei A.
Wang, Xiaolong
author_facet Wang, Renhao
Sun, Yu
Tandon, Arnuv
Gandelsman, Yossi
Chen, Xinlei
Efros, Alexei A.
Wang, Xiaolong
contents Prior work has established Test-Time Training (TTT) as a general framework to further improve a trained model at test time. Before making a prediction on each test instance, the model is first trained on the same instance using a self-supervised task such as reconstruction. We extend TTT to the streaming setting, where multiple test instances - video frames in our case - arrive in temporal order. Our extension is online TTT: The current model is initialized from the previous model, then trained on the current frame and a small window of frames immediately before. Online TTT significantly outperforms the fixed-model baseline for four tasks, on three real-world datasets. The improvements are more than 2.2x and 1.5x for instance and panoptic segmentation. Surprisingly, online TTT also outperforms its offline variant that accesses strictly more information, training on all frames from the entire test video regardless of temporal order. This finding challenges those in prior work using synthetic videos. We formalize a notion of locality as the advantage of online over offline TTT, and analyze its role with ablations and a theory based on bias-variance trade-off.
format Preprint
id arxiv_https___arxiv_org_abs_2307_05014
institution arXiv
publishDate 2023
record_format arxiv
spellingShingle Test-Time Training on Video Streams
Wang, Renhao
Sun, Yu
Tandon, Arnuv
Gandelsman, Yossi
Chen, Xinlei
Efros, Alexei A.
Wang, Xiaolong
Computer Vision and Pattern Recognition
Machine Learning
Prior work has established Test-Time Training (TTT) as a general framework to further improve a trained model at test time. Before making a prediction on each test instance, the model is first trained on the same instance using a self-supervised task such as reconstruction. We extend TTT to the streaming setting, where multiple test instances - video frames in our case - arrive in temporal order. Our extension is online TTT: The current model is initialized from the previous model, then trained on the current frame and a small window of frames immediately before. Online TTT significantly outperforms the fixed-model baseline for four tasks, on three real-world datasets. The improvements are more than 2.2x and 1.5x for instance and panoptic segmentation. Surprisingly, online TTT also outperforms its offline variant that accesses strictly more information, training on all frames from the entire test video regardless of temporal order. This finding challenges those in prior work using synthetic videos. We formalize a notion of locality as the advantage of online over offline TTT, and analyze its role with ablations and a theory based on bias-variance trade-off.
title Test-Time Training on Video Streams
topic Computer Vision and Pattern Recognition
Machine Learning
url https://arxiv.org/abs/2307.05014