B IDIRECTIONAL V ARIATIONAL I NFERENCE FOR N ON -A UTOREGRESSIVE T EXT - TO -S PEECH
Anonymous authors
Paper under double-blind review
A BSTRACT
Although early text-to-speech (TTS) models such as Tacotron 2 have succeeded in generating human-like speech, their autoregressive architectures have several limitations: (1) They require a lot of time to generate a mel-spectrogram consisting of hundreds of steps. The autoregressive speech generation shows a lack of robustness due to its error propagation property.
In this paper, we propose a novel non-autoregressive TTS model called BVAE-TTS, which eliminates the architectural limitations and generates a mel-spectrogram in parallel.
BVAETTS adopts a bidirectional-inference variational autoencoder (BVAE) that learns hierarchical latent representations using both bottom-up and top-down paths to increase its expressiveness.
To apply BVAE to TTS, we design our model to utilize text information via an attention mechanism.
By using attention maps that BVAETTS generates, we train a duration predictor so that the model uses the predicted duration of each phoneme at inference.
In experiments conducted on LJSpeech dataset, we show that our model generates a mel-spectrogram 27 times faster than Tacotron 2 with similar speech quality.
Furthermore, our BVAE-TTS outperforms Glow-TTS, which is one of the state-of-the-art non-autoregressive TTS models, in terms of both speech quality and inference speed while having 58% fewer parameters.
I NTRODUCTION
End-to-end text-to-speech (TTS) systems have recently attracted much attention, as neural TTS models such as Tacotron 2 began to generate high-quality speech that is very similar to the human voice (Sotelo et al., 2017;
Wang et al., 2017; Shen et al., 2018; Ping et al., 2018; Li et al., 2019).
Typically, those TTS systems ﬁrst generate a mel-spectrogram from a text using a sequence-to-sequence (seq2seq) model (Sutskever et al., 2014) and then synthesize speech from the mel-spectrogram using a neural vocoder like WaveGlow (Prenger et al., 2019).
Early neural TTS systems have used an autoregressive (AR) architecture to generate a melspectrogram mainly because of its two beneﬁts.
First, the AR generation eases the difﬁculty of modeling mel-spectrogram distribution by factorizing the distribution into the product of homogeneous conditional factors in sequential order.
Second, the seq2seq based AR architecture helps the model predict the length of the target mel-spectrogram from an input text, which is a non-trivial task because there are no pre-deﬁned rules between the lengths of text and mel-spectrogram.
Although they facilitate high-quality speech synthesis, AR TTS models have several shortcomings. First, they cannot generate a mel-spectrogram in parallel, so the inference time increases linearly with mel-spectrogram time steps.
Second, the AR-based generation suffers from accumulated prediction error, resulting in being vulnerable to the out-of-domain data, e.g. very long input text, or text patterns not existing in the training dataset.
In this work, we present a novel non-AR TTS model called BVAE-TTS that achieves fast and robust high-quality speech synthesis.
BVAE-TTS generates a mel-spectrogram in parallel by adopting a bidirectional-inference variational autoencoder (BVAE) (Sønderby et al., 2016; Kingma et al., 2016; Maaløe et al., 2019; Vahdat & Kautz, 2020) consisting of 1-D convolutional networks.
For the highquality speech synthesis, BVAE-TTS learns mel-spectrogram distribution jointly with hierarchical latent variables in a bidirectional manner, which is the property of BVAE that uses both bottom-up and top-down paths.
Furthermore, to match the length of the target mel-spectrogram at inference,
BVAE-TTS has an additional module called duration predictor, which predicts how many steps of a mel-spectrogram will be generated from each phoneme. To train the duration predictor, we employ an attention mechanism in BVAE-TTS to make BVAE-TTS utilize the text while learning attention maps between the text and the mel-spectrogram, where the mapping information is used for duration labels. Our BVAE-TTS has advantages over the previous non-AR TTS models as follows:
• It has a simpler training process compared to the previous non-AR TTS models such as ParaNet (Peng et al., 2020) and FastSpeech (Ren et al., 2019). In the previous TTS models, well-trained AR teacher models are needed for duration labels or knowledge-distillation. Although FastSpeech 2 (Ren et al., 2020) removes the dependency on the teacher model, it still requires additional duration labels and acoustic features that are prepared in advance using other speech analysis methods.
In contrast, BVAE-TTS requires only the text-speech paired dataset without any helps from the teacher model. • It is more ﬂexible in designing its architecture compared to the previous ﬂow-based non-AR TTS models such as Flow-TTS (Miao et al., 2020) and Glow-TTS (Kim et al., 2020). The ﬂow-based models have architectural constraints caused by their bijective transformation property, which leads to deeper models with a lot of parameters.
On the contrary, the VAE-based model is free from the architectural constraints. In experiments conducted on LJSpeech dataset, we ﬁrst compare our BVAE-TTS with Tacotron 2 and Glow-TTS in terms of speech quality, inference speed, and model size.
The results show that our model achieves 27 times speed improvement on average over Tacotron 2 in generating a mel-spectrogram from a text with a similar level of speech quality.
Furthermore, BVAE-TTS outperforms the state-of-the-art ﬂow-based non-AR TTS model, Glow-TTS, in both speech quality and inference time, while having 58% fewer model parameters.
Additionally, we analyze how the latent representations are learned by BVAE-TTS.
In this analysis, we conﬁrm that the bottom part of
BVAE-TTS captures the variation of mel-spectrograms that can occur from a text.
The source code and the audio samples of our model will be available online.
Related work: In the meantime, several TTS systems have utilized VAE to relax the one-to-many mapping nature in TTS, so improve the naturalness and the controllability of the systems.
For example, (Hsu et al., 2018) and (Zhang et al., 2019) incorporate VAE to Tacotron 2 to learn the style or prosody of the input speech.
However, previous uses of VAE have been limited to an auxiliary network in TTS based on the main AR TTS model.
To the best of our knowledge, our BVAE-TTS is the ﬁrst parallel TTS model that directly uses the VAE architecture to the task of TTS. More discussions about other related works on the previous non-AR TTS models are in Section 5.
B ACKGROUND 2. B
[Figure]
Variational autoencoder (VAE) is a neural network generative model p θ ( x , z ) parameterized by θ , where x is an observed data and z is a latent vector.
In practice, since we only have a dataset X = { x 1 , ..., x N } without the knowledge about z , θ is typically optimized by maximizing the likelihood:
[Equation]
However, the integral over z is intractable to compute.
Therefore, the VAE introduces an approximate posterior q φ ( z | x ) and does variational inference while maximizing the evidence lower bound (ELBO):
[Equation]
In practice, for easy sampling and easy computation of the KL-divergence, each of the prior p ( z ) and the approximate posterior q φ ( z | x ) is usually modeled as a multivariate normal distribution with a diagonal covariance matrix.
For a more expressive model, the latent vector z can be factorized into { z 1 , ..., z K } with hierarchical dependency, where K is the number of hierarchies.
Then, each of the prior and the approximate posterior is represented as p θ ( z ) =
Π k p θ ( z k | z <k ) and q φ ( z | x ) =
Π k q φ ( z k | z <k , x ) , respectively.
In (Sønderby et al., 2016; Kingma et al., 2016; Vahdat & Kautz, 2020), the variational inference is designed in a bidirectional way based on bottom-up path and top-down path, while letting the inference network (left) and generative network (right) share their parameters as shown in Figure 1.
First, along the bottom-up path, BVAE extracts hierarchical features from x and stores them inside of it.
Then, along the top-down path, BVAE does the variational inference and reconstructs the input data considering the stored hierarchical features together.
This architecture helps the model effectively learn the hierarchies between the latent variables, and equation (2) is changed as follows:
[Equation]
2. D URATION PREDICTOR IN NON - AUTOREGRESSIVE TEXT - TO - SPEECH
To achieve the non-autoregressive (non-AR) text-to-speech (TTS) model, the model needs to predict the length of the target mel-spectrogram from a text. This is because there is no way to access to the length of the target mel-spectrogram at inference.
However, this is a challenging task considering that there are no pre-deﬁned rules between the lengths of text and mel-spectrogram.
Recently, several non-AR TTS models (Ren et al., 2019; Zeng et al., 2020; Kim et al., 2020) resolved the issue by introducing a module called duration predictor.
The duration predictor is a module that predicts how many mel-spectrogram steps will be generated from each phoneme.
First, using the duration predictor, the non-AR TTS models compute durations ˆ D =
{ ˆ d 1 , ..., ˆ d S } corresponding to each phoneme based on phoneme representations H = { h 1 , ..., h S } , where each ˆ d i is a positive integer that is rounded off from a positive real number, and S is the number of phonemes.
Then, H is expanded to the length of the target mel-spectrogram T , by repeating each h i as many steps as ˆ d i .
Finally, the non-AR TTS models generate a mel-spectrogram in parallel by decoding the expanded phoneme representations.
In practice, since there are no ground-truth duration labels for the training of the duration predictor, the non-AR models obtain the duration labels using various methods, and we adopt a method used in FastSpeech (Ren et al., 2019). From well-aligned attention maps, the duration labels are obtained according to d i = (cid:80) t
= T t =1 [argmax s a s,t
== i ] , where a s,t represents an attention weight given from the t -th mel-spectrogram step to the s -th phoneme.  M ETHODOLOGY
In this section, we explain a novel non-autoregressive (non-AR) TTS model, BVAE-TTS, which is based on the bidirectional-inference variational autoencoder (BVAE).
As shown in Figure 2-(a), during training, BVAE-TTS is given a mel-spectrogram with a phoneme sequence, and it is trained to reconstruct the mel-spectrogram while maximizing the ELBO.
Here, the duration predictor is jointly trained using the attention maps that BVAE-TTS generates during training.
As shown in
Figure 2-(c), at inference BVAE-TTS generates a mel-spectrogram from a phoneme sequence using the duration predictor as described in Section 2.2, while using its top-down path for decoding the expanded phoneme representations. In Appendix A.1, pseudo-codes for the training and inference of BVAE-TTS are contained for detailed descriptions. The other aspects of BVAE-TTS are described in the following sub-sections in more detail.
3. U SING BVAE FOR TEXT - TO - SPEECH
Unlike the previous BVAE models (Sønderby et al., 2016; Kingma et al., 2016; Vahdat & Kautz, 2020) are trained to generate natural images, our model should learn to generate a mel-spectrogram
BVAE Layer
B X attention
Pre-net
Projection
Text Encoder
Phonemes
BVAE Block
BVAE Layer BVAE Layer
Downsample
Upsample
K, V
Q dim ↑ dim ↓ 𝜵𝓛 𝒇(∙) 𝒆𝒙𝒑 V
V Duration Predictor 𝑴𝑺𝑬 log෡𝒅 log𝒅 (a) Training pre-conv conv conv conv post-conv conv conv conv conv z { 𝝁 𝒌 ,𝚺 𝒌 }, { 𝚫𝝁 𝒌 𝟐 ,𝚫𝚺 𝒌 𝟐 } { 𝚫𝝁 𝒌 𝟏 ,𝚫𝚺 𝒌 𝟏 } (b) BVAE Layer
B X
Duration Predictor
Projection
Text Encoder
Phonemes
Upsample
V dim ↓
BVAE Layer BVAE Layer
BVAE Layer 𝒆𝒙𝒑 V (c) Inference that is not only natural but also corresponding to the input text.
To this end, we add a dot-product attention network (Bahdanau et al., 2015) on top of the BVAE, which is a channel for BVAE-TTS to learn how to utilize the text properly.
First, using a text encoder, key ( K ) and value ( V ) are obtained from a phoneme sequence, and from the bottom-up path, query ( Q ) is obtained.
Here, obtaining Q is different from the bottom-up paths of the previous BVAE studies used in the image domain, where only the parameters for posterior approximation are obtained. Second, based on the dot-product attention with Q , K , and V , the V are expanded to V exp to ﬁt the length of the top-down path, and then the V exp is inputted into the top-down path of BVAE-TTS.
Lastly, the BVAE-TTS does both the variational inference and mel-spectrogram reconstruction along the top-down path using the expanded text representations with the following objectives:
[Equation]
where x represents mel-spectrogram, y represents text, z represents latent representation, and mean absolute error (MAE) loss is used for the L recon .
In addition to that, a duration predictor is jointly trained to predict durations corresponding to each phoneme in the logarithmic domain using mean square error (MSE) loss, L dur =
E [(log d i − log ˆ d i ) 2 ] , where the target d i and ˆ d i are obtained as described in Section 2.2.
The duration predictor takes as input the V obtained from the text encoder, and here the V is detached from the computational graph to prevent it from affecting the BVAE training.
3. BVAE-TTS ARCHITECTURE
In this section, we describe the architecture of BVAE-TTS that hierarchically learns the latent representations based on BVAE blocks consisting of BVAE layers.
BVAE block: As shown in Figure 2-(a), the main part of BVAE-TTS is the stacked BVAE blocks, each consisting of BVAE layers.
To guide the multi-scale ﬁne-to-coarse latent features to be contained in the latent hierarchies, the time dimension of input mel-spectrogram is downsampled using bilinear downsampling operations (Zhang, 2019) in even-numbered BVAE blocks along the bottomup path.
Here, the numbering of BVAE blocks starts with one and increases from bottom to top.
On the contrary, its time dimension is upsampled again along the top-down path, by repeating the signals in the BVAE blocks where the downsamplings have occurred.
In odd-numbered BVAE-blocks, the channel dimension is decreased along the bottom-up path and is increased along the top-down path.
It is done at the pre- or post-convolutional network of the ﬁrst BVAE layer in the BVAE blocks shown in Figure 2-(b).
BVAE layer:
The main element of the BVAE block is the BVAE layer.
As shown in Figure 2-(b), at each bottom-up and top-down path, the parameters of the prior and the approximate posterior distributions { µ k , Σ k } , { ∆ µ k 1 , ∆ Σ k 1 } , { ∆ µ k 2 , ∆ Σ k 2 } are obtained from 1-D convolutional networks.
Then, the prior distribution p θ ( z k | z <k , y ) , and the approximate posterior distribution q φ ( z k | z <k , x , y ) are deﬁned as follow:
[Equation]
where diagonal covariance matrices Σ are used after applying a softplus function to guarantee that they are positive.
This parameterization follows (Vahdat & Kautz, 2020), where the approximate posterior q φ ( z k | z <k , x , y ) is relative to the prior p θ ( z k | z <k , y ) .
With this parameterization, when the prior moves, the approximate posterior moves accordingly while making the BVAE training easier and more stable.
During training, the latent representation z k is sampled from q φ ( z k | z <k , x , y ) , and sampled from p θ ( z k | z <k , y ) at inference.
Other details on the BVAE-TTS architecture such as text encoder or duration predictor are in Appendix A.2.
3. B RIDGE THE GAP BETWEEN TRAINING AND INFERENCE
When BVAE-TTS reconstructs a mel-spectrogram during training, text representations are expanded via the attention network. In contrast, text representations are expanded via the duration predictor at inference.
Therefore, to bridge the gap between the attention-based mel-spectrogram generation and the duration-based mel-spectrogram generation, we use the following techniques in this work.
Straight-Through argmax:
In the duration-based generation, the predicted duration of each phoneme is used after it is rounded to the nearest integer.
It means that there is a corresponding phoneme for every time step of a mel-spectrogram.
Therefore, during training, we use a trick called Straight-Through (ST) argmax, where the text representations are passed to the top-down path via the arg max operation instead of the weighted sum in the attention mechanism.
However, during backpropagation, the parameter update is conducted as if the signal obtained from the arg max was a result of the weighted sum.
Jitter:
To make the model more robust to the errors of the duration predictor, we apply jitter to the text representations, where the text representations obtained from the arg max operation are replaced with neighboring text representations with each probability of 25% during training.
We also experimentally observe that applying jitter makes the learning of the attention maps more stable, so the attention maps are not defused throughout the training and stay diagonal.
Positional encoding biasing & Guided attention:
In order to reduce the gap between the attentionbased generation and the duration-based generation, it is important for the learned attention maps to have diagonal shapes.
Therefore, we use two additional techniques to directly help BVAE-TTS learn the diagonal attention maps.
First, we add positional encoding vectors with different angular speeds to query and key as an inductive bias following (Ping et al., 2018).
Second, we use an additional guided attention loss L guide that gives penalties for attention weights deviating from the diagonal following (Tachibana et al., 2018).
For more details on the techniques in this section, see Appendix A.3.
With the above techniques, BVAE-TTS is trained with the following objective:
[Equation]
where the α is a warm-up constant that linearly increases from 0 to 1 over the ﬁrst 20% of training.
This technique is proposed in (Sønderby et al., 2016) to weaken the variational regularization in the early stages of training.
E XPERIMENTS
In this section, we describe the experimental setup and the results obtained from the quantitative and qualitative experiments that are conducted to evaluate our BVAE-TTS.
For comparison, we use two state-of-the-art TTS models: Tacotron 2 1 for an AR TTS model and Glow-TTS 2 for a non-AR TTS model.
Here, we use the pre-trained weights of the models that are publicly available.
4. E XPERIMENTAL SETUP
In the experiments, we mainly use the LJSpeech dataset (Ito & Johnson, 2017) consisting of 12500 / 100 / 500 samples for training / validation / test splits, respectively.
For speech data, we convert raw waveforms into log-mel-spectrograms with 1024 window length and 256 hop length and use them as target sequences of our BVAE-TTS model.
For text data, we convert raw texts into phoneme sequences using grapheme-to-phoneme library (Park, 2019) and use them as input sequences of
BVAE-TTS.
We train the BVAE-TTS consisting of 4 BVAE blocks for 300K iterations with a batch size of 128.
For an optimizer, we use the Adamax Optimizer (Kingma & Ba, 2015) with β 1 = 0 .
9 , β 2 = {+
+}
.using the learning rate scheduling used in (Vaswani et al., 2017), where initial learning rate of 1e3 and warm-up step of 4000 are used.
Training of BVAE-TTS takes about 48 hours on Intel(R) Xeon(R)
Gold 5120 CPU (2.2GHz) and NVIDIA V100 GPU on the Pytorch 1.16.0 library with
Python 3.6.10 over the Ubuntu 16.04 LTS.
For more details on the hyperparameters, see Appendix A.4.
4. E XPERIMENTAL RESULTS
In this section, we compare BVAE-TTS with Tacotron 2 and Glow-TTS in terms of speech quality, inference time, and model size.
For the quality evaluation, we use pre-trained WaveGlow 3 as a vocoder that converts mel-spectrograms to waveforms.
When we sample latent representations in
Glow-TTS and BVAE-TTS, we use the temperature of 0.333 for the models for better speech quality.
(Kingma & Dhariwal, 2018)
[Table]
Speech quality: In this experiment, we measure the Mean-Opinion-Score (MOS) for audio samples generated by each TTS model using ﬁfty sentences randomly sampled from the in-domain LJSpeech test set (MOS-ID).
In addition, we measure another MOS on ﬁfty sentences randomly sampled from the test-clean set of LibriTTS (Zen et al., 2019) to see the generalization ability on the out-of-domain text data (MOS-OOD).
Via Amazon Mechanical Turk (AMT), we assign ﬁve testers living in the
United States to each audio sample, and ask them to listen to the audio sample and give a score between 1 to 5 in 9-scale based on its naturalness.
The MOS results shown in Table 1 demonstrate the superiority of our BVAE-TTS, where it outperforms the state-of-the-art non-AR TTS model, Glow-TTS, on both MOS-ID and MOS-OOD.
Although BVAE-TTS does not surpass Tacotron 2 in MOS-ID, our model achieves better results on MOS-OOD. It shows its robustness to the out-of-domain text over the autoregressive TTS model which suffers from the accumulated prediction error.
For a better understanding of the speech quality generated by the models, we strongly encourage the readers to listen to the audio samples in the supplementary material.
Inference time: We measure inference times taken to generate a mel-spectrogram from a text on the 500 sentences of LJSpeech test set in GPU environment.
The average inference time of each TTS model is shown in Table 1.
As can be seen in the table, our BVAE-TTS shows 27.2 times faster inference speed on average compared with Tacotron 2, and it is also 1.78 times faster than GlowTTS.
Besides, due to the sequential generation property of the AR TTS model, the gap between the inference speed of BVAE-TTS and Tacotron 2 increases with a longer length of an input text.
See Appendix B for more details.
Model size: As can be seen in the last column of Table 1, BVAE-TTS has the smallest number of parameters of 16.0M while maintaining high-quality speech synthesis.
Furthermore, BVAE-TTS gets smaller (to 12.0M) at inference because the layers belonging to the bottom-up path are not used to generate a mel-spectrogram, and the number of parameters is 58% fewer parameters compared to Glow-TTS.
This shows that the training principle of BVAE-TTS that hierarchically learns the latent features while adjusting hidden dimensions enables our model to have small parameters.
It is contrary to the ﬂow-based TTS models such as Flow-TTS (Miao et al., 2020) and Glow-TTS (Kim et al., 2020), where many parameters are used due to its architectural constraint.
4. M ODEL ANALYSIS
As our BVAE-TTS is the ﬁrst VAE-based parallel TTS model, we conduct several analyses on it.
First, we analyze BVAE-TTS to see how the hierarchies are contained in the latent representations and how the variance in mel-spectrograms is learned.
Then, we verify the effectiveness of the techniques used in BVAE-TTS such as Straight-Through (ST) argmax and jitter through ablation studies.
4.3. A NALYSIS ON HIERARCHY
[Figure]
In this experiment, we conduct an analysis on the hierarchical latent representation learning of BVAE-TTS.
To see how the latent features of the mel-spectrograms are learned in the hierarchies, we observe the variations of the mel-spectrograms sampled from the same text, while using different temperatures for different BVAE blocks.
Speciﬁcally, we set a target BVAE block among the four BVAE blocks, and increase the variance of the target BVAE block by using a temperature of 1.0 or 2.0 or 5.0.
On the contrary, we lower the variance of the non-target BVAE blocks using a temperature of 0.333.
Then, we sample 100 different mel-spectrograms each from the same text, while varying the target BVAE block and its temperature.
Figure 3 shows the averages of pixel-by-pixel standard deviations measured on the randomly sampled 100 mel-spectrograms from the same text.
The block numbers in the ﬁgure are given from one to four starting from the BVAE block at the bottom.
In this experiment, we observe that the variance of the speech is mostly contained in the latent representations of BVAE blocks 1, 2, which are close to the mel-spectrogram.
However, there is not much variance in the generated mel-spectrograms when we increase the temperature of BVAE blocks 3, 4, which are close to the text representations.
Therefore, we can conclude that the global content is mostly contained in the expanded text representations obtained using the text encoder and the duration predictor, and the BVAE blocks 3,focus on building the content rather than its style.
Note that while Figure 3 shows standard deviations measured using one examplar sentence, “One, Two, Three.”, the tendency is consistent regardless of the input sentence.
Mel-spectrogram samples obtained in this experiment are in Appendix C.
4.3. A BLATION STUDY
[Figure]
We conduct ablation studies to see the effects of applying jitter and Straight-Through (ST) argmax to the soft attention mechanism in BVAE-TTS, and the results are shown in Figure 4.
Here, since jitter is included in ST-argmax (jitter is applied to the output of the arg max ), the ablation study of not using ST-argmax represents when BVAE-TTS is trained using a normal soft attention mechanism.
The most noticeable differences appear in the attention maps that they learn.
As shown in the ﬁrst row of Figure 4-(a), (b), applying jitter shows the effectiveness for BVAE-TTS to learn well-aligned attention maps.
It results in using more accurate duration labels to train the duration predictor, which leads to more natural speech.
We observe that BVAE-TTS without applying jitter still generates a clear speech even though it is a little unnatural, where it obtains a 3.68 MOS on the LJSpeech dataset.
As shown in the bottom mel-spectrogram of Figure 4-(c), the BVAE-TTS without STargmax technique just generates a stuttering sound.
As shown in Figure 4-(a), although the BVAE-TTS also does not learn the perfect attention map,
BVAE-TTS successfully generates a mel-spectrogram at inference.
Since the text is forced to be used monotonically in the duration-based generation, it makes the model more robust to the attention errors while making fewer pronouncing mistakes.
In addition, when using the duration predictor, it is also possible to locally control the speed of speech by adjusting the predicted durations.
The experiment on the speed control is included in Appendix D.
D ISCUSSION
To overcome the limitations that the autoregressive (AR) TTS models have, various non-AR architectures have been recently proposed.
On one hand, there have been feed-forward neural networks such as ParaNet (Peng et al., 2020), and FastSpeech 1, 2 (Ren et al., 2019; 2020), which use knowledge distillation or additional duration labels and acoustic features.
Although they succeeded in enabling their models to predict the lengths of the target mel-spectrograms, the feed-forward architectures did not ﬁt the one-to-many mapping problems in TTS.
Therefore, FastSpeech (Ren et al., 2019) uses as targets mel-spectrograms that the AR teacher model generates.
This is because much of the diversity in original mel-spectrograms has been eliminated in the mel-spectrograms generated by the AR teacher model.
Besides, FastSpeech 2 (Ren et al., 2020) even directly uses additional pre-obtained acoustic features such as pitch and energy to relax the one-to-many mapping nature in
TTS.
In contrast to the models, it can be seen that BVAE-TTS is asked to solve one-to-one mapping problems during training because there is only one possible target for the reconstruction task.
As a result, BVAE-TTS can generate natural and diverse samples while learning latent features in mel-spectrograms.
On the other hand, there have been generative ﬂow-based non-AR TTS models such as Flow-TTS (Miao et al., 2020) and Glow-TTS (Kim et al., 2020).
While their speech quality is comparable to that of AR TTS models, ﬂow-based generative models usually have a problem that they require a lot of model parameters.
In the models, the dimensions of the hidden representations in the ﬂow networks should be the same through the whole network, and their bipartite ﬂow requires many layers and larger hidden size because of its lack of expressiveness (Ping et al., 2019; Lee et al., 2020).
Contrary to ﬂow-based TTS models, our BVAE-TTS is free from the above issue.
In this work, by designing BVAE-TTS in hierarchical architecture with varying hidden dimensions, we can outperform the ﬂow-based TTS model, Glow-TTS in both speech quality and speed, while having a much smaller model size.
C ONCLUSION
In this work, we propose BVAE-TTS, which is the ﬁrst VAE-based non-AR TTS model that generates a mel-spectrogram from a text in parallel.
To use the BVAE architecture in text-to-speech, we combine BVAE with an attention mechanism to utilize a text, and to extract duration labels for the training of the duration predictor.
In experiments, BVAE-TTS achieves to generate speech 27x faster than Tacotron 2 with similar speech quality, and also outperforms Glow-TTS in terms of both speech quality and inference time with 58% fewer parameters.
Since our VAE-based TTS model shows competitive performance and has many advantages over the previous non-AR TTS models, we hope it becomes a good starting point for future VAE-based TTS research.
A D ETAILS ON BVAE-TTS
A. A LGORITHMS Data:
Result: ˆ x : a reconstructed mel-spectrogram
L total , L recon , L KL , L dur , L guide : total loss, and losses that make up the total loss
{+K V ← TextEncoder ( y ) ; h ← PreNet ( x ) for b ← 0 to B − 1 do if b %2 = = 
 then h ← Downsample( h ) ; endfor l ← 0 to L − 1 do h, ∆ µ+}
, ∆ Σ
← BVAE-TTS b l ] .
BottomUp( h ) ;
BVAE-
∆ µ
← ∆ µ
BVAE-TTS b, l
∆ Σ
← Softplus (∆ Σ
; end end
Q ← IncreaseDimension( h ) ; h exp , A ← Attention( Q + PE query , K + PE key , V ) ; h exp ← DecreaseDimension( h exp ) ;
L KL ←for b ← B − 1 to 0 do for l ← L − 1 to 0 do h exp , kl loss ← BVAE
TTS b l
TopDown( h
L
← L KL + kl loss end if b %2 =
= 1 then
Upsample
; end end
ˆ x ← Projection ( h exp ) ; D ← ExtractDurations( A ) ; ˆ D ← DurationPredictor( V ) ; L recon ← MAE ( x , ˆ x ) ; L dur ← MSE( log (D), log ( ˆ D ) ) ; L guide ← GuidedAttentionLoss( A ) ; L total ← L recon + α · L KL + L dur + L guide ;
Data
Result: h
← DecreaseDimension exp
; for b ← B do for l
L − 1 to do h
if
; end end
; A. A RCHITECTURE DETAILS Text encoder: To obtain key ( K ) and value ( V ) used in the attention mechanism, we use a text encoder used in DeepVoice 3 (Ping et al., 2018), which consists of a series of 1-D convolutional network with a gated linear unit and a residual connection; y = √ 0 .
5( x +conv 1 ( x ) ∗ σ (conv 2 ( x ))) .
The last hidden representations of the text encoder are used as K , and V is computed according to V = √ 0 .
5( K + E ) , where E represents phoneme embedding vectors.
Duration predictor: We use a duration predictor used in FastSpeech (Ren et al., 2019), which consists of two Conv1d-ELU-LayerNorm-Dropout blocks, followed by a linear projection layer.
Then, we take exponential of the output values and add ones to them to guarantee that the predicted durations are larger than one.
Pre-net & Projection layer & Dimension matching layers: To adjust dimensions properly, there are four additional layers; Pre-net, Projection layer, and two additional dimension matching layers.
Pre-net consists of two Conv1d-Dropout-ELU blocks, and Projection layer consists of two Conv1dELU-LayerNorm-Dropout blocks followed by a Conv1d layer with sigmoid activation.
For the dimension matching layers, we use a Conv1d layer with kernel width 5 to increase the dimension, and use a linear layer to decrease the dimension.
BVAE block & layer:
To increase the receptive ﬁeld of the model, the pre- & post- convolutional networks use dilated convolution with the dilation factors 1, 2, 4 for each BVAE layer in each BVAE block.
Every convolutional network except for the networks that output the parameters of the prior and posterior distributions are followed by an ELU layer.
When the residual path is merged with the original path, we combine the signals according to y = √ 0 .
5( x + f ( x )) to stabilize training.
Following (Vahdat & Kautz, 2020), we apply the spectral normalization (Miyato et al., 2018) to the convolutional networks located on the residual path to stabilize training.
A. A DDITIONAL TECHNIQUES
Straight-Through argmax: Let A = { a s,t } be the attention matrix obtained from the dot-product operation in the attention mechanism.
Then, the original attention mechanism expands the value V to V exp according to V exp =
A (cid:124) · V .
However, with the ST-argmax, another one-hot attention matrix A onehot = { a (cid:48) s,t } is ﬁrstly obtained, where a (cid:48) s,t = 1 if a s,t == max s ( a s,t ) , else 0.
Then V exp is obtained as follow:
[Equation]
where by using the detach function, we let the gradient ﬂow to A instead of A onehot during backpropagation.
Jitter: To apply jitter to the hidden states, we apply jitter to A onehot obtained from ST-argmax, of which example is shown in Figure 5.
By using a function ‘ random.choices ’ that returns an element chosen from a list according to probabilities p , applying jitter to A onehot can be implemented according to a (cid:48) s,t = random.choices ([ a (cid:48) s,t − 1 , a (cid:48) s,t , a (cid:48) s,t +1 ] , p = [0 . 25 , 0 . 5 , 0 . 25]) .
[Figure]
Positional encoding biasing: We add positional encoding vectors with different angular speed w s to query Q and key K to introduce inductive bias of diagonal attention: [Equation]
[Equation]
(11) where pos is the position, i is the dimension, and d is the dimension of Q and K .
For Q , w s = 1 , and for K , w s = T/S , where T and S are the numbers of time steps of Q and K .
Then, the attention matrix A is obtained according to A
=
( K + PE key ) · ( Q + PE query ) (cid:124) .
Guided attention: To directly lead the attention map to be formed in diagonal, we additionally introduce L guide as follow:
[Equation]
where g is set to 0.2.
[Figure]
Hyperparameter BVAE-TTS
Phoneme Embedding Dimension
Text Encoder Layers
Text Encoder Hidden Dimension
Text Encoder Conv1D Kernel Width
Text Encoder Conv1D Filter Size Text Encoder Dropout 0. Pre-net Layers Pre-net Dropout 0.
Pre-net Hidden Dimension Downsampling Conv 1D kernel [0.25, 0.5, 0.25] Projection Layers Projection Dropout 0.
Projection Conv1D Kernel Width
Projection Conv1D Filter Size Duration Predictor Conv1D Kernel Width Duration Predictor Conv1D Filter Size Duration Predictor Dropout 0.
BVAE Blocks
BVAE Layers per block BVAE Conv1D Kernel Width Hidden Dimensions of BVAE blocks 128, 128, 64, Total Number of Parameters 16.0M ( 12.0M )
B P ARALLEL SYNTHESIS
[Figure]
To see the beneﬁt of parallel mel-spectrogram synthesis, we measure the inference time of Tacotron 2, Glow-TTS, and BVAE-TTS on the 500 sentences of LJSpeech test set in GPU environment.
Figure 6 shows that the inference times of the non-AR TTS models are almost constant even if the length of the input text gets longer.
On the contrary, the inference time of Tacotron2 linearly increases.
C A NALYSIS ON HIERARCHY
While changing the target BVAE block and its temperature as described in 4.3.1, we observe the generated mel-spectrogram samples.
As shown in Figure 7-9, the variance of the mel-spectrograms is clear when we increase the temperature of the two bottom BVAE blocks.
On the contrary, the melspectrograms are almost the same when we increase the temperature of the top two BVAE blocks.
Especially, when we set the temperature of ‘BVAE block 2’ to 5.0, the mel-spectrograms are the most diverse with good speech quality.
blockblockblockblockblockblockblockblockblockblockblockblock
[Figure]
[Figure]
blockblockblockblockblockblockblockblockblockblock
[Figure]
blockblockblockblockblockblockblockblockblockblockblockblock
[Figure]
[Figure]
D
S PEED CONTROL
[Figure]
While AR TTS models generally suffer from the lack of controllability, BVAE-TTS can control the ﬁne-grained speed of speech by multiplying a positive constant to predicted durations by the duration predictor.
Figure 10 shows three mel-spectrograms produced by BVAE-TTS using the same sentence, “One, Two, Three.”.
While changing the target word from “One” to “Three”, we multiply 2.0 to the durations of phonemes belonging to the target word, and multiply 0.7 to the durations of phonemes belonging to the non-target words.
In this experiment, we observe that BVAE-TTS successfully generates speech while varying the pronouncing speed of each word in a single sentence.
Interestingly, we observe that our model intensiﬁes the pronunciation of the target word, showing the capability of appropriately adjusting the prosody according to the speed.