JEPA模型隐式学习数据密度,可直接用于异常检测与数据筛选
Gaussian Embeddings: How JEPAs Secretly Learn Your Data Density
- 利用雅可比矩阵计算样本的密度概率,无需额外训练
- 在合成数据、ImageNet等多场景验证有效,精度稳定
- 适合需要数据质量评估或异常检测的研究者
联合嵌入预测架构(JEPAs)能自适应解决多种下游任务。其结合两个目标:(i) 潜空间预测项,即轻微扰动样本的表示可由原样本表示预测;(ii) 反坍缩项,防止所有样本表示相同。我们发现,反坍缩项不仅防坍缩,还可严格估计数据密度。任何成功训练的JEPA均可高效、闭式计算样本 $x$ 的概率,用于数据清洗、异常检测或密度估计。该理论对数据集和架构均无依赖。我们在合成数据、控制数据及ImageNet上,以及I-JEPA、DINOv2等不同自监督方法和多模态模型(如MetaCLIP)中验证了结果。我们将提取密度的方法命名为 {f JEPA-SCORE}。
原文摘要 · Abstract (English)
Joint Embedding Predictive Architectures (JEPAs) learn representations able to solve numerous downstream tasks out-of-the-box. JEPAs combine two objectives: (i) a latent-space prediction term, i.e., the representation of a slightly perturbed sample must be predictable from the original sample's representation, and (ii) an anti-collapse term, i.e., not all samples should have the same representation. While (ii) is often considered as an obvious remedy to representation collapse, we uncover that JEPAs' anti-collapse term does much more--it provably estimates the data density. In short, any successfully trained JEPA can be used to get sample probabilities, e.g., for data curation, outlier detection, or simply for density estimation. Our theoretical finding is agnostic of the dataset and architecture used--in any case one can compute the learned probabilities of sample $x$ efficiently and in closed-form using the model's Jacobian matrix at $x$. Our findings are empirically validated across datasets (synthetic, controlled, and Imagenet) and across different Self Supervised Learning methods falling under the JEPA family (I-JEPA and DINOv2) and on multimodal models, such as MetaCLIP. We denote the method extracting the JEPA learned density as {\bf JEPA-SCORE}.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。