English
Related papers

Related papers: Why and How Auxiliary Tasks Improve JEPA Represent…

200 papers

Joint Embedding Predictive Architectures (JEPA) are a novel self supervised training technique that have shown recent promise across domains. We introduce BERT-JEPA (BEPA), a training paradigm that adds a JEPA training objective to…

Computation and Language · Computer Science 2026-01-05 Taj Gillin , Adam Lalani , Kenneth Zhang , Marcel Mateos Salles

In this work, we introduce Mask-JEPA, a self-supervised learning framework tailored for mask classification architectures (MCA), to overcome the traditional constraints associated with training segmentation models. Mask-JEPA combines a…

Computer Vision and Pattern Recognition · Computer Science 2024-07-16 Dong-Hee Kim , Sungduk Cho , Hyeonwoo Cho , Chanmin Park , Jinyoung Kim , Won Hwa Kim

Trajectory similarity computation is an essential technique for analyzing moving patterns of spatial data across various applications such as traffic management, wildlife tracking, and location-based services. Modern methods often apply…

Machine Learning · Computer Science 2024-06-21 Lihuan Li , Hao Xue , Yang Song , Flora Salim

While auxiliary tasks play a key role in shaping the representations learnt by reinforcement learning agents, much is still unknown about the mechanisms through which this is achieved. This work develops our understanding of the…

Machine Learning · Computer Science 2021-02-26 Clare Lyle , Mark Rowland , Georg Ostrovski , Will Dabney

Model-based reinforcement learning (MBRL) agents operating in high-dimensional observation spaces, such as Dreamer, rely on learning abstract representations for effective planning and control. Existing approaches typically employ…

Machine Learning · Computer Science 2026-04-15 Michael Hauri , Friedemann Zenke

Aerodynamic surrogate models are increasingly used to replace repeated high-fidelity CFD evaluations in many-query design settings, but current approaches still face two important limitations: they often scale poorly to the very large…

This paper presents that the masked-modeling principle driving the success of large foundational vision models can be effectively applied to audio by making predictions in a latent space. We introduce Audio-based Joint-Embedding Predictive…

Sound · Computer Science 2024-01-12 Zhengcong Fei , Mingyuan Fan , Junshi Huang

Real-world reinforcement learning (RL) environments, whether in robotics or industrial settings, often involve non-visual observations and require not only efficient but also reliable and thus interpretable and flexible RL approaches. To…

Machine Learning · Computer Science 2024-02-19 Moritz Lange , Noah Krystiniak , Raphael C. Engelhardt , Wolfgang Konen , Laurenz Wiskott

Motivated by the challenge of seamless cross-dataset transfer in EEG signal processing, this article presents an exploratory study on the use of Joint Embedding Predictive Architectures (JEPAs). In recent years, self-supervised learning has…

Machine Learning · Computer Science 2024-10-10 Pierre Guetschel , Thomas Moreau , Michael Tangermann

Multivariate time series underpin modern critical infrastructure, making the prediction of anomalies a vital necessity for proactive risk mitigation. While Joint-Embedding Predictive Architectures (JEPA) offer a promising framework for…

Machine Learning · Computer Science 2026-02-05 Yanan He , Yunshi Wen , Xin Wang , Tengfei Ma

We introduce a two-stage self-supervised framework that combines the Joint-Embedding Predictive Architecture (JEPA) with a Density Adaptive Attention Mechanism (DAAM) for learning robust speech representations. Stage~1 uses JEPA with DAAM…

Image-based Joint-Embedding Predictive Architecture (I-JEPA) offers a promising approach to visual self-supervised learning through masked feature prediction. However with the inherent visual uncertainty at masked positions, feature…

Machine Learning · Computer Science 2026-05-06 Chen Huang , Xianhang Li , Vimal Thilak , Etai Littwin , Josh Susskind

Semi-supervised learning has emerged as a powerful paradigm for leveraging large amounts of unlabeled data to improve the performance of machine learning models when labeled data are scarce. Among existing approaches, methods derived from…

Machine Learning · Computer Science 2026-04-29 Ali Aghababaei-Harandi , Aude Sportisse , Massih-Reza Amini

In this paper, we explore an approach to auxiliary task discovery in reinforcement learning based on ideas from representation learning. Auxiliary tasks tend to improve data efficiency by forcing the agent to learn auxiliary prediction and…

Machine Learning · Computer Science 2024-07-23 Banafsheh Rafiee , Sina Ghiassian , Jun Jin , Richard Sutton , Jun Luo , Adam White

This paper demonstrates an approach for learning highly semantic image representations without relying on hand-crafted data-augmentations. We introduce the Image-based Joint-Embedding Predictive Architecture (I-JEPA), a non-generative…

Computer Vision and Pattern Recognition · Computer Science 2023-04-14 Mahmoud Assran , Quentin Duval , Ishan Misra , Piotr Bojanowski , Pascal Vincent , Michael Rabbat , Yann LeCun , Nicolas Ballas

Autonomous driving, as an agent operating in the physical world, requires the fundamental capability to build \textit{world models} that capture how the environment evolves spatiotemporally in order to support long-term planning. At the…

Computer Vision and Pattern Recognition · Computer Science 2026-02-16 Haoran Zhu , Anna Choromanska

Representation alignment (REPA) guides generative training by distilling representations from a strong, pretrained vision encoder to intermediate diffusion features. We investigate a fundamental question: what aspect of the target…

Computer Vision and Pattern Recognition · Computer Science 2025-12-12 Jaskirat Singh , Xingjian Leng , Zongze Wu , Liang Zheng , Richard Zhang , Eli Shechtman , Saining Xie

Current multimodal learning strategies primarily optimize in the original token space. Such a framework is easy to incorporate with the backbone of pretrained language model, but might result in modality collapse. To alleviate such issues,…

Machine Learning · Computer Science 2025-06-19 Hongyang Lei , Xiaolong Cheng , Qi Qin , Dan Wang , Kun Fan , Huazhen Huang , Qingqing Gu , Yetao Wu , Zhonglin Jiang , Yong Chen , Luo Ji

We present Clin-JEPA, a multi-phase co-training framework for joint-embedding predictive (JEPA) pretraining on EHR patient trajectories. JEPA architectures have enabled latent-space planning in robotics and high-quality representation…

Video representation learning is an increasingly important topic in machine learning research. We present Video JEPA with Variance-Covariance Regularization (VJ-VCR): a joint-embedding predictive architecture for self-supervised video…

Computer Vision and Pattern Recognition · Computer Science 2024-12-17 Katrina Drozdov , Ravid Shwartz-Ziv , Yann LeCun