UCL Discovery Stage
UCL home » Library Services » Electronic resources » UCL Discovery Stage

Trinet: Stabilizing Self-Supervised Learning From Complete or Slow Collapse

Cao, L; Wang, J; Yang, B; Su, D; Yu, D; (2023) Trinet: Stabilizing Self-Supervised Learning From Complete or Slow Collapse. In: Proceedings of ICASSP 2023 - 2023 IEEE International Conference on Acoustics, Speech and Signal Processing (ICASSP). IEEE: Rhodes, Greece. Green open access

[thumbnail of 2301.00656.pdf]
Preview
Text
2301.00656.pdf - Accepted Version

Download (675kB) | Preview

Abstract

Self-supervised learning (SSL) models confront challenges of abrupt informational collapse or slow dimensional collapse. We propose TriNet, which introduces a novel triple-branch architecture for preventing collapse and stabilizing the pretraining. TriNet learns the SSL latent embedding space and incorporates it to a higher level space for predicting pseudo target vectors generated by a frozen teacher. Our experimental results show that the proposed method notably stabilizes and accelerates pre-training and achieves a relative word error rate reduction (WERR) of 6.06% compared to the state-of- the-art (SOTA) Data2vec for a downstream benchmark ASR task. We will release our code at https://github.com/tencent-ailab/.

Type: Proceedings paper
Title: Trinet: Stabilizing Self-Supervised Learning From Complete or Slow Collapse
Event: ICASSP 2023 - 2023 IEEE International Conference on Acoustics, Speech and Signal Processing (ICASSP)
Dates: 4 Jun 2023 - 10 Jun 2023
ISBN-13: 9781728163277
Open access status: An open access version is available from UCL Discovery
DOI: 10.1109/ICASSP49357.2023.10094725
Publisher version: https://doi.org/10.1109/ICASSP49357.2023.10094725
Language: English
Additional information: This version is the author accepted manuscript. For information on re-use, please refer to the publisher’s terms and conditions.
Keywords: Self-supervised learning, collapse, pseudo label, self-learning, bootstrapping
UCL classification: UCL
UCL > Provost and Vice Provost Offices > UCL BEAMS
UCL > Provost and Vice Provost Offices > UCL BEAMS > Faculty of Engineering Science
UCL > Provost and Vice Provost Offices > UCL BEAMS > Faculty of Engineering Science > Dept of Computer Science
URI: https://discovery-pp.ucl.ac.uk/id/eprint/10183765
Downloads since deposit
1,221Downloads
Download activity - last month
Download activity - last 12 months
Downloads by country - last 12 months

Archive Staff Only

View Item View Item