提出正交梯度优化器,让视频流学习不因顺序依赖而性能下降。
Learning from Streaming Video with Orthogonal Gradients
- 用正交梯度改造优化器,打破连续视频输入带来的相关性
- 在三个任务中均超越AdamW,提升流式视频表征性能
- 适用于任何优化器,尤其适合无法打乱数据的实时视频场景
我们研究从连续视频流中进行自监督表征学习的挑战。传统方法依赖打乱视频以满足独立同分布(IID)假设,但在连续输入场景下该假设失效,导致性能下降。我们在三个任务上验证了从打乱到顺序学习的性能损失:单视频表征方法DoRA、多视频数据集上的VideoMAE,以及未来视频预测。为应对这一问题,我们提出对标准优化器的几何改进——利用训练过程中的正交梯度来解耦批次间的相关性。该方法可适配任意优化器,本文在SGD和AdamW上进行了验证。实验表明,使用正交优化器后,模型在流式视频训练下的下游任务表现显著提升,在三个场景中均优于强基线AdamW。
原文摘要 · Abstract (English)
We address the challenge of representation learning from a continuous stream of video as input, in a self-supervised manner. This differs from the standard approaches to video learning where videos are chopped and shuffled during training in order to create a non-redundant batch that satisfies the independently and identically distributed (IID) sample assumption expected by conventional training paradigms. When videos are only available as a continuous stream of input, the IID assumption is evidently broken, leading to poor performance. We demonstrate the drop in performance when moving from shuffled to sequential learning on three tasks: the one-video representation learning method DoRA, standard VideoMAE on multi-video datasets, and the task of future video prediction. To address this drop, we propose a geometric modification to standard optimizers, to decorrelate batches by utilising orthogonal gradients during training. The proposed modification can be applied to any optimizer -- we demonstrate it with Stochastic Gradient Descent (SGD) and AdamW. Our proposed orthogonal optimizer allows models trained from streaming videos to alleviate the drop in representation learning performance, as evaluated on downstream tasks. On three scenarios (DoRA, VideoMAE, future prediction), we show our orthogonal optimizer outperforms the strong AdamW in all three scenarios.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。