arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2511.18424cs.CV

跨模态联合嵌入预测架构CrossJEPA:用于从2D图像高效学习3D表示的架构

CrossJEPA: Cross-Modal Joint-Embedding Predictive Architecture for Efficient 3D Representation Learning from 2D Images

University of Moratuwa(摩图瓦大学);Technische Universität Darmstadt(达姆施塔特技术大学)

Avishka Perera, Kumal Hewagamage, Saeedha Nazar, Kavishka Abeywardana, Hasitha Gallella, Ranga Rodrigo, Mohamed Afham

AI总结:

CrossJEPA通过跨模态联合嵌入预测架构,利用图像基础模型知识,实现高效3D表示学习,达到SOTA性能,且训练高效、内存占用低。

摘要:

图像到点的跨模态学习已兴起,以解决3D表示学习中大规模3D数据集稀缺的问题。然而,目前利用2D数据的方法往往导致大型、训练缓慢的模型,使其在资源受限环境中部署困难。此类模型的架构设计因此至关重要,决定了其性能、内存占用和计算效率。联合嵌入预测架构(JEPA)因其简单性和效率在自监督学习中广受欢迎,但在跨模态设置中却研究较少,部分原因是人们误认为掩码是JEPA的固有特性。在此背景下,我们提出了CrossJEPA,一种简单的跨模态联合嵌入预测架构,利用图像基础模型的知识,并训练一个预测器来推断特定渲染的2D视图的嵌入表示,从而从点云对应的嵌入表示中引入JEPA风格的预训练策略。通过将预测器条件化于跨域投影信息,CrossJEPA从目标域专属的语义中净化监督信号。我们进一步利用冻结教师设计与一次目标嵌入缓存机制,实现了一种折衷效率。CrossJEPA在合成ModelNet40(94.2%)和现实世界ScanObjectNN(88.3%)基准上实现了新的SOTA,仅使用14.1M预训练参数(8.5M在点编码器中),并在标准单GPU上约6小时预训练时间。这些结果使CrossJEPA成为一种性能优异、内存高效且训练快速的3D表示学习框架,通过知识蒸馏实现。我们从直观、理论和经验上分析了CrossJEPA,并广泛消除了我们的设计选择。代码将被提供。

英文摘要:

Image-to-point cross-modal learning has emerged to address the scarcity of large-scale 3D datasets in 3D representation learning. However, current methods that leverage 2D data often result in large, slow-to-train models, making them computationally expensive and difficult to deploy in resource-constrained environments. The architecture design of such models is therefore critical, determining their performance, memory footprint, and compute efficiency. The Joint-embedding Predictive Architecture (JEPA) has gained wide popularity in self-supervised learning for its simplicity and efficiency, but has been under-explored in cross-modal settings, partly due to the misconception that masking is intrinsic to JEPA. In this light, we propose CrossJEPA, a simple Cross-modal Joint Embedding Predictive Architecture that harnesses the knowledge of an image foundation model and trains a predictor to infer embeddings of specific rendered 2D views from corresponding 3D point clouds, thereby introducing a JEPA-style pretraining strategy beyond masking. By conditioning the predictor on cross-domain projection information, CrossJEPA purifies the supervision signal from semantics exclusive to the target domain. We further exploit the frozen teacher design with a one-time target embedding caching mechanism, yielding amortized efficiency. CrossJEPA achieves a new state-of-the-art in linear probing on the synthetic ModelNet40 (94.2%) and the real-world ScanObjectNN (88.3%) benchmarks, using only 14.1M pretraining parameters (8.5M in the point encoder), and about 6 pretraining hours on a standard single GPU. These results position CrossJEPA as a performant, memory-efficient, and fast-to-train framework for 3D representation learning via knowledge distillation. We analyze CrossJEPA intuitively, theoretically, and empirically, and extensively ablate our design choices. Code will be made available.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑