A Policy-based Approach to the SpecAugment
Method for Low Resource E2E ASR
Rui Li∗§, Guodong Ma∗§, Dexin Zhao†§, Ranran Zeng†, Xiaoyu Li†and Hao Huang∗‡
∗School of Information Science and Engineering, Xinjiang University, Urumqi, China
†China Telecom Beijing Research Institute, Beijing, China
‡Xinjiang Provincial Key Laboratory of Multi-lingual Information Technology, Urumqi, China
Corresponding author; E-mail: hwanghao@gmail.com
§Equal contributions
Abstract—SpecAugment is a very effective data augmentation
method for both HMM and E2E-based automatic speech recog-
nition (ASR) systems. Especially, it also works in low-resource
scenarios. However, SpecAugment masks the spectrum of time
or the frequency domain in a fixed augmentation policy, which
may bring relatively less data diversity to the low-resource ASR.
In this paper, we propose a policy-based SpecAugment (Policy-
SpecAugment) method to alleviate the above problem. The idea
is to use the augmentation-select policy and the augmentation-
parameter changing policy to solve the fixed way. These policies
are learned based on the loss of validation set, which is applied
to the corresponding augmentation policies. It aims to encourage
the model to learn more diverse data, which the model relatively
requires. In experiments, we evaluate the effectiveness of our
approach in low-resource scenarios, i.e., the 100 hours librispeech
task. According to the results and analysis, we can see that the
above issue can be obviously alleviated using our proposal. In
addition, the experimental results show that, compared with the
state-of-the-art SpecAugment, the proposed Policy-SpecAugment
has a relative WER reduction of more than 10% on the Test/Dev-
clean set, more than 5% on the Test/Dev-other set, and an absolute
WER reduction of more than 1% on all test sets.
I. INTRODUCTION
Recently, end-to-end (E2E) automatic speech recognition
(ASR) [1]–[6] based neural networks have achieved a large
improvements. Meanwhile, the E2E ASR simplifies the pro-
cessing of system construction, which establishes a direct
mapping from the acoustic feature sequences to the modeling
unit sequences. With the emergence of E2E ASR, researchers
[7]–[22] explore different E2E ASR scenarios and partly focus
on the data augmentation and training strategy due to the
nature of data-hungry and easy over-fitting. However, most of
the existing work on data augmentation technology is just to
explore a fixed way to bring more abundant data to the model.
For example, the state-of-the-art SpecAugment method [7] uses
three spectrum disturbance strategies in a fixed augmentation
policy for input speech spectrum features. Under a certain
amount of training data, the fixed augmentation policy may
tend to be stable faster so that it will not bring too much
data diversity in a model learning stage. Therefore, we believe
that, with the use of SpecAugment, the information brought to
the model can be enriched. In addition, when the model is in
different learning states, it may more need to learn the data
applied to different the combinations of augmentation policy.
But the masking strategies of SpecAugment are completely
random which is not related to model state.
Based on the above discussions, we propose a policy-
based SpecAugment to improve the performance of low re-
source end-to-end (E2E) ASR systems, which is named Policy-
SpecAugment. In our proposed Policy-SpecAugment method,
we will calculate the loss value of the validation set under the
action of each augmentation strategies in the model learning
stage. The loss value can be represented the fitting degree
of the model to the corresponding augmentation strategies
at the previous epoch trained model, so as to reflect which
augmentation strategy should be studied at the current model
training epoch. Then, the losses will be used to calculate the
probabilities of the augmentation-select policy and the factor
of the augmentation-parameter changing policy, which aims to
encourage the augmentation method to produce the various data
needed by ASR model. The specific details will be introduced
in section IV. For fair comparison, the augmentation strategies
we used are consistent with SpecAugment, including time
masking, frequency masking and time warping. We will briefly
show the three classic and effective augmentation strategies
in Section III. In the 100 hours librispeech task [23], we
use ESPnet1 [24] to confirm our Policy-SpecAugment. The
experimental results show that, compared with the state-of-
the-art SpecAugment, our proposed Policy-SpecAugment has
a relative increase of more than 10% on the test/dev-clean set,
a relative increase of more than 5% on the test/dev-other set,
and an absolute increase of more than 1% on all test sets.
The paper is organized as follows. Section II is to review
the prior related works. In Section III, we will briefly show the
three data augmentation methods in SpecAugment. And Section
IV presents the proposed policy-based SpecAugment method in
detail. In addition, Section V describes the experiment setups
and results. After that, we perform further analysis in Section
VI and conclude in Section VII.
II. RELATED WORKS
Data augmentation is a method of increasing the diversity of
training data [25]–[27] to prevent the model over-fitting. Cur-
rently, there are many data augmentation methods to improve
ASR system performance. In [28], audio speed perturbation
was proposed, which aims to augment the varied speed data
for the model. In [29], room impulse responses were proposed
to simulate far-field data. In [30], speech synthesis methods
arXiv:2210.08520v1 [cs.SD] 16 Oct 2022