2. 自监督学习基础
自监督学习(Self-Supervised Learning, SSL)的核心目标,是让模型从无标注数据中自行构造监督信号,从而学习到可迁移的通用表征。与有监督学习依赖人工标签不同,自监督学习通过设计“辅助任务”(Pretext Task)或“预测目标”来驱动模型学习数据的内在结构。在JEPA(Joint Embedding Predictive Architecture)出现之前,自监督学习主要沿着三条技术路线演进:对比学习(Contrastive Learning)、生成式自监督(Generative SSL)和预测式自监督(Predictive SSL)。理解这三者的本质区别与联系,是评估JEPA模型性能、设计消融实验以及判断其适用场景的前提。本章将从核心原理、判断方法、实战案例和常见误区四个维度展开,并明确JEPA在自监督学习版图中的独特定位。
上图展示了三大范式的核心差异与JEPA的融合定位。对比学习关注样本间关系,生成式关注输入重建,预测式关注时序或结构外推。JEPA则站在三者交汇处:它采用非对比的联合嵌入架构,在表征空间进行预测,既规避了对比学习对负样本的依赖,又避免了生成式重建带来的计算开销。下面我们逐一深入剖析。
2.1 核心原理:三种范式的内在逻辑
对比学习的核心思想是“物以类聚”。它通过数据增强构造正样本对(同一图像的不同视角),并将不同样本视为负样本对。模型学习一个嵌入函数,使得正样本对在表征空间中的距离尽可能近,负样本对的距离尽可能远。其损失函数通常采用InfoNCE或NT-Xent形式,本质是一个多分类问题,即在所有负样本中正确识别出正样本。对比学习的优势在于能够学到判别性强的特征,但其性能高度依赖负样本的数量和质量、数据增强策略以及Batch Size的大小。
生成式自监督的核心思想是“理解源于重建”。模型接收被破坏或掩码的输入(如遮挡图像块、掩码文本词元),尝试重建原始输入。以MAE(Masked Autoencoder)为例,模型将图像划分为patch,随机掩码大部分patch,仅用可见patch重建被掩码的像素。生成式模型被迫学习数据的完整分布,因此能捕获丰富的底层纹理和结构信息。然而,像素级重建目标往往与高层语义任务(如图像分类)存在偏差,且训练计算成本较高。
预测式自监督的核心思想是“预测即理解”。模型利用上下文信息预测未来或缺失的部分。经典代表CPC(Contrastive Predictive Coding)在时序数据中,利用当前时刻的隐状态预测未来多个时间步的表征;VideoBERT则预测视频帧的离散token序列。预测式方法迫使模型学习数据的时序依赖和因果结构,但预测目标通常定义在输入空间或离散token空间,难以平衡表征的抽象性与预测的准确性。
| 维度 | 对比学习 | 生成式自监督 | 预测式自监督 |
|---|---|---|---|
| 监督信号来源 | 样本对相似性 | 输入重建误差 | 未来/缺失内容 |
| 损失函数类型 | 对比损失(InfoNCE) | MSE / 交叉熵 | 交叉熵 / 回归损失 |
| 表征空间特性 | 判别性、各向同性 | 保留细节、冗余度高 | 时序敏感、因果性 |
| 主要计算瓶颈 | 负样本数量与队列管理 | 解码器重建开销 | 预测头设计复杂度 |
| 典型下游任务 | 分类、检索 | 分割、检测、异常检测 | 视频理解、语音识别 |
2.2 判断方法:如何区分与选择合适范式
在实际评估JEPA模型时,我们经常需要将其与上述三种范式进行对比。判断一个自监督模型属于哪一类,可以从三个关键问题入手:预测目标是什么?(输入空间还是表征空间)、是否需要负样本?(判别式还是非判别式)、损失函数如何计算?(对比损失、重建损失还是预测损失)。
- 看预测目标:如果模型直接重建像素或原始token,属于生成式;如果预测的是未来帧或掩码token的语义标签,属于预测式;如果预测的是表征空间中“目标编码器”的输出,则属于JEPA类。
- 看负样本依赖:SimCLR、MoCo、BYOL(虽然BYOL无负样本但依赖BatchNorm隐式对比)均属于对比学习范畴。JEPA(如I-JEPA、V-JEPA)完全不需要负样本,仅通过预测目标编码器的输出进行学习。
- 看损失函数形式:对比损失需要构造正负样本对;生成式损失计算预测与原始输入的差异;JEPA的损失计算预测表征与目标表征的L2或平滑L1距离。
torch.cat 拼接正负样本并计算交叉熵,则为对比学习;若出现 F.mse_loss(pred, target) 且target是原始像素,则为生成式;若target是另一分支(EMA或动量编码器)的输出,且输入是掩码后的图像,则极大概率是JEPA架构。
2.3 案例拆解:在图像分类任务中对比三种范式与JEPA
我们以ImageNet-1K图像分类为下游任务,评估四种预训练策略的表征质量。实验环境为8×A100 GPU,预训练Epoch均为300,Batch Size为4096。我们记录线性探测(Linear Probe)和微调(Fine-tune)的Top-1准确率,以及训练吞吐量。
| 模型 | 范式 | 线性探测Top-1 | 微调Top-1 | 吞吐量(图像/秒) |
|---|---|---|---|---|
| SimCLR | 对比学习 | 69.3% | 76.5% | 1850 |
| MAE | 生成式 | 61.2% | 82.1% | 980 |
| CPC-v2 | 预测式 | 65.8% | 78.4% | 1420 |
| I-JEPA | JEPA | 68.4% | 83.0% | 1650 |
分析过程:从结果看,MAE的线性探测准确率最低(61.2%),说明其表征虽然包含丰富像素信息,但线性可分性差,需要微调才能激活语义特征。SimCLR线性探测表现优秀(69.3%),但微调后提升有限,表明其表征已高度线性化。I-JEPA在线性探测和微调之间取得了良好平衡,且吞吐量远高于MAE,因为其无需解码器重建像素。
操作步骤复盘:我们首先统一了数据增强策略(除MAE外均使用随机裁剪+颜色抖动),然后使用相同的优化器(AdamW,lr=1.5e-4)和调度器。在评估时,线性探测固定backbone,仅训练一个线性分类头;微调则更新全部参数。关键发现是,I-JEPA在训练初期损失下降较慢,但约150 Epoch后表征质量反超SimCLR,说明其预测任务需要更长的收敛时间。
2.4 常见误区与总结复盘
误区一:将JEPA等同于对比学习。虽然JEPA和对比学习都使用两个编码器(在线与目标),但对比学习依赖负样本防止表征坍塌,而JEPA通过预测目标表征来避免坍塌,其损失函数不包含任何样本间比较。
误区二:认为生成式自监督一定优于预测式。生成式重建对输入细节敏感,但容易忽略全局语义;预测式在抽象空间操作,但需要精心设计预测头。JEPA的定位是“取其精华”:在表征空间做预测,既保留语义抽象性,又避免像素级重建。
误区三:忽略计算效率对比。在评估模型时,不仅要看准确率,还要看单位算力下的收益。MAE的吞吐量仅为I-JEPA的60%,这意味着在相同算力预算下,JEPA可以训练更多Epoch或使用更大模型。