PASE: Leveraging the Phonological Prior of WavLM for Low-Hallucination Generative Speech Enhancement
Abstract
Generative models have shown remarkable performance in speech enhancement (SE), achieving superior perceptual quality over traditional discriminative approaches. However, existing generative SE approaches often overlook the risk of hallucination under severe noise, leading to incorrect spoken content or inconsistent speaker characteristics, which we term linguistic and acoustic hallucinations, respectively. We argue that linguistic hallucination stems from models' failure to constrain valid phonological structures and it is a more fundamental challenge. While language models (LMs) are well-suited for capturing the underlying speech structure through modeling the distribution of discrete tokens, existing approaches are limited in learning from noise-corrupted representations, which can lead to contaminated priors and hallucinations. To overcome these limitations, we propose the Phonologically Anchored Speech Enhancer (PASE), a generative SE framework that leverages the robust phonological prior embedded in the pre-trained WavLM model to mitigate hallucinations. First, we adapt WavLM into a denoising expert via representation distillation to clean its final-layer features. Guided by the model's intrinsic phonological prior, this process enables robust denoising while minimizing linguistic hallucinations. To further reduce acoustic hallucinations, we train the vocoder with a dual-stream representation: the high-level phonetic representation provides clean linguistic content, while a low-level acoustic representation retains speaker identity and prosody. Experimental results demonstrate that PASE not only surpasses state-of-the-art discriminative models in perceptual quality, but also significantly outperforms prior generative models with substantially lower linguistic and acoustic hallucinations.
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 the Phonologically Anchored Speech Enhancer (PASE), which effectively mitigates linguistic and acoustic hallucinations in generative speech enhancement through innovative use of phonological priors and dual-stream representations. This work represents a meaningful advancement in the field of audio processing, addressing critical challenges in speech enhancement with a novel and effective approach.
Methodology
The proposed PASE framework innovatively leverages the phonological prior from the pre-trained WavLM model, employing a denoising representation distillation strategy that effectively reduces linguistic hallucinations. The dual-stream representation approach for vocoder training is a significant methodological advancement, allowing for the preservation of both linguistic content and speaker characteristics. The paper presents a clear and systematic approach to addressing the challenges of hallucination in generative speech enhancement, which is a notable contribution to the field.
Results & evaluation
The experimental design is robust, utilizing a comprehensive dataset and a variety of evaluation metrics that capture both perceptual quality and the integrity of linguistic content. The results demonstrate PASE's superiority over existing models in reducing hallucinations while maintaining high perceptual quality. The extensive ablation studies provide valuable insights into the effectiveness of the proposed methods, reinforcing the claims made by the authors.
Reproducibility
The paper provides detailed implementation details, including configurations for the DeWavLM and vocoder components, as well as the training process. However, the lack of specific information about the primary institution and the absence of a clear citation for the datasets used may hinder full reproducibility. The provided URLs for the demo and project repository enhance accessibility for further exploration.
Limitations
While the paper presents a strong methodology and promising results, it does not fully address potential limitations in terms of computational complexity and the scalability of the proposed approach in real-world applications. Additionally, the reliance on a specific pre-trained model (WavLM) may limit the generalizability of the findings to other contexts or datasets.
Broader impact
The implications of this work are significant for applications in speech enhancement, particularly in environments with high levels of noise. By reducing hallucinations, PASE could improve the reliability of speech recognition systems and enhance user experiences in various domains, including telecommunications, assistive technologies, and entertainment. The main contribution of this paper is the introduction of the Phonologically Anchored Speech Enhancer (PASE), which effectively mitigates linguistic and acoustic hallucinations in generative speech enhancement through innovative use of phonological priors and dual-stream representations. This work represents a meaningful advancement in the field of audio processing, addressing critical challenges in speech enhancement with a novel and effective approach.
Author of this paper? Send a correction.