用软聚类目标替代硬标签,实现无重聚类的自监督语音表征学习。
S-JEPA : Soft Clustering Anchors for Self-Supervised Speech Representation Learning

- 通过KL散度匹配高斯混合模型的软后验,避免硬聚类带来的语音模糊性丢失。
- 在90M参数以下模型中达到最低词错误率,且情感识别性能达HuBERT-Base的一半参数水平。
- 无需离线重聚类或教师蒸馏,训练过程连续,适合大规模语音任务部署。
自监督语音编码器通常通过预测掩码位置的离散硬聚类标签进行训练,这种做法会压缩类别边界处的声学模糊性,并需在迭代间中断训练以重新聚类整个语料库。我们提出S-JEPA,一种基于JEPA架构的编码器-预测器对,通过KL散度使预测结果与掩码位置上高斯混合模型(GMM)的软后验相匹配。训练分两阶段连续进行:第一阶段使用固定GMM对MFCC特征建模,第二阶段在线更新基于编码器特征的GMM,输入层由无标签信号自适应选择,从而消除了离线重聚类步骤及手动选定聚类层的需要。在SUPERB评估协议下,S-JEPA在参数量低于90M的自监督方法中实现了最低词错误率(WER),且在情感识别任务上达到与HuBERT-Base相当的性能,仅需其约一半参数。对保留语音的帧级熵分析显示,存在双峰分布,部分帧熵接近理想二分类均等情形,直接证明软目标有效保留了硬目标会破坏的声学模糊性。代码已开源。
原文摘要 · Abstract (English)
Self-supervised speech encoders are predominantly trained by predicting discrete hard cluster IDs at masked positions, a recipe that collapses acoustic ambiguity at category boundaries and requires interrupting training to re-cluster the entire corpus between iterations. We introduce S-JEPA, a JEPA-style encoder-predictor pair trained to match the soft posteriors of a Gaussian Mixture Model at masked positions via KL divergence. Training runs as one continuous optimization trajectory in two phases: a fixed GMM over MFCC features, then an online GMM over encoder features, with the input layer selected adaptively from a label-free signal, removing both the offline re-cluster step and the hand-tuned choice of which transformer layer to cluster on. Under the SUPERB protocol, S-JEPA achieves the lowest WER among evaluated SSL methods below 90M parameters and matches HuBERT-Base on emotion recognition at roughly half its parameter count, establishing a new Pareto frontier without offline re-clustering or teacher distillation. An analysis of the predictor's per-frame entropy on held-out speech reveals a bimodal distribution with a substantial minority of frames near the entropy of a perfect two-cluster tie, providing direct empirical evidence that the soft-target objective preserves the acoustic ambiguity that hard targets would collapse. Code is available at https://github.com/gioannides/s-jepa.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。