Audio ML PapersLandmark papers
← All papers
RESEARCH PAPER / 2026-02-20

Rethinking Flow and Diffusion Bridge Models for Speech Enhancement

Dahan Wang, Jun Gao, Tong Lei ...Full author list in the original paper.

Nanjing University · 40th AAAI Conference on Artificial Intelligence (AAAI-26)

Affiliations: Nanjing University

Abstract

Flow matching and diffusion bridge models have emerged as leading paradigms in generative speech enhancement, modeling stochastic processes between paired noisy and clean speech signals based on principles such as flow matching, score matching, and Schrödinger bridge. In this paper, we present a framework that unifies existing flow and diffusion bridge models by interpreting them as constructions of Gaussian probability paths with varying means and variances between paired data. Furthermore, we investigate the underlying consistency between the training/inference procedures of these generative models and conventional predictive models. Our analysis reveals that each sampling step of a well-trained flow or diffusion bridge model optimized with a data prediction loss is theoretically analogous to executing predictive speech enhancement. Motivated by this insight, we introduce an enhanced bridge model that integrates an effective probability path design with key elements from predictive paradigms, including improved network architecture, tailored loss functions, and optimized training strategies. Experiments on denoising and dereverberation tasks demonstrate that the proposed method outperforms existing flow and diffusion baselines with fewer parameters and reduced computational complexity. The results also highlight that the inherently predictive nature of this generative framework imposes limitations on its achievable upper-bound performance.

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 unified framework for flow and diffusion bridge models in speech enhancement, which enhances performance through innovative methodologies and insights. This work significantly advances the field by bridging generative and predictive modeling approaches, offering a comprehensive solution to challenges in speech enhancement.

Methodology

The paper presents a unified framework that integrates flow matching and diffusion bridge models for speech enhancement, providing a novel interpretation of these models as Gaussian probability paths. The methodology is robust, combining theoretical insights with practical improvements in network architecture and training strategies. The introduction of a time embedding mechanism and an enhanced loss function demonstrates a thoughtful approach to optimizing performance while reducing complexity.

Results & evaluation

The experiments are well-structured, utilizing two datasets for denoising and dereverberation tasks. The results show a clear performance advantage over existing baselines, with comprehensive metrics that validate the effectiveness of the proposed model. The ablation studies further strengthen the findings by isolating the impact of various modifications.

Reproducibility

The paper includes sufficient implementation details, including dataset descriptions, training configurations, and hyperparameter settings, which enhance reproducibility. The availability of code on GitHub supports this aspect, allowing other researchers to replicate the experiments.

Limitations

While the proposed model shows significant improvements, the authors acknowledge that its inherently predictive nature may impose an upper limit on performance compared to purely predictive models. Additionally, the reliance on specific architectures may limit generalizability to other tasks or domains.

Broader impact

The research has potential applications in various speech processing tasks, including real-time communication systems, hearing aids, and assistive technologies for the hearing impaired. The integration of predictive paradigms into generative models could inspire further innovations in speech enhancement and related fields. The main contribution of this paper is the introduction of a unified framework for flow and diffusion bridge models in speech enhancement, which enhances performance through innovative methodologies and insights. This work significantly advances the field by bridging generative and predictive modeling approaches, offering a comprehensive solution to challenges in speech enhancement.

Author of this paper? Send a correction.