Title: HIGH FIDELITY SPEECH SYNTHESIS WITH ADVERSARIAL NETWORKS

Abstract: Generative adversarial networks have seen rapid development in recent years and have led to remarkable improvements in generative modelling of images. However, their application in the audio domain has received limited attention, and autoregressive models, such as WaveNet, remain the state of the art in generative modelling of audio signals such as human speech. To address this paucity, we introduce GAN-TTS, a Generative Adversarial Network for Text-to-Speech. Our architecture is composed of a conditional feed-forward generator producing raw speech audio, and an ensemble of discriminators which operate on random windows of different sizes. The discriminators analyse the audio both in terms of general realism, as well as how well the audio corresponds to the utterance that should be pronounced. To measure the performance of GAN-TTS, we employ both subjective human evaluation (MOS -Mean Opinion Score), as well as novel quantitative metrics (Fréchet DeepSpeech Distance and Kernel DeepSpeech Distance), which we find to be well correlated with MOS. We show that GAN-TTS is capable of generating high-fidelity speech with naturalness comparable to the state-of-the-art models, and unlike autoregressive models, it is highly parallelisable thanks to an efficient feed-forward generator.

1. INTRODUCTION:

The Text-to-Speech (TTS) task consists in the conversion of text into speech audio. In recent years, the TTS field has seen remarkable progress, sparked by the development of neural autoregressive models for raw audio waveforms such as WaveNet (van den Oord et al., 2016) , SampleRNN (Mehri et al., 2017) and WaveRNN (Kalchbrenner et al., 2018) . A notable limitation of these models is that they are difficult to parallelise over time: they predict each time step of an audio signal in sequence, which is computationally expensive and often impractical. A lot of recent research on neural models for TTS has focused on improving parallelism by predicting multiple time steps in parallel, e.g. using flow-based models (van den Oord et al., 2018; Ping et al., 2019; Prenger et al., 2019; Kim et al., 2019) . Such highly parallelisable models are more suitable to run efficiently on modern hardware.

An alternative approach for parallel waveform generation would be to use Generative Adversarial Networks (GANs, Goodfellow et al., 2014) . GANs currently constitute one of the dominant paradigms for generative modelling of images, and they are able to produce high-fidelity samples that are almost indistinguishable from real data. However, their application to audio generation tasks has seen relatively limited success so far. In this paper, we explore raw waveform generation with GANs, and demonstrate that adversarially trained feed-forward generators are indeed able to synthesise high-fidelity speech audio. Our contributions are as follows:

• We introduce GAN-TTS, a Generative Adversarial Network for text-conditional highfidelity speech synthesis. Its feed-forward generator is a convolutional neural network, coupled with an ensemble of multiple discriminators which evaluate the generated (and real) audio based on multi-frequency random windows. Notably, some discriminators take the linguistic conditioning into account (so they can measure how well the generated audio corresponds to the input utterance), while others ignore the conditioning, and can only assess the general realism of the audio. • We propose a family of quantitative metrics for speech generation based on Fréchet Inception Distance (FID, Heusel et al., 2017) and Kernel Inception Distance (KID, Bińkowski et al., 2018) , where we replace the Inception image recognition network with the Deep-Speech audio recognition network. The code for our metrics is publicly available online1 . • We present quantitative and subjective evaluation of TTS-GAN and its ablations, demonstrating the importance of our architectural choices. Our best-performing model achieves a MOS of 4.2, which is comparable to the state-of-the-art WaveNet MOS of 4.4, and establishes GANs as a viable option for efficient TTS.

2. RELATED WORK:

2.1 AUDIO GENERATION Most neural models for audio generation are likelihood-based: they represent an explicit probability distribution and the likelihood of the observed data is maximised under this distribution. Autoregressive models achieve this by factorising the joint distribution into a product of conditional distributions (van den Oord et al., 2016; Mehri et al., 2017; Kalchbrenner et al., 2018; Arik et al., 2017) . Another strategy is to use an invertible feed-forward neural network to model the joint density directly (Prenger et al., 2019; Kim et al., 2019) . Alternatively, an invertible feed-forward model can be trained by distilling an autoregressive model using probability density distillation (van den Oord et al., 2018; Ping et al., 2019) , which enables it to focus on particular modes. Such mode-seeking behaviour is often desirable in conditional generation settings: we want the generated speech signals to sound realistic and correspond to the given text, but we are not interested in modelling every possible variation that occurs in the data. This reduces model capacity requirements, because parts of the data distribution may be ignored. Note that adversarial models exhibit similar behaviour, but without the distillation and invertibility requirements.

Many audio generation models, including all of those discussed so far, operate in the waveform domain: they directly model the amplitude of the waveform as it evolves over time. This is in stark contrast to most audio models designed for discriminative tasks (e.g. audio classification): such models tend to operate on time-frequency representations of audio (spectrograms), which encode certain inductive biases with respect to the human perception of sound, and usually discard all phase information in the signal. While phase information is often inconsequential for discriminative tasks, generated audio signals must have a realistic phase component, because fidelity as judged by humans is severely affected otherwise. Because no special treatment for the phase component of the signal is required when generating directly in the waveform domain, this is usually more practical.

Tacotron (Wang et al., 2017) and MelNet (Vasquez & Lewis, 2019) constitute notable exceptions, and they use the Griffin-Lim algorithm (Griffin & Lim, 1984) to reconstruct missing phase information, which the models themselves do not generate. Models like Deep Voice 2 & 3 (Gibiansky et al., 2017; Ping et al., 2018) and Tacotron 2 (Shen et al., 2018) achieve a compromise by first generating a spectral representation, and then using a separate autoregressive model to turn it into a waveform and fill in any missing spectral information. Because the generated spectrograms are imperfect, the waveform model has the additional task of correcting any mistakes. Char2wav (Sotelo et al., 2017) uses intermediate vocoder features in a similar fashion.

2.2. GENERATIVE ADVERSARIAL NETWORKS:

Generative Adversarial Networks (GANs, Goodfellow et al., 2014) form a subclass of implicit generative models that relies on adversarial training of two networks: the generator, which attempts to produce samples that mimic the reference distribution, and the discriminator, which tries to differentiate between real and generated samples and, in doing so, provides a useful gradient signal to the generator. Following rapid development, GANs have achieved state-of-the-art results in image (Zhang et al., 2019; Brock et al., 2019; Karras et al., 2019) and video generation (Clark et al., 2019) , and have been successfully applied for unsupervised feature learning (Donahue et al., 2017; Dumoulin et al., 2017a; Donahue & Simonyan, 2019) , among many other applications.

Despite achieving impressive results in these domains, limited work has so far shown good performance of GANs in audio generation. Two notable exceptions include WaveGAN (Donahue et al., 2019) and GANSynth (Engel et al., 2019) , which both successfully applied GANs to simple datasets of audio data. The former is the most similar to this work in the sense that it uses GANs to generate raw audio; results were obtained for a dataset of spoken commands of digits from zero to nine. The latter provides state-of-the-art results for a dataset of single note recordings from various musical instruments (NSynth, Engel et al., 2017) GANs have seen limited application at large scale in non-visual domains so far. Two seconds of audio at 24kHz2 has a dimensionality of 48000, which is comparable to RGB images at 128 × 128 resolution. Until recently, high-quality GAN-generated images at such or higher resolution were uncommon (Zhang et al., 2019; Karras et al., 2019) , and it was not clear that training GANs at scale would lead to extensive improvements (Brock et al., 2019) .

Multiple discriminators have been used in GANs for different purposes. For images, Denton et al. (2015) ; Zhang et al. (2017) ; Karras et al. (2018) proposed to use separate discriminators for different resolutions. Similar approaches have also been used in image-to-image transfer (Huang et al., 2018) and video synthesis (Saito & Saito, 2018) . Clark et al. (2019) , on the other hand, combine a 3D-discriminator that scores the video at lower resolution and a 2D-frame discriminator which looks at individual frames. In adversarial feature learning, Donahue & Simonyan (2019) combine outputs from three discriminators to differentiate between joint distributions of images and latents. Discriminators operating on windows of the input have been used in adversarial texture synthesis (Li & Wand, 2016) and image translation (Isola et al., 2017; Zhu et al., 2017) .

In parallel with our work, MelGAN (Kumar et al., 2019) used GANs for mel-spectrogram inversion, with multiple discriminators operating at different frequencies. MelGAN enforces the mapping between input spectrogram conditioning and generated audio via an L 1 reconstruction loss in discriminator feature space, in contrast with our work, which enforces the mapping adversarially using conditional discriminators. When combined with a Text2mel model, their generator achieves a MOS of 3.7.

3. GAN-TTS:

3.1 DATASET Our text-to-speech models are trained on a dataset which contains high-fidelity audio of human speech with the corresponding linguistic features and pitch information. The linguistic features encode phonetic and duration information, while the pitch is represented by the logarithmic fundamental frequency log F 0 . In total, there are 567 features. We do not use ground-truth duration and pitch for subjective evaluation; we instead use duration and pitch predicted by separate models. The dataset is formed of variable-length audio clips containing single sequences, spoken by a professional voice actor in North American English. For training, we sample 2 second windows (filtering out shorter examples) together with corresponding linguistic features. The total length of the filtered dataset is 44 hours. The sampling frequency of the audio is 24kHz, while the linguistic features and pitch are computed for 5ms windows (at 200Hz). This means that the generator network needs to learn how to convert the linguistic features and pitch into raw audio, while upsampling the signal by a factor of 120. We apply a µ-law transform to account for the logarithmic perception of volume (see Appendix C).

3.2. GENERATOR:

A summary of generator G's architecture is presented in Table 2 in Appendix A.2. The input to G is a sequence of linguistic and pitch features at 200Hz, and its output is the raw waveform at 24kHz. The generator is composed of seven blocks (GBlocks, Figure 1a ), each of which is a stack of two residual blocks (He et al., 2016) . As the generator is producing raw audio (e.g. a 2s training clip corresponds to a sequence of 48000 samples), we use dilated convolutions (Yu & Koltun, 2016) to ensure that the receptive field of G is large enough to capture long-term dependencies. The four kernel size-3 convolutions in each GBlock have increasing dilation factors: 1, 2, 4, 8. Convolutions are preceded by Conditional Batch Normalisation (Dumoulin et al., 2017b) , conditioned on the linear embeddings of the noise term z ∼ N (0, I 128 ) in the single-speaker case, or the concatenation of z and a onehot representation of the speaker ID in the multi-speaker case. The embeddings are different for each BatchNorm instance. A GBlock contains two skip connections, the first of which performs upsampling if the output frequency is higher than the input, and it also contains a size-1 convolution if the number of output channels is different from the input. GBlocks 3-7 gradually upsample the temporal dimension of hidden representations by factors of 2, 2, 2, 3, 5, while the number of channels is reduced by GBlocks 3, 6 and 7 (by a factor of 2 each). The final convolutional layer with Tanh activation produces a single-channel audio waveform.

3.3. ENSEMBLE OF RANDOM WINDOW DISCRIMINATORS:

Instead of a single discriminator, we use an ensemble of Random Window Discriminators (RWDs) which operate on randomly sub-sampled fragments of the real or generated samples. The ensemble allows for the evaluation of audio in different complementary ways, and is obtained by taking a Cartesian product of two parameter spaces: (i) the size of the random windows fed into the discriminator; (ii) whether a discriminator is conditioned on linguistic and pitch features. For example, in our best-performing model, we consider five window sizes (240, 480, 960, 1920, 3600 samples) , which yields 10 discriminators in total. Notably, the number of discriminators only affects the training computation requirements, as at inference time only the generator network is used, while the discriminators are discarded. However, thanks to the use of relatively short random windows, the proposed ensemble leads to faster training than conventional discriminators.

Using random windows of different size, rather than the full generated sample, has a data augmentation effect and also reduces the computational complexity of RWDs, as explained next. In the first layer of each discriminator, we reshape (downsample) the input raw waveform to a constant temporal dimension ω = 240 by moving consecutive blocks of samples into the channel dimension, i.e. from

[ωk, 1] to [ω, k],

where k is the downsampling factor (e.g. k = 8 for input window size 1920). This way, all the RWDs have the same architecture and similar computational complexity despite different window sizes. We confirm these design choices experimentally in Section 5.

The conditional discriminators have access to linguistic and pitch features, and can measure whether the generated audio matches the input conditioning. This means that random windows in conditional discriminators need to be aligned with the conditioning frequency to preserve the correspondence between the waveform and linguistic features within the sampled window. This limits the valid sampling to that of the frequency of the conditioning signal (200Hz, or every 5ms). The unconditional discriminators, on the contrary, only evaluate whether the generated audio sounds realistic regardless of the conditioning. The random windows for these discriminators are sampled without constraints at full 24kHz frequency, which further increases the amount of training data.

More formally, let λ = 120 denote the frequency ratio between waveform and linguistic features and let D C k ( , * ; θ) and D U k ( ; θ ) be conditional and unconditional discriminator networks parametrized by θ and θ , respectively, which downsample the waveform input by a factor of k.3 . We define conditional and unconditional RWDs as stochastic functions:

EQUATION

where x and c are respectively the waveform and linguistic features and a l:r = (a l , a l+1 , . . . , a r-1 ) T denotes a vector slice.

The final ensemble discriminator combines 10 different RWD's:

RWD * ω (x, c; θ * ) = k∈{1,2,4,8,15} cRWD k,ω (x, c; θ k ) + uRWD k,ω (x; θ k ), θ * = k (θ k ∪ θ k ).

(3) Algorithm 1 in Appendix D shows pseudocode for computation of RWD * . In Section 5 we describe other combinations of RWDs as well as a full, deterministic discriminator which we used in our ablation study.

3.4. DISCRIMINATOR ARCHITECTURE:

The full discriminator architecture is shown in Figure 2 . The discriminators consists of blocks (DBlocks) that are similar to the GBlocks used in the generator, but without batch normalisation. The architectures of standard and conditional DBlocks are shown in Figures 1b and 1c respectively. The only difference between the two DBlocks is that in the conditional DBlock, the embedding of the linguistic features is added after the first convolution. The first and the last two DBlocks do not downsample (i.e. keep the temporal dimension fixed). Apart from that, we add at least two downsampling blocks in the middle, with downsample factors depending on k, so as to match the frequency of the linguistic features (see Appendix A.2 for details). Unconditional RWDs are composed entirely of DBlocks. In conditional RWDs, the input waveform is gradually downsampled by DBlocks, until the temporal dimension of the activation is equal to that of the conditioning, at which point a conditional DBlock is used. This joint information is then passed to the remaining DBlocks, whose final output is average-pooled to obtain a scalar. The dilation factors in the DBlocks' convolutions follow the pattern 1, 2, 1, 2, . . . -unlike the generator, the discriminator operates on a relatively small window, and we did not observe any benefit from using larger dilation factors.

4. EVALUATION:

We provide subjective human evaluation of our model using Mean Opinion Scores (MOS), as well as quantitative metrics.

4.1. MOS:

We evaluate our model on a set of 1000 sentences, using human evaluators. Each evaluator was asked to mark the subjective naturalness of a sentence on a 1-5 Likert scale, comparing to the scores reported by van den Oord et al. (2018) for WaveNet and Parallel WaveNet.

Although our model was trained to generate 2 second audio clips with the starting point not necessarily aligned with the beginning of a sentence, we are able to generate samples of arbitrary length. This is feasible due to the fully convolutional nature of the generator and carried out using a convolutional masking trick, detailed in Appendix A.1. Human evaluators scored full sentences with a length of up to 15 seconds. 

4.2. SPEECH DISTANCES:

We introduce a family of quantitative metrics for generative models of speech, which include the unconditional and conditional Fréchet DeepSpeech Distance (FDSD, cFDSD) and Kernel Deep-Speech Distance (KDSD, cKDSD). These metrics follow common metrics used in evaluation of GANs for images, Fréchet Inception Distance (FID, Heusel et al., 2017) and Kernel Inception Distance (KID, Bińkowski et al., 2018) .

FID and KID compute the Fréchet distance and the Maximum Mean Discrepancy (MMD, Gretton et al., 2012) respectively between representations of reference and generated distributions extracted from a pre-trained Inception network (Szegedy et al., 2016) . To obtain analogous metrics for speech, we extract the features from an open-source implementation of an accurate speech recognition model, DeepSpeech2 (Amodei et al., 2016) . Specifically, we use the implementation available in the NVIDIA OpenSeq2Seq library (Kuchaiev et al., 2018) and extract features from the last layer, whose output is used in the CTC loss during training. We use representations in the resulting feature space to compute the Fréchet distance and MMD (See Appendix B.1 for details).

We note that Kilgour et al. (2019) proposed a similar metric, Fréchet Audio Distance. This metric uses a network trained for audio event classification on the AudioSet dataset (Gemmeke et al., 2017) as a feature extractor, whereas we use a network that was trained for speech recognition.

As conditioning plays a crucial role in our task, we compute two variants of these metrics, conditional (cFDSD, cKDSD) and unconditional (FDSD, KDSD). Both Fréchet and Kernel distance provide scores with respect to a reference real sample and require both the real sample and the generated one to be independent and identically distributed. Assume that variables x real and x G 3 . Next to each block we present the dimensionality and frequency of its outputs; ch. -number of output channels.

are drawn from the real and and generated distributions, while c is drawn from the distribution of linguistic features. In the conditional case, cFDSD and cKDSD compute distances between conditional distributions p(x G |c) and p(x real |c). In the unconditional case, FDSD and KDSD compare p(x G ) and p(x real ).

Both metrics are estimated using 10,000 generated and reference samples, drawn independently with the same (in the conditional case), or independent (in the unconditional case) linguistic features. This procedure is detailed in Appendix B.3.

The main reason for using both Fréchet and Kernel distances is the popularity of FID in the image domain, despite the issue of its biased estimator, as shown by Bińkowski et al. (2018) . Thanks to the availability of an unbiased estimator of MMD, this issue does not apply to kernel-based distances.

For instance, they yield zero values for real data, which allows comparison in the conditional case. We give more details on these distances in Appendix B.2.

5. EXPERIMENTS:

In this section we discuss the experiments, comparing GAN-TTS with WaveNet and carrying out ablations that validate our architectural choices.

As mentioned in Section 3, the main architectural choices made in our model include the use of multiple RWDs, conditional and unconditional, with a number of different downsampling factors. We thus consider the following ablations of our best-performing model:

1. full-input discriminator FullD = D C 1 , 2. single conditional RWD: cRWD 1 ,

Published as a conference paper at ICLR 2020 3. multiple conditional RWDs: cRWD {1,2,4,8,15} = k∈{1,2,4,8,15} cRWD k , 4. single conditional and single unconditional RWD: cRWD 1 + uRWD 1 , 5. five independent cRWDs and uRWDs:

(cRWD 1 + uRWD 1 ) ×5 (x, c) := 5 i=1 cRWD 1 (x, c; θ i ) + uRWD 1 (x; θ i ), 6. 10 RWDs without downsampling but with different window sizes:

RWD 1,240×{1,2,4,8,15} = k∈{1,2,4,8,15} (cRWD 1,240k + uRWD 1,240k )

7. 10 RWDs with longer window: RWD * 480 .

All other parameters of these models were the same as in the proposed one. In Appendix D we present details of the hyperparameters used during training as well as pseudocode for training GAN-TTS.

5 Our best model achieves worse yet comparable scores to the strong baselines, WaveNet and Parallel WaveNet. This performance, however, has not yet been achieved using adversarial techniques and is still very good, especially when compared to parametric text-to-speech models. These results are however not directly comparable due to dataset differences; for instance WaveNet and Parallel WaveNet were trained on 65 hours of data, a bit more than GAN-TTS.

Our ablation study confirms the importance of combining multiple RWDs. The deterministic full discriminator achieved the worst scores. All multiple-RWD models achieved better results than a single cRWD 1 ; all models that used unconditional RWDs were better than those that did not. Comparing 10-discriminator models, it is clear that combinations of different window sizes were beneficial, as a simple ensemble of 10 fixed-size windows was significantly worse. All three 10-RWD models with varying discriminator sizes achieved similar mean opinion scores, with the downsampling model with base window size 240 performing best.

We also observe a noticeable correlation between human evaluation scores (MOS) and the proposed metrics, which demonstrates that these metrics are well-suited for the evaluation of neural audio synthesis models.

5.2. DISCUSSION:

Random window discriminators. Although it is difficult to say why RWDs work much better than the full discriminator, we conjecture that this is because of the relative simplicity of the dis-tributions that the former must discriminate between, and the number of different samples we can draw from these distributions. For example, the largest window discriminators used in our best model discriminate between distributions supported on R 3600 , and there are respectively 371 and 44,401 different windows that can be sub-sampled from a 2s clip (real or generated) by conditional and unconditional RWDs of effective window size 3600. The full discriminator, on the other hand, always sees full real or generated examples sampled from a distribution supported on R 48000 .

Computational efficiency. Our Generator has a larger receptive field (590ms, i.e. 118 steps at the frequency of the linguistic features) and three times fewer FLOPs (0.64 MFLOP/sample) than Parallel WaveNet (receptive field size: 320ms, 1.97 MFLOP/sample). However, the discriminators used in our ensemble compare windows of shorter sizes, from 10ms to 150ms. Since these windows are much shorter than the entire generated clips, training with ensembles of such RWDs is faster than with FullD. In terms of depth, our generator has 30 layers, which is a half of Parallel WaveNet's, while the depths of the discriminators vary between 11 and 17 layers, as discussed in Appendix A.2.

Stability. The proposed model enjoyed very stable training, with gradual improvement of subjective sample quality and decreasing values of the proposed metrics. Despite training for as many as 1 million steps, we have not experienced model collapses often reported in GAN literature and studied in detail by Brock et al. (2019) .

6. CONCLUSION:

We have introduced GAN-TTS, a GAN for raw audio text-to-speech generation. Unlike state-ofthe-art text-to-speech models, GAN-TTS is adversarially trained and the resulting generator is a feed-forward convolutional network. This allows for very efficient audio generation, which is important in practical applications. Our architectural exploration lead to the development of a model with an ensemble of unconditional and conditional Random Window Discriminators operating at different window sizes, which respectively assess the realism of the generated speech and its correspondence with the input text. We showed in an ablation study that each of these components is instrumental to achieving good performance. We have also proposed a family of quantitative metrics for generative models of speech: (conditional) Fréchet DeepSpeech Distance and (conditional) Kernel DeepSpeech Distance, and demonstrated experimentally that these metrics rank models in line with Mean Opinion Scores obtained through human evaluation. The metrics are publicly available for machine learning community, as is the DeepSpeech recognition model they are based on. Our quantitative results as well as subjective evaluation of the generated samples showcase the feasibility of text-to-speech generation with GANs. A ARCHITECTURE DETAILS A.1 MASKING CONVOLUTIONS TO GENERATE LONGER SAMPLES Since our generator is a fully-convolutional network, in theory it is capable of generating samples of arbitrary length. However, since deep learning frameworks usually require processing fixed-size samples in batches for efficiency reasons, our inputs of different lengths need to be zero-padded to fit in a fixed-size tensor. Convolutional layers, including the ones used in our model, often pad their inputs to create outputs of the desired dimensionality, hence we only need to ensure that the padded part of the input tensors to all layers is always zero. As shown in Figure 3 , this would not normally be the case after the second convolutional layer, since convolutions (with kernel sizes greater than one) would propagate non-zero values outside the border between meaningful input and padding. A simple way to address this issue is masking, i.e. multiplying the input by a zero-one mask tensor, directly before each convolutional layer. This enables batched sampling of utterances of different length, which is efficient on many hardware platforms, optimised for batching. 2 : Architecture of GAN-TTS's Generator. t denotes the temporal dimension, while ch denotes the number of channels. The rightmost three columns describe dimensions of the output of the corresponding layer.

In Table 2 we present the details of Generator architecture. Overall, the generator has 30 layers, most of which are parts of dilated residual blocks. 

B DEEPSPEECH DISTANCES -DETAILS B.1 DEEPSPEECH2:

Our evaluation metrics extract high-level features from raw audio using the pre-trained DeepSpeech2 model from the NVIDIA OpenSeq2Seq library (Kuchaiev et al., 2018) . Let w = 480 be a 20ms window of raw audio at 24kHz, and let f : R w -→ R 1600 be a function that maps such a window through the DeepSpeech2 network up to the 1600-dimensional output of the layer labeled ForwardPass/ds2_encoder/Reshape_2:0. We use default values for all settings of the DeepSpeech2 model; f also includes the model's preprocessing layers.

For a 2s audio clip a ∈ R 100w , we define

EQUATION

where a i:j = (a i , a i+1 , . . . , a j-1 ) is a vector slice.

The function DS therefore computes 1600 features for each 20ms window, sampled evenly with 10ms overlap, and then takes the average of the features along the temporal dimension.

B.2 METRICS IN DISTRIBUTION SPACE:

Given samples X ∈ R m×d and Y ∈ R n×d , where d is the representation dimension, the Fréchet distance and MMD can be computed using the following estimators: x ← x j:j+ωk assign the slice of waveform x ← x j:j+ωk assign the slice of waveform return 19: end procedure orthogonal regularisation (Brock et al., 2016; 2019) and exponential moving averaging to the generator weights with a decay rate of 0.9999 for sampling. We also use cross-replica BatchNorm (Ioffe & Szegedy, 2015) , which aggregates batch statistics from all devices across which the batch is split and standing statistics during sampling. The latter means that we accumulate batch statistics from 100 forward passes through the generator before the actual sampling takes place, allowing for inference at arbitrary batch sizes.

EQUATION

In fact, accumulating standing statistics makes the BatchNorm layers in the generator independent of any characteristics of the samples produced during inference. This technique is thus vital for sampling audio of unspecified length: producing samples that are longer than those used during training typically requires using a smaller batch size, with partially padded samples (See Appendix A.1). These smaller batches would naturally have different statistics than the batches used during training. We trained our models on Cloud TPU v3 Pods with data parallelism over 128 replicas for 1 million generator and discriminator updates, which usually took up to 48 hours.

In Algorithm 1 we present the pseudocode for training GAN-TTS. 

:

https://github.com/mbinkowski/DeepSpeechDistances

24kHz is a commonly used frequency for speech, because the absence of frequencies above 12kHz does not meaningfully affect fidelity.

Here we only require that ∈ R ωk and * ∈ R (ωk/λ)×567 for any window size ω. Since we consider fully-convolutional architectures with average pooling after the top layer, D C k and D U k do not depend on the value of ω. We describe architecture details of D C k 's and D U k 's for k ∈ {1, 2,

4, 8, 15} in Appendix A.2.

Figures and Tables:

Figure 1: Residual blocks used in the model. Convolutional layers have the same number of input and output channels and no dilation unless stated otherwise. h -hidden layer representation, llinguistic features, z -noise vector, m -channel multiplier, m = 2 for downsampling blocks (i.e. if their downsample factor is greater than 1) and m = 1 otherwise, M -G's input channels, M = 2N in blocks 3, 6, 7, and M = N otherwise; size refers to kernel size.

Figure 2: Multiple Random Window Discriminator architecture. The discriminator combines outputs from 5 unconditional (uRWDs, left) and 5 conditional (cRWDs, right) discriminators; one of each group is detailed in the diagram. The number of downsampling blocks is fixed for uRWDs and depends on the input window size ωk for cRWDs, see Table3. Next to each block we present the dimensionality and frequency of its outputs; ch. -number of output channels.

Figure3: Masking scheme for sampling different-length samples. Top: processing a batch of samples of different lengths padded with zeros leads to interference between padding and non-padding after the second convolution, not seen during training. Bottom: masking after each convolution ensures that the meaningful input seen by each layer is padded with zeros only.

where µ X , µ Y and Σ X , Σ Y are the means and covariance matrices of X and Y respectively, while k : R d × R d -→ R is a positive definite kernel function. FollowingBińkowski et al. (2018) we use the polynomial kernel k(x, y) = 1 d x T y + 1 3 . (7) Algorithm 1 TTS-GAN require: N -waveform length, λ -waveform-conditioning frequency ratio, ω -base window size, n steps -number of training steps, n batch -batch size, η D , η G -discriminator and generator learning rates. 1: procedure RWD*(x, c, θ * ) Compute value of RWDs ensemble 2:

procedure 1: procedure TTS-GAN(ψ, θ * ) Main training algorithm 2:for s := 1 to n steps do 1 , c 1 ), . . . ,(x n batch , c n batch )) RWD * (x i , c i , θ * )) + + (1 + RWD * (x i , c i , θ * )) + ) Adam(∇ θ * l D , η D ) c 1 , . . . , c n batch ) iid ∼ p(c) sample batch of iid linguistic features 13: z ← (z 1 , . . . , z n batch ) RWD * (x i , c i , θ * ) compute G's loss 16: ψ ← Adam(∇ ψ l G , η G )

Figure 4: Learning curve for the GAN-TTS model in terms of cFDSD.

Figure 4 presents the stable and gradual decrease of cFDSD during training.

Results from prior work, the ablation study and the proposed model. Mean opinion scores for natural speech, WaveNet and Parallel WaveNet are taken from van den Oord et al. (2018) and are not directly comparable due to dataset differences. For natural speech we present estimated FDSD -non-zero due to the bias of the estimator -and theoretical values of KDSD and cKDSD. cFDSD is unavailable; see Appendix B.2.

Table content: <table><tr><td>.1 RESULTS</td><td/><td/><td/><td/><td/></tr><tr><td>model</td><td/><td>MOS</td><td colspan="2">FDSD cFDSD</td><td>KDSD ×10 5</td><td>cKDSD ×10 5</td></tr><tr><td>natural speech</td><td/><td>4.55 ± 0.075</td><td>0.161</td><td>N/A</td><td>0</td><td>0</td></tr><tr><td colspan="2">WaveNet, van den Oord et al. (2016)</td><td>4.41 ± 0.069</td><td/><td/><td/></tr><tr><td colspan="3">Parallel WaveNet, van den Oord et al. (2018) 4.41 ± 0.078</td><td/><td/><td/></tr><tr><td>FullD</td><td/><td>1.889 ± 0.057</td><td>4.51</td><td>4.46</td><td>785</td><td>782</td></tr><tr><td>cRWD 1</td><td/><td colspan="2">3.394 ± 0.058 0.362</td><td>0.247</td><td>35.2</td><td>30.9</td></tr><tr><td>cRWD {1,2,4,8,15}</td><td/><td colspan="2">3.498 ± 0.059 0.398</td><td>0.284</td><td>42.1</td><td>37.9</td></tr><tr><td>cRWD 1 + uRWD 1</td><td/><td colspan="2">3.502 ± 0.057 0.259</td><td>0.144</td><td>16.6</td><td>12.3</td></tr><tr><td>(cRWD 1 + uRWD 1 )</td><td>×5</td><td colspan="2">3.526 ± 0.054 0.194</td><td>0.073</td><td>5.59</td><td>1.34</td></tr><tr><td colspan="2">RWD 1,240×{1,2,4,8,15} RWD  *  480 GAN-TTS (RWD  *  )</td><td colspan="2">4.154 ± 0.050 0.184 4.195 ± 0.045 0.193 4.213 ± 0.046 0.184</td><td>0.061 0.069 0.060</td><td>3.73 5.28 3.84</td><td>0.54 0.98 0.37</td></tr></table>



Table content: <table><tr><td colspan="2">A.2 GENERATOR AND DISCRIMINATOR DETAILS</td><td/><td/></tr><tr><td>layer/input</td><td>t</td><td>freq.</td><td>ch</td></tr><tr><td>linguistic features, z</td><td>400</td><td colspan="2">200Hz 567</td></tr><tr><td>conv, kernel size 3</td><td>400</td><td colspan="2">200Hz 768</td></tr><tr><td>GBlock</td><td>400</td><td colspan="2">200Hz 768</td></tr><tr><td>GBlock</td><td>400</td><td colspan="2">200Hz 768</td></tr><tr><td>GBlock, upsample ×2</td><td>800</td><td colspan="2">400Hz 384</td></tr><tr><td colspan="2">GBlock, upsample ×2 1600</td><td colspan="2">800Hz 384</td></tr><tr><td colspan="4">GBlock, upsample ×2 3200 1600Hz 384</td></tr><tr><td colspan="4">GBlock, upsample ×3 9600 4800Hz 192</td></tr><tr><td colspan="3">GBlock, upsample ×5 48000 24kHz</td><td>96</td></tr><tr><td>conv, kernel size 3</td><td colspan="2">48000 24kHz</td><td>1</td></tr><tr><td>Tanh</td><td/><td/><td/></tr></table>

shows the numbers of residual DBlocks and downsample factors in these blocks for different initial downsample factors of RWDs.

Table content: <table><tr><td colspan="3">Published as a conference paper at ICLR 2020</td><td/><td/><td/><td/></tr><tr><td>k</td><td/><td>D C k</td><td/><td/><td>D U k</td><td/></tr><tr><td/><td>factors</td><td colspan="2">num. blocks depth</td><td>factors</td><td colspan="2">num. blocks depth</td></tr><tr><td colspan="2">1 1, 5, 3, 2, 2, 2, 1, 1</td><td>8</td><td>17</td><td>1, 5, 3, 1, 1</td><td>5</td><td>11</td></tr><tr><td>2</td><td>1, 5, 3, 2, 2, 1, 1</td><td>7</td><td>15</td><td>1, 5, 3, 1, 1</td><td>5</td><td>11</td></tr><tr><td>4</td><td>1, 5, 3, 2, 1, 1</td><td>6</td><td>13</td><td>1, 5, 3, 1, 1</td><td>5</td><td>11</td></tr><tr><td>8</td><td>1, 5, 3, 1, 1</td><td>5</td><td>11</td><td>1, 5, 3, 1, 1</td><td>5</td><td>11</td></tr><tr><td>15</td><td>1, 2, 2, 2, 1, 1</td><td>6</td><td>13</td><td>1, 2, 2, 1, 1</td><td>5</td><td>11</td></tr></table>

Downsample factors in discriminators for different initial stride values k.All conditional discriminators eventually add the representations of the waveform and the linguistic features. This happens once the temporal dimension of the main residual stack is downsampled to the dimension of the linguistic features, i.e. by a factor of 120. Downsampling is carried out via an initial reshape operation (by a factor k varying per RWD) and then in residual blocks, whose downsample factors are prime divisors of 120/k, in decreasing order. For unconditional discriminators, we use only the first two largest prime divisors of 120/k.Algorithm 1 in Appendix D presents pseudocode for forward pass through RWDensemble.

Table content: <table/>

Peer review (iclr2020):
I want thank the authors for solving this long-standing GAN challenge in raw waveform synthesis. With all due respect, previous GAN trials for audio synthesis are inspiring, but their audio qualities are far away from the state-of-the-art results. Although the speech fidelity of GAN-TTS is still worse than WaveNet and Parallel WaveNet from the posted sample, it has begun to close the significant performance gap that has existed between autoregressive models and GANs for raw audios. Overall, this is a very good paper with significant contributions to the filed. Detailed comment: 1, In WaveNet, the conditional features (linguistic / mel-spectrogram) are added as bias terms in the convolutional layers. Did the authors tried this alternative architecture for the generator, which uses the white noisy z as network input (similar as flow-based models, e.g., Parallel WaveNet) and the conditional features as bias term in the convolutional layers? 2, Could the authors comment the importance of serval architecture choices in this work? From Table 1, it seems to me that the ensemble of random window discriminators is the most important (perhaps the only important) contributing factor for the success. For example, the MOS score was boosted from 1.889 to 4.213 by replacing a single full discriminator to the ensemble of RWDs. 3, The notations in Eq. (1) and (2) are messy. Although I can figure their meaning from the context, one may clarify certain notations if they appear at the first time. 4, The stable training (NO model collapses) is pretty impressive. Could the authors shed some light on the potential reason? Does the ensemble of RWD regularizes the training? What*s your experience for training FullD (does not have random window ) and cRWD_1 (only has one random window discriminator)? Are they still very stable? Also, could the authors comment on the importance of large batch size -- 1024 for stable training of GAN-TTS? 5, Although there is a notable difference, one may properly mention previous work Yamamoto et al. (2019), which uses GAN as an auxiliary loss within ClariNet and obtains high-fidelity speech ( https://r9y9.github.io/demos/projects/interspeech2019/ ). Yamamoto et al. Probability Density Distillation with Generative Adversarial Networks for High-Quality Parallel Waveform Generation. 2019. === update === Thank you for the detailed response. 2, Thanks for the elaboration. 4, It would be very interesting to see an analysis of model stability with smaller batch sizes.