The mean squared error (MSE) is a ubiquitous loss function for speech enhancement, but its problem is that the error cannot reflect the auditory perception quality. This is because MSE causes models to over-emphasize low-frequency components which has high energy, leading to the inadequate modeling of perceptually important high-frequency information. To overcome this limitation, we propose a perceptually-weighted loss function grounded in psychoacoustic principles. Specifically, it leverages equal-loudness contours to assign frequency-dependent weights to the reconstruction error, thereby penalizing deviations in a way aligning with human auditory sensitivity. The proposed loss is model-agnostic and flexible, demonstrating strong generality. Experiments on the VoiceBank+DEMAND dataset show that replacing MSE with our loss in a GTCRN model elevates the WB-PESQ score from 2.17 to 2.93-a significant improvement in perceptual quality.
Figures & tables
Figure 1: Overview of the Loud-loss computation pipeline.
Freq
SPL
Freq
SPL
Freq
SPL
Freq
SPL
(Hz)
(dB)
(Hz)
(dB)
(Hz)
(dB)
(Hz)
(dB)
20
99.85
25
93.94
31.5
88.17
40
82.63
50
77.78
63
73.08
80
68.48
100
64.37
125
60.59
160
56.70
200
53.41
250
50.40
315
47.58
400
44.98
500
43.05
630
41.34
800
40.06
1000
40.01
1250
41.82
1600
42.51
Table 1: Required SPL (dB) versus frequency (Hz) for the 40-phon equal-loudness contour.
Exp. ID
Models
Type
Loss
WB-PESQ
NB-PESQ
ESTOI
-
Unprocessed
-
-
1.97
2.88
0.787
1
GTCRN †
Mapping
MSE
2.17
3.04
0.812
2
GTCRN † + PCS
Mapping
MSE
2.48
3.20
0.807
3
GTCRN †
Mapping
Loud-loss
2.93
3.51
0.836
4
GTCRN ‡
Masking
MSE
2.57
3.24
0.824
5
GTCRN ‡
Masking
Loud-loss
2.92
3.50
0.841
Table 2: Objective evaluation results on the VoiceBank+DEMAND test set. Boldface indicates the best score in each column. † denotes magnitude mapping-based methods, while ‡ denotes masking-based methods.
Figure 2: Comparison of enhanced speech spectrograms generated using different loss functions.
Exp. ID
Description
WB-PESQ
3
Baseline (Loud-loss)
2.93
Analysis of Sub-band Splitting
11
w/o Sub-band Overlap
2.81
12
Uniform Band Split
2.40
13
Per-bin Weighting (no bands)
2.56
Analysis of Weighting Scheme
Table 3: Ablation study of the Loud-loss on the GTCRN † model. "Baseline" refers to our full proposed method (Exp. 3).
Loss
DNSMOS
NISQA
P.808
SIG
BAK
OVRL
MOS
Noise
Color
Disc.
Loud.
MSE
3.0477
2.9139
3.0105
2.3416
2.3468
2.0555
2.6022
3.0961
3.0108
PFPL
3.2226
2.9757
3.4343
2.5276
2.4191
2.8315
2.7386
3.1316
3.1564
MetricGAN
3.2266
2.9424
3.4958
2.5285
2.5918
2.9450
2.7597
3.5369
3.0876
Loud-loss
3.2272
2.8972
3.6656
2.5500
2.5189
3.0390
2.7500
3.4316
3.1426
Table 4: Evaluation on the DNS3 blind test set.
Method
Extra net
Params (M)
Time (ms)
Rel. to MSE
Memory (MB)
FLOPs (G)
MSE
None
0.028
18.43±3.14
1.00×
313.6
2.96
PFPL
Frozen wav2vec
32.565
56.65±4.88
3.07×
871.1
296.14
MetricGAN
Discriminator
0.209
162.92±14.04
8.84×
329.8
7.67
Loud-loss
None
0.028
19.38±2.88
1.05×
313.6
2.96
Table 5: Training-cost comparison with other perceptually motivated methods.
Speech enhancement improves intelligibility and listening quality by suppressing noise, reverberation, and competing speakers in recorded or live audio.