Soft Clustering Anchors for Self-Supervised Speech Representation Learning in Joint Embedding Prediction Architectures
Abstract
Joint Embedding Predictive Architectures (JEPA) offer a promising approach to self-supervised speech representation learning, but suffer from representation collapse without explicit grounding. We propose GMM-Anchored JEPA, which fits a Gaussian Mixture Model once on log-mel spectrograms and uses its frozen soft posteriors as auxiliary targets throughout training. A decaying supervision schedule allows GMM regularization to dominate early training before gradually yielding to the JEPA objective. Unlike HuBERT and WavLM, which require iterative re-clustering, our approach clusters input features once with soft rather than hard assignments. On ~50k hours of speech, GMM anchoring improves ASR (28.68% vs. 33.22% WER), emotion recognition (67.76% vs. 65.46%), and slot filling (64.7% vs. 59.1% F1) compared to a WavLM-style baseline with matched compute. Cluster analysis shows GMM-anchored representations achieve up to 98% entropy compared to 31% for WavLM-style, indicating substantially more uniform cluster utilization. Code is made available at https://github.com/gioannides/clustering-anchored-jepa.
Keywords
Cite
@article{arxiv.2602.09040,
title = {Soft Clustering Anchors for Self-Supervised Speech Representation Learning in Joint Embedding Prediction Architectures},
author = {Georgios Ioannides and Adrian Kieback and Judah Goldfeder and Linsey Pang and Aman Chadha and Aaron Elkins and Yann LeCun and Ravid Shwartz-Ziv},
journal= {arXiv preprint arXiv:2602.09040},
year = {2026}
}
Comments
15 pages, 5 figures. Code: github.com/gioannides/clustering-anchored-jepa