ddlBoJack/Awesome-Speech-Pretraining

Paper, Code and Statistics for Self-Supervised Learning and Pre-Training on Speech.

214

47 commits

updated Jan 18, 2024

See the code

README

Table of Contents generated with DocToc

Awesome-Speech-Pretraining

Papers, Resources, and Statistics for Self-Supervised Learning and Pre-Training on Speech.

🌟 represents important papers.

Papers

2018

2019

2020

2021

2022

2023

Speech + Text

SSL for Audio

SSL for TTS

SSL Model Distillation, Compression and Acceleration

Resources

Speech processing Universal PERformance Benchmark (SUPERB)

Self-Supervised Speech Pre-training and Representation Learning (S3PRL)

Statistics

Statistics on speech pretraining.

wav2vec 2.0

Pre-training

SizeTransformerSamplesBatch SizeTrain Time
BASE12 blocks, model dimension 768, FFN 3072, 8 heads1.4m(cropped)/GPU1.6h400k updates, 64 V100 * 1.6d
LARGE24 blocks, model dimension 1024, FFN 4096, 16 heads1.2m(cropped)/GPU2.7h250k updates, 128 V100 * 2.3d(Librispeech)
600k updates, 128 V100 * 5.2d(LibriVox)

Fine-tuning

wav2vec-u

MethodFeature ExtractorBatch SizeTrain Time
wav2vec-Uwav2vec 2.0 LARGE160 unlabeled audio + 160 text samples150k steps, single V100 * 12h
wav2vec-U + self trainingwav2vec 2.0 LARGE/80k updates, 8 V100(Librispeech)
13k updates, 4V100(TIMIT)

HuBERT

Pre-training

SizeFeature ExtractorBatch SizeStageTrain Time
BASEwav2vec 2.0 BASE(95M)87.5s1: MFCC 250k steps
2: 6-th transformer layer 400k steps
9.5h/100k steps, 32GPUs(Librispeech-960)
LARGEwav2vec 2.0 LARGE(317M)56.25s3: 9-th transformer layer from BASE HuBERT 400k steps9.5h/100k steps, 128GPUs(Libri-light-60k)
X-LARGEConformer XXL(964M)22.5s3: 9-th transformer layer from BASE HuBERT 400k steps9.5h/100k steps, 256GPUs(Libri-light-60k)

Fine-tuning

Contributors

ddlBoJack

45 commits

RookieJunChen

2 commits

ddlBoJack/Awesome-Speech-Pretraining

Paper, Code and Statistics for Self-Supervised Learning and Pre-Training on Speech.

214

47 commits

updated Jan 18, 2024

See the code

README

Table of Contents generated with DocToc

Awesome-Speech-Pretraining

Papers, Resources, and Statistics for Self-Supervised Learning and Pre-Training on Speech.

🌟 represents important papers.

Papers

2018

2019

2020

2021

2022

2023

Speech + Text

SSL for Audio

SSL for TTS

SSL Model Distillation, Compression and Acceleration

Resources

Speech processing Universal PERformance Benchmark (SUPERB)

Self-Supervised Speech Pre-training and Representation Learning (S3PRL)

Statistics

Statistics on speech pretraining.

wav2vec 2.0

Pre-training

SizeTransformerSamplesBatch SizeTrain Time
BASE12 blocks, model dimension 768, FFN 3072, 8 heads1.4m(cropped)/GPU1.6h400k updates, 64 V100 * 1.6d
LARGE24 blocks, model dimension 1024, FFN 4096, 16 heads1.2m(cropped)/GPU2.7h250k updates, 128 V100 * 2.3d(Librispeech)
600k updates, 128 V100 * 5.2d(LibriVox)

Fine-tuning

wav2vec-u

MethodFeature ExtractorBatch SizeTrain Time
wav2vec-Uwav2vec 2.0 LARGE160 unlabeled audio + 160 text samples150k steps, single V100 * 12h
wav2vec-U + self trainingwav2vec 2.0 LARGE/80k updates, 8 V100(Librispeech)
13k updates, 4V100(TIMIT)

HuBERT

Pre-training

SizeFeature ExtractorBatch SizeStageTrain Time
BASEwav2vec 2.0 BASE(95M)87.5s1: MFCC 250k steps
2: 6-th transformer layer 400k steps
9.5h/100k steps, 32GPUs(Librispeech-960)
LARGEwav2vec 2.0 LARGE(317M)56.25s3: 9-th transformer layer from BASE HuBERT 400k steps9.5h/100k steps, 128GPUs(Libri-light-60k)
X-LARGEConformer XXL(964M)22.5s3: 9-th transformer layer from BASE HuBERT 400k steps9.5h/100k steps, 256GPUs(Libri-light-60k)

Fine-tuning

Contributors

ddlBoJack

45 commits

RookieJunChen

2 commits