Bone-Conduction Diffusion (BCDM-DC-L)
- The paper introduces a multimodal conditional diffusion framework that leverages bone-conducted inputs with air-conducted signals for robust speech enhancement under extreme noise.
- It details a mathematically grounded approach using forward and reverse stochastic differential equations, with an Euler–Maruyama scheme for efficient inference.
- Empirical results demonstrate superior performance over baselines, showing consistent improvements in PESQ, POLQA, and ESTOI across varying noise levels.
The Bone-conduction Diffusion model, large variant (BCDM-DC-L), is a multimodal speech enhancement architecture that integrates air- and bone-conducted microphone inputs via a conditional diffusion generative framework. Designed to address severe performance degradation in single-channel speech enhancement systems under extreme noise, BCDM-DC-L explicitly leverages the complementary, noise-robust characteristics of bone-conducted speech to guide the denoising process in the spectral domain through a mathematically grounded stochastic differential equation (SDE) mechanism (Khanagha et al., 18 Jan 2026).
1. Mathematical Foundation
BCDM-DC-L models the clean speech short-time Fourier transform (STFT) slice , the noisy air-conducted STFT , and the bone-conducted STFT . The generative process is defined as a continuous-time SDE—specifically, a drift-diffusion process:
where the drift term
pulls the process toward the noisy air-conducted mixture, and the diffusion scheduling is
with , . is standard Wiener noise.
The corresponding reverse-time SDE is
where the "score" 0 is approximated by a neural network 1. Discretization over 2 steps yields the forward transition:
3
and the reverse (denoising) process is modeled as
4
where 5 is constructed via an Euler–Maruyama step utilizing the learned score 6.
Training minimizes the (conditional) Fisher divergence between true and model score:
7
In practice, the equivalent "noise-prediction" parameterization is adopted:
8
No additional variational bounds are used.
2. Model Architecture
The BCDM-DC-L architecture is based on a multi-resolution NCSN++ U-Net, comprising symmetric encoder-decoder paths. The "large" (L) variant adopts channel widths 9 and ResNet blocks of depth 2, with approximately 67.4 million parameters.
Bone-conduction conditioning (DC) is realized via a separate BigGAN-style ResNet downsampling encoder for 0 featuring depth 2 and matching channel sizes. Decoder-stage integration concatenates: (i) main decoder skip-connection features, (ii) the previous decoder output, and (iii) the bone-conduction encoder's features; a 1 convolution reduces channel dimensionality post-concatenation.
Data flow outline:
- 2 noisy mixture spectrogram
- 3 forward SDE state sampled from 4
- For 5:
- 6
- 7
- 8
- Output 9 (complex STFT of enhanced speech)
3. Training Methodology
The model is trained with the loss
0
where 1 with 2 provided directly from SDE analysis.
Optimization employs Adam (learning rate 3; 4), batch size 8, with exponential moving average parameter tracking (decay 0.999) and 5 diffusion steps per sample.
4. Dataset and Preprocessing
Data is sourced from ABCS: 42 hours, 100 speakers, with aligned air- and bone-conducted speech. Noise is synthetically added to air signals using CHiME3, training SNR sampled uniformly in 6 dB and testing on SNRs 7 dB. STFT parameters are: window size 510, hop 128, yielding 256 frequency bins and 256 time frames post-processing, resulting in 8 complex-valued inputs. Each STFT channel is per-utterance zero-mean, unit-variance normalized.
5. Inference and Sampling Procedure
At inference, a Predictor–Corrector sampler is used, cycling through:
- Predictor: Euler–Maruyama step using 9.
- Corrector: Langevin step; 0.
Typically, 1 steps are used, but step counts 2–3 are supported. Step size 4 is adapted to maintain sampler SNR. The model achieves real-time feasibility (e.g., 256×256 input, 5 in 150 ms with a single GPU); real-time/causal deployment is projected possible with 8–16 steps (630 ms/frame). The small model variant achieves competitive results (PESQ) with as few as 20 steps, corresponding to roughly 40 score evaluations.
6. Empirical Performance and Model Analysis
BCDM-DC-L demonstrates consistent superiority over prior single- and multimodal baselines (BiNet, DCCRN) in multiple metrics:
| SNR (dB) | POLQA | PESQ | ESTOI |
|---|---|---|---|
| –10 | 2.44 | 2.02 | 0.76 |
| +15 | 3.70 | 3.25 | 0.94 |
Relative improvements over the strongest baseline are 7 to 8 in POLQA, 9 to 0 in PESQ, and 1 to 2 in ESTOI across all SNRs evaluated.
Ablation studies contrast "DC" (bone-conduction conditioning) versus "IC" (independent conditioning): DC (67.4 M parameters) outperforms IC (65.6 M) by 3 to 4 PESQ but exhibits slower performance scaling as the number of reverse steps increases. Smaller BCDM-DC-S variants surpass DCCRN at 5 steps and BiNet at 6.
7. Significance and Context
BCDM-DC-L demonstrates the effectiveness of integrating bone-conducted signals through conditional diffusion modeling, leveraging the intrinsic noise-resilience of bone transmission and the generative flexibility of score-based SDEs. These results substantiate the potential of multimodal SDE-guided diffusion models to robustly enhance speech in adverse acoustic regimes, setting a new reference for quantitative performance in the multimodal speech enhancement domain (Khanagha et al., 18 Jan 2026). The explicit SDE-based fusion of modalities, and efficient inference via predictor–corrector samplers, positions this approach for further adoption both in batch and real-time applications where robust speech quality under noise is requisite.