BrainWavLM: Fine-tuning Speech Representations with Brain Responses to Language

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Vattikonda, Nishitha, Vaidya, Aditya R., Antonello, Richard J., Huth, Alexander G.
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866929713221468160
author Vattikonda, Nishitha
Vaidya, Aditya R.
Antonello, Richard J.
Huth, Alexander G.
author_facet Vattikonda, Nishitha
Vaidya, Aditya R.
Antonello, Richard J.
Huth, Alexander G.
contents Speech encoding models use auditory representations to predict how the human brain responds to spoken language stimuli. Most performant encoding models linearly map the hidden states of artificial neural networks to brain data, but this linear restriction may limit their effectiveness. In this work, we use low-rank adaptation (LoRA) to fine-tune a WavLM-based encoding model end-to-end on a brain encoding objective, producing a model we name BrainWavLM. We show that fine-tuning across all of cortex improves average encoding performance with greater stability than without LoRA. This improvement comes at the expense of low-level regions like auditory cortex (AC), but selectively fine-tuning on these areas improves performance in AC, while largely retaining gains made in the rest of cortex. Fine-tuned models generalized across subjects, indicating that they learned robust brain-like representations of the speech stimuli. Finally, by training linear probes, we showed that the brain data strengthened semantic representations in the speech model without any explicit annotations. Our results demonstrate that brain fine-tuning produces best-in-class speech encoding models, and that non-linear methods have the potential to bridge the gap between artificial and biological representations of semantics.
format Preprint
id arxiv_https___arxiv_org_abs_2502_08866
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle BrainWavLM: Fine-tuning Speech Representations with Brain Responses to Language
Vattikonda, Nishitha
Vaidya, Aditya R.
Antonello, Richard J.
Huth, Alexander G.
Computation and Language
Speech encoding models use auditory representations to predict how the human brain responds to spoken language stimuli. Most performant encoding models linearly map the hidden states of artificial neural networks to brain data, but this linear restriction may limit their effectiveness. In this work, we use low-rank adaptation (LoRA) to fine-tune a WavLM-based encoding model end-to-end on a brain encoding objective, producing a model we name BrainWavLM. We show that fine-tuning across all of cortex improves average encoding performance with greater stability than without LoRA. This improvement comes at the expense of low-level regions like auditory cortex (AC), but selectively fine-tuning on these areas improves performance in AC, while largely retaining gains made in the rest of cortex. Fine-tuned models generalized across subjects, indicating that they learned robust brain-like representations of the speech stimuli. Finally, by training linear probes, we showed that the brain data strengthened semantic representations in the speech model without any explicit annotations. Our results demonstrate that brain fine-tuning produces best-in-class speech encoding models, and that non-linear methods have the potential to bridge the gap between artificial and biological representations of semantics.
title BrainWavLM: Fine-tuning Speech Representations with Brain Responses to Language
topic Computation and Language
url https://arxiv.org/abs/2502.08866