Audio-JEPA: Joint-Embedding Predictive Architecture for Audio Representation Learning
Abstract
Building on the Joint-Embedding Predictive Architecture (JEPA) paradigm, a recent self-supervised learning framework that predicts latent representations of masked regions in high-level feature spaces, we propose Audio-JEPA (Audio Joint-Embedding Predictive Architecture), tailored specifically for audio data. Audio-JEPA uses a simple Vision Transformer backbone to predict latent representations of masked spectrogram patches rather than reconstructing raw audio. We pre-train on unlabeled AudioSet clips (10s, 32kHz) with random patch masking on mel-spectrograms. We evaluate on the X-ARES suite covering speech, music, and environmental sound tasks. Although our implementation is a straightforward translation of the original model to audio, the results still show comparable performance to wav2vec 2.0 and data2vec while using less than one-fifth of their training data and with no hyper-parameter tuning. All code and pretrained checkpoints will be released on GitHub.
Keywords
Cite
@article{arxiv.2507.02915,
title = {Audio-JEPA: Joint-Embedding Predictive Architecture for Audio Representation Learning},
author = {Ludovic Tuncay and Etienne Labbé and Emmanouil Benetos and Thomas Pellegrini},
journal= {arXiv preprint arXiv:2507.02915},
year = {2025}
}