arXiv:2607.04044cs.CVstat.ML2026-07

用孪生编码器提升JEPA模型的自监督表征学习效果

SiamJEPA: On the Role of Siamese Student Encoders in JEPA

论文配图:SiamJEPA: On the Role of Siamese Student Encoders in JEPA
图 1 · 摘自论文原文
  • 采用孪生编码器+EMA教师网络,增强表征可分性
  • 训练初期加速收敛,在有限数据下性能超越单编码器JEPA
  • 为自监督学习提供新架构思路,适合研究表征学习者

近期,联合嵌入预测架构(JEPA)因其在自监督表征学习中的潜力而受到广泛关注。与重建像素的掩码自编码器不同,JEPA通过预测被遮蔽区域的潜在嵌入来学习表征。现有方法如I-JEPA和V-JEPA通常在学生网络中使用单个编码器,而采用孪生编码器更契合脑启发式表征学习框架,但其在JEPA中的作用尚未深入探索。本文提出SiamJEPA,一种配备指数移动平均(EMA)教师网络的掩码孪生学生编码器。该模型可视为脑启发模型PhiNet的JEPA版本。在ImageNet线性探测上的大量实验表明,孪生编码器作为有效正则化项,提升了表征可分性,并加速了训练初期的学习过程。此外,SiamJEPA在训练预算有限时持续优于同类单编码器JEPA,且线性探测准确率高于需要更长训练时间的掩码自编码器(MAE)。结果表明,孪生学生编码器不仅是架构选择,更是预测表征学习的重要归纳偏置。这些发现为JEPA模型设计提供了新见解,表明引入孪生学生结构是一种简单而有效的自监督学习改进方法。

原文摘要 · Abstract (English)

Recently, Joint Embedding Predictive Architectures (JEPAs) have attracted significant attention in the computer vision and machine learning communities as a promising framework for self-supervised representation learning. Unlike masked autoencoders that reconstruct pixels, JEPA models learn representations by predicting latent embeddings of masked regions. Existing JEPA-based methods, such as I-JEPA and V-JEPA, typically employ a single encoder in the student network. In contrast, using Siamese encoders for student network is more naturally aligned with brain-inspired representation learning frameworks, yet their role in JEPA models remains largely unexplored. In this paper, we investigate the effect of Siamese student encoders in JEPA-based representation learning. To this end, we propose SiamJEPA, masked Siamese student encoders equipped with an exponential moving average (EMA) teacher network. SiamJEPA can also be viewed as a JEPA formulation of the brain-inspired representation learning model PhiNet. Through extensive experiments on ImageNet linear probing, we demonstrate that Siamese encoders act as an effective regularizer for the JEPA objective, improving representation separability and accelerating learning during the early stages of training. Furthermore, SiamJEPA consistently outperforms comparable single-encoder JEPA variants under limited training budgets and achieves higher linear probing accuracy than Masked Autoencoders (MAE) which requires longer training. Our findings reveal that Siamese student encoders are not merely an architectural choice but constitute an important inductive bias for predictive representation learning. These results provide new insights into the design of JEPA-based models and suggest that incorporating Siamese student architectures offers a simple yet effective approach for improving self-supervised representation learning.

自监督学习表征学习孪生网络JEPA

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