Convexity-based Pruning of Speech Representation Models

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Dorszewski, Teresa, Tětková, Lenka, Hansen, Lars Kai
Formato: Preprint
Publicado: 2024
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866908145383636992
author Dorszewski, Teresa
Tětková, Lenka
Hansen, Lars Kai
author_facet Dorszewski, Teresa
Tětková, Lenka
Hansen, Lars Kai
contents Speech representation models based on the transformer architecture and trained by self-supervised learning have shown great promise for solving tasks such as speech and speaker recognition, keyword spotting, emotion detection, and more. Typically, it is found that larger models lead to better performance. However, the significant computational effort involved in such large transformer systems is a challenge for embedded and real-world applications. Recent work has shown that there is significant redundancy in the transformer models for NLP and massive layer pruning is feasible (Sajjad et al., 2023). Here, we investigate layer pruning in audio models. We base the pruning decision on a convexity criterion. Convexity of classification regions has recently been proposed as an indicator of subsequent fine-tuning performance in a range of application domains, including NLP and audio. In empirical investigations, we find a massive reduction in the computational effort with no loss of performance or even improvements in certain cases.
format Preprint
id arxiv_https___arxiv_org_abs_2408_11858
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Convexity-based Pruning of Speech Representation Models
Dorszewski, Teresa
Tětková, Lenka
Hansen, Lars Kai
Computation and Language
Sound
Audio and Speech Processing
Speech representation models based on the transformer architecture and trained by self-supervised learning have shown great promise for solving tasks such as speech and speaker recognition, keyword spotting, emotion detection, and more. Typically, it is found that larger models lead to better performance. However, the significant computational effort involved in such large transformer systems is a challenge for embedded and real-world applications. Recent work has shown that there is significant redundancy in the transformer models for NLP and massive layer pruning is feasible (Sajjad et al., 2023). Here, we investigate layer pruning in audio models. We base the pruning decision on a convexity criterion. Convexity of classification regions has recently been proposed as an indicator of subsequent fine-tuning performance in a range of application domains, including NLP and audio. In empirical investigations, we find a massive reduction in the computational effort with no loss of performance or even improvements in certain cases.
title Convexity-based Pruning of Speech Representation Models
topic Computation and Language
Sound
Audio and Speech Processing
url https://arxiv.org/abs/2408.11858