Mitigating Attention Sinks and Massive Activations in Audio-Visual Speech Recognition with LLMs
By: Anand , Umberto Cappellazzo , Stavros Petridis and more
Potential Business Impact:
Makes computers understand talking better, even with bad sound.
Large language models (LLMs) have recently advanced auditory speech recognition (ASR), visual speech recognition (VSR), and audio-visual speech recognition (AVSR). However, understanding of their internal dynamics under fine-tuning remains limited. In natural language processing, recent work has revealed attention sinks, tokens that attract disproportionately high attention, and associated massive activations in which some features of sink tokens exhibit huge activation in LLMs. In this work, we are the first to study these phenomena in multimodal speech recognition. Through a detailed analysis of audio-visual LLMs, we identify attention sinks and massive activations not only at the BOS token but also at intermediate low-semantic tokens across ASR, VSR, and AVSR. We show that massive activations originate in the MLP layers and correspond to fixed feature indices across all sink tokens. We further show that intermediate sink tokens exhibit high cosine similarity to the BOS token, thereby amplifying attention and activation. Building on these insights, we introduce a simple decorrelation loss that reduces cosine similarity between BOS and other tokens, effectively mitigating intermediate sinks and massive activations. Furthermore, our method improves word error rate (WER) under high audio-visual feature downsampling while remaining stable at lower downsampling rates.
Similar Papers
Mitigating Attention Sinks and Massive Activations in Audio-Visual Speech Recognition with LLMS
Audio and Speech Processing
Makes computers understand talking better, even with bad sound.
See What You Are Told: Visual Attention Sink in Large Multimodal Models
CV and Pattern Recognition
Makes AI better at looking at pictures.
Attention Sinks and Compression Valleys in LLMs are Two Sides of the Same Coin
Machine Learning (CS)
Fixes how AI understands and remembers information.