atlas

Research map / Label-efficient and robust learning / Self-supervised and contrastive

Multi-resolution HuBERT: Multi-resolution Speech Self-Supervised Learning with Masked Unit Prediction

ICLR 2024 · Label-efficient and robust learning › Self-supervised and contrastive · approach: Self-supervised speech representations · 1 citations (OpenAlex)

Read the paperCheck your idea against 108,967 papers

AI-extracted summary. Written by AI from the paper's full text. Numbers and claims can be wrong or missing context, so check the paper before you cite or build on it.

Proposes MR-HuBERT, a hierarchical Transformer trained with multi-resolution HuBERT-style masked unit prediction (20ms and 40ms), which improves over HuBERT on LibriSpeech ASR, SUPERB, and ML-SUPERB while cutting inference MACs by 9-13%.

Problem

Speech SSL models process speech at a fixed 20ms frame rate, ignoring that different resolutions carry different information; prior multi-resolution work needed separate pre-trained models, and no work integrated multi-resolution information into pre-training.

Approach

A single hierarchical Transformer encodes speech at high (20ms) and low (40ms) resolutions with up/down-sampling modules between encoders, trained with summed masked unit prediction losses at both resolutions, where low-resolution targets are obtained by skip-subsampling high-resolution K-means units.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
replace single-resolution HuBERT with multi-resolution HuBERTHuBERT-large · LibriSpeech 1-hour labeled, test-cleanWER: 14.40 -> 6.37Helped
replace single-resolution HuBERT with multi-resolution HuBERTHuBERT-large · SUPERB Understandingscore: 932.6 -> 949.7Helped
replace single-resolution HuBERT with multi-resolution HuBERTHuBERT-large · SUPERB Enhancementscore: 456.0 -> 609.5Helped
train a multilingual multi-resolution model on VoxpopuliHuBERT-base · ML-SUPERB 10-minute/1-hourSUPERB score: 746.2 / 812.7 -> 957.2 / 986.8Helped
use multi-resolution inferenceHuBERT-base · Inference, base modelMACs: 431G -> 394GHelped
add a third resolution (20, 40, 80 ms) (ablation)mono-base · LibriSpeech 10-hour labeled, test-cleanWER: 8.46 -> 8.84Hurt
vary encoder layer allocation to (5, 2, 5) (ablation)mono-base · LibriSpeech 1-hour labeled, test-cleanWER: 19.26 -> 18.63Helped
simplify upsampling and downsampling modules (ablation)mono-base · LibriSpeech 1-hour labeled, test-cleanWER: 19.26 -> 18.33Helped
use a single prediction target instead of multi-resolution prediction (ablation)mono-base · LibriSpeech 10-hour labeled, test-cleanWER: 8.46 -> 9.52Hurt
use a single resolution (20, 20 ms) (ablation)mono-base · LibriSpeech 100-hour labeled, test-cleanWER: 4.92 -> 4.93Hurt
reduce each encoder from four layers to three (ablation)mono-base · LibriSpeech 100-hour labeled, test-cleanWER: 4.92 -> 5.55Hurt
use HuBERT-base-40 units for low-resolution prediction targets instead of subsampled high-resolution units (ablation)mono-base · LibriSpeech 10-hour labeled, test-cleanWER: 8.46 -> 9.36Hurt

Surprising findings

Limitations

Experimental setup

Datasets: LibriSpeech LibriLight VoxPopuli ML-SUPERB

Benchmarks: LibriSpeech SUPERB ML-SUPERB

Metrics: word error rate phone error rate accuracy F1 CER BLEU STOI PESQ SI-SDRi SUPERB score

Baselines: HuBERT-base HuBERT-base+ HuBERT-large HuBERT-large* mHuBERT-base mHuBERT-base*

Builds on: HuBERT wav2vec 2.0 SUPERB ML-SUPERB

Closest papers by meaning

Writing a paper in this area? scime draft is a free LaTeX editor with AI included, real-time co-authors and a check of your references and venue rules before you submit. It also imports OpenAI Prism projects.

Open scime draft

Get the next scime Dispatch and a monthly field update

Free. We send a confirmation link first; every email has a one-click unsubscribe. Privacy