Listen Like a Teacher: Mitigating Whisper Hallucinations using Adaptive Layer Attention and Knowledge Distillation
Abstract
The Whisper model, an open-source automatic speech recognition system, is widely adopted for its strong performance across multilingual and zero-shot settings. However, it frequently suffers from hallucination errors, especially under noisy acoustic conditions. Previous works to reduce hallucinations in Whisper-style ASR systems have primarily focused on audio preprocessing or post-processing of transcriptions to filter out erroneous content. However, modifications to the Whisper model itself remain largely unexplored to mitigate hallucinations directly. To address this challenge, we present a two-stage architecture that first enhances encoder robustness through Adaptive Layer Attention (ALA) and further suppresses hallucinations using a multi-objective knowledge distillation (KD) framework. In the first stage, ALA groups encoder layers into semantically coherent blocks via inter-layer correlation analysis. A learnable multi-head attention module then fuses these block representations, enabling the model to jointly exploit low- and high-level features for more robust encoding. In the second stage, our KD framework trains the student model on noisy audio to align its semantic and attention distributions with a teacher model processing clean inputs. Our experiments on noisy speech benchmarks show notable reductions in hallucinations and word error rates, while preserving performance on clean speech. Together, ALA and KD offer a principled strategy to improve Whisper's reliability under real-world noisy conditions.
The following analysis is generated by AI and may contain errors. It is separate from the authors’ abstract and is not peer review.
Research summary
The main contribution of this paper is the introduction of a two-stage framework that enhances the robustness of the Whisper ASR model through Adaptive Layer Attention and Multi-Objective Knowledge Distillation. This comprehensive analysis highlights the technical contributions, innovative methodology, and significant potential impact on the field of automatic speech recognition.
Methodology
The proposed methodology introduces two significant innovations: Adaptive Layer Attention (ALA) and Multi-Objective Knowledge Distillation (MOKD). ALA enhances the encoder's robustness by dynamically fusing representations from semantically coherent blocks of layers, addressing the common issue of relying solely on the final layer outputs. This approach is well-grounded in the principles of deep learning, leveraging inter-layer correlation analysis to improve feature extraction under noisy conditions. MOKD further refines the model by aligning the student model's representations with those of a clean teacher model, effectively mitigating hallucination errors. The combination of these two techniques represents a thoughtful and systematic approach to improving ASR systems, particularly in challenging environments.
Results & evaluation
The experiments are comprehensive, covering multiple languages and a range of noise conditions, which strengthens the validity of the findings. The use of diverse datasets (e.g., CommonVoice, LibriSpeech) and the introduction of noise from the DEMAND dataset provide a realistic assessment of the model's performance. The results demonstrate significant improvements in word error rates (WER) and semantic accuracy (SeMaScore), particularly under low signal-to-noise ratios (SNR), showcasing the effectiveness of the proposed methods. However, a more detailed comparison with existing state-of-the-art models could enhance the evaluation.
Reproducibility
The paper provides sufficient implementation details, including training configurations, datasets used, and evaluation metrics, which are essential for reproducibility. However, the absence of URLs for code or demo implementations limits the ease with which other researchers can replicate the study. Including a GitHub repository or similar would significantly enhance reproducibility.
Limitations
While the proposed methods show promise, the paper does not address potential limitations, such as the computational overhead introduced by ALA and MOKD, which could impact real-time applications. Additionally, the reliance on a clean teacher model may not always be feasible in practical scenarios where such models are not available.
Broader impact
The work has significant implications for improving the reliability of ASR systems in real-world applications, particularly in noisy environments. By reducing hallucination errors, the proposed methods can enhance user trust in ASR technologies, potentially leading to broader adoption in critical applications such as healthcare, customer service, and accessibility tools. The main contribution of this paper is the introduction of a two-stage framework that enhances the robustness of the Whisper ASR model through Adaptive Layer Attention and Multi-Objective Knowledge Distillation. This comprehensive analysis highlights the technical contributions, innovative methodology, and significant potential impact on the field of automatic speech recognition.
Author of this paper? Send a correction.