arXiv:2509.24317cs.LGcs.CV2025-09被引 10

用固定教师模型替代动态更新,让视频自监督学习更高效透明

Rethinking JEPA: Compute-Efficient Video SSL with Frozen Teachers

  • 教师模型冻结后直接预测掩码区域的潜在表示
  • 相同训练算力下,性能超越最新V-JEPA 2模型
  • 学生模型对教师质量不敏感,适合大规模扩展

视频联合嵌入预测架构(V-JEPA)通过指数移动平均(EMA)更新的教师模型,在潜空间中预测被掩码区域,以学习通用视频表征。然而EMA机制复杂且耦合师生结构。本文提出SALT(静态教师非对称潜空间训练):先用像素重建目标训练目标编码器,再冻结教师,让学生在掩码区域预测教师潜表示。该方法解耦优化过程,提升透明性、效率与可扩展性。实验表明,相同预训练计算量下,学生模型在多个基准上优于V-JEPA 2;其准确率-算力权衡曲线全面领先。更重要的是,学生性能对教师质量不敏感——即使使用小型低效教师,仍能获得高性能学生。这提示应将算力主要分配给学生。SALT为视频表示学习提供了一种简单、可扩展、高效的替代方案。

原文摘要 · Abstract (English)

Video Joint Embedding Predictive Architectures (V-JEPA) learn generalizable off-the-shelf video representation by predicting masked regions in latent space with an exponential moving average (EMA)-updated teacher. While EMA prevents representation collapse, it complicates scalable model selection and couples teacher and student architectures. We revisit masked-latent prediction and show that a frozen teacher suffices. Concretely, we (i) train a target encoder with a simple pixel-reconstruction objective under V-JEPA masking, then (ii) freeze it and train a student to predict the teacher's latents on masked regions. This leads to a two-stage, unregularized scheme that we refer to as SALT (Static-teacher Asymmetric Latent Training). SALT decouples optimization into pixel reconstruction (teacher) and masked latent prediction (student), increasing transparency, efficiency, and scalability while preserving the ability of representation to generalize under frozen evaluation. Empirically, our student models outperform recently proposed V-JEPA 2 encoders under frozen backbone evaluation across diverse benchmarks. They are also more compute-optimal: at matched pretraining FLOPs, our method achieves higher probing accuracy, and its scaling curves dominate V-JEPA's accuracy-FLOPs Pareto frontier. Finally, we find that student quality is remarkably robust to teacher quality: high-performing students emerge even with small, sub-optimal teachers. This points to a compute budget allocation that should overwhelmingly favor the student. These results position SALT as a simple, scalable, and compute-efficient alternative to EMA-based self-distillation for video representation learning.

视频自监督模型压缩计算效率

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。