Demo

“Over the past few years, speech synthesis systems have seen rapid advances thanks to deep learning. As anyone who owns a voice assistant knows, artificial voices are becoming more and more natural and convincing. The good news is you can recreate this impressive technology yourself, using high quality open-source tools.”

With video

Agenda

• Intro & why deep learning
• Breaking down the problem (seq2seq & audio synthesis)
• Solution (acoustic model & vocoder)
  • Step-by-step guide to getting started
• Q&A

Aims

• Leave the talk able to train a near state-of-art TTS system, with a voice of your choice, from scratch.
• Understand the problem domain and common architectures for solutions.
• That the paragraph below won’t be gibberish by the end of the session!

a recurrent sequence-to-sequence feature prediction network with attention which predicts a sequence of mel spectrogram frames from an input character sequence, combined with a vocoder which generates time-domain waveform samples conditioned on the predicted mel spectrogram frames. — Tacotron 2 paper

Hearing is Believing: Generating Realistic Speech with Deep Learning

A bit of history

• Humans have been synthesising speech with computers for decades
• Prior to the emergence of DNNs, two approaches dominated:
  • Concatenative synthesis
  • Parametric synthesis
• But over the past ~5 years, deep learning methods have become the SOTA

Why do deep learning methods dominate?

A 13 stage (!) TTS system from Bell Labs

A typical modern TTS pipeline

Why do deep learning methods dominate?

Second, because they sound “better”.

Why do deep learning methods dominate?

Second, because they sound “better”.

How can we measure how good TTS systems sound?

• Ultimately, we have to rely on human judgement
• We want to do that in a structured way
• Industry standard is Mean Opinion Score (MOS)
  • Ask a pool of human reviewers to score the naturalness of the speech on a five point scale (1 = Bad, 2 = Poor, 3 = Fair, 4 = Good, 5 = Excellent)
  • Take the average of these scores

MOS: Deep learning vs. legacy systems

“And here’s another example of speech generated by our deep learning system. Now let’s try and gain a deeper understanding of the relevant problem domains.”

Sequence-to-sequence (seq2seq) problem

Some observations about seq2seq

Some observations about seq2seq

Observation #1: Need more than a simple, start to finish, one-to-one mapping between input tokens & output tokens.

Some observations about seq2seq

Observation #1: Need more than a simple, start to finish, one-to-one mapping between input tokens & output tokens.

Does observation #1 apply to text to speech?

Some observations about seq2seq

Observation #2: We often need to consider multiple items in the input sequence to produce the right item in the output sequence

Example: He makes a cake

vs.

Example: He makes me happy

Possible translations of “make” into French

faire, fabriquer, préparer, établir, former, prendre, passer, rendre, faciliter, réaménagé, forcer, obliger, atteindre, gagner, réussir, marquer, tourner, arriver, passer, entrer…

Example: He makes a cake → Il fait un gâteau

vs.

Example: He makes me happy → Il me rend heureux

Example: He makes a cake → Il fait un gâteau

vs.

Example: He makes me happy → Il me rend heureux

Sequence-to-sequence models generally include attention mechanisms, which learn which input items we should be paying attention to when generating each output item.

Does observation #2 apply to text to speech?

Example 1: Tokens later in the input sequence

Example 2: Tokens earlier in the input sequence

The Panda eats, shoots and leaves
The Panda eats shoots and leaves

Some observations about seq2seq

Observation #3: When generating output items, we need to consider (some of) the output sequence we’ve already generated.

Example: Il a mangé un sandwich

Could be translated as:

• He has eaten a sandwich
• He ate a sandwich

If a model “looks back” at the output sequence (more formally, if it generates each output item by conditioning on previously generated items) we say it is “autoregressive” or “recurrent”.

Autoregressive models typically give a more fluent output, but they pose performance challenges (as we’ll see).

Does observation #3 apply to text to speech?

Short answer: yes.

For speech synthesis, deep learning techniques generally outperform traditional approaches.

Characteristics of the sequence-to-sequence (seq2seq) problem

1. No one-to-one (or one-to-N) mapping between input items and output items
2. An output item could depend on a weighted combination of input items (attention)
3. We may need to look back at the output sequence generated so far to ensure fluency (autoregressive)

Audio synthesis

We’re generating waveforms

Waveform frequency

How to store waveforms digitally?

Reduced sample rate

Choosing a sample rate

Sample rate comparison

24kHz:
16kHz:
8kHz:

(Source: “Prayer St Francis” by shadoWisp on freesound, licensed under CCBY 3.0)

Choosing a sample rate

At this stage you hopefully understand what a waveform is (amplitude changing over time), and how it can be digitized (by taking thousands of discretes samples per second of the changing amplitude).

Recall that in seq2seq problems, we’ll usually (with autoregressive models) need to look back in the sequence generated so far. Let’s say we wanted to check the previous second of audio that was generated, to ensure fluency.

Solutions?

1. Avoid autoregressive models (likely to hurt quality)
2. Find a more efficient representation for our output sequence than a waveform

Enter the spectrogram

First, recall that for a pure tone, like this:

We can describe it very efficiently, e.g. a 800Hz sine wave, at 60dB, lasting for 1 second (no need for thousands of samples!).

What if we want to describe a more complex sound?

Like a cello (below), or speech?

Another useful concept is additive synthesis/harmonics

Here’s an audible example

300Hz tone:
400Hz tone:
500Hz tone:

300Hz + 400Hz + 500Hz tone:

Let’s say this is 800Hz wave + 200Hz wave

Another useful tool: discrete Fourier transform

Another useful tool: discrete Fourier transform

Mel spectrogram idea #1

Effectively a data compression technique. Like many compression techniques, we’ll optimise for human perception:

As with image compression techniques we’ll ignore differences that humans can’t perceive, and preserve differences which humans can percieve.

Mel spectrogram idea #2

Let’s figure out a way to accurately, but efficiently describe a short snippet of audio (~1/20th of a second).

Mel spectrogram idea #2

Mel spectrogram idea #3

Now we can efficiently describe for a single “frame” of audio (~1/20th of a second), to describe a longer audio waveform, we just repeat the process as many times as neccessary.

Example spectrogram

Spectrogram: test yourself

Which is the female speaker, which the male speaker? (They’re saying the same sentence).

Spectrogram: advantages 1

We’ve dramatically reduced the space required to describe an audio clip. For example a 10 second clip sampled at 22050Hz would contain 220,000 data points when represented as a waveform. As a mel spectrogram, we need only approximately 80 × 300 or 24,000 - an order of magnitude reduction.

Additionally, for the purposes of autoregressive models, looking back 1 second now only means looking back ~30 or so steps (reduction by 3 orders of magnitude).

Spectrogram: advantages 2

Because we’re ignoring frequencies that humans can’t hear, and scaling our frequency scale to match human perception, only differences in our audio files which are perceptible to humans should be registered in our spectrogram (and vice versa).

Spectrogram: the big disadvantage

Spectrograms are a lossy format, as we’ll see if we convert audio into a spectrogram, then naively convert it back to an audio waveform (i.e. an audio file):

Before:
After:

Spectrogram: why the loss of fidelity?

Well, we’re using a finite number of channels (e.g. 80) to capture all the possible frequencies in the spectrum of human hearing.

Phase

Subtle phase shifts of component frequencies distort our resultant waveform

Solving the phase problem

Our mel spectrogram doesn’t include phase information, that’s the key reason for the unpleasant “tinny” distortions when I convert it back to audio.

Short answer: no! We have to be ruthless, our spectrograms have greatly reduced the footprint of our data, so ditching phase is a reasonable sacrifice!

Additionally, as you can see above, the phase information doesn't have a clean structure in the way that our mel spec does (it sort of looks like noise). This will be hard to compress, and won't be a great input to our model.

Our example from before used an algorithm called Griffin-Lim, which sets the phase randomly (and then does repeatedly fowards- and backwards- Fourier transforms). It gives an OK approximation, with some distortion.

However, as we’ll see in the next section, we’ll be able to recover the phase information almost perfectly, using a specially trained deep learning model (called a vocoder).

Audio synthesis: summary

• Ultimately we want to produce a time domain waveform which describes changes in amplitude (i.e. air pressure, “loudness”) over time
• We’ll use a mel spectrogram as a convenient compressed representation
  • Mel spectrograms describe sounds as a weighted combination of (usually 80) human perceptible frequencies channels, each 1/20th of a second (or so)
  • Mel spectograms overcome the “curse of dimensionality” for autoregressive TTS models
• Mel spectograms throw away phase information. We’ll need to reconstruct it using either an approximate method (Griffin-Lim, sounds OK) or using a vocoder (sounds much better)

A modern TTS pipeline

Tacotron 2 - A bit of history

• Tacotron 1 was introduced in a March 2017 paper by Google researchers
• Quickly followed up by Tacotron 2 (December 2017) which improved on + simplified the original
• Google’s implementation is closed source, but high quality open-source implementations exist (as we’ll see)

Tacotron 2 - How does it stack up?

• Broadly speaking, we typically compare models in terms of quality/MOS, robustness, training efficiency, inference efficiency
• Tacotron 2 provides: near SOTA quality, good robustness; but relatively low training & inference efficiency
• Partly Tacotron 2 remains SOTA because the paper’s original model is solid, but also because it’s been improved further thanks to, for example, alternative attention mechanisms

Tacotron 2 - Alternatives

There are many other models out there (beyond the scope of this talk)! Many focus on improved training and/or inference efficiency vs. Tacotron 2.

Some worth checking out include VITS, FastSpeech 2, Transformer Network, AdaSpeech 2, GlowTTS, FastPitch, Flowtron, TalkNet, Grad-TTS.

Tacotron 2 - Architecture

Tacotron 2, loss & learning process

Text: “Hello world”

For each example in our training set we try to predict the spectrogram as accurately as possible, given the transcript. We make our prediction r frames at a time. (We call r the “reduction factor”. Typically r = 2)

Tacotron 2, loss & learning process

Text: “Hello world”

We predict the next r frames. But for the purposes of our prediction, our previously predicted frames are replaced by the frames from the spectrogram in the training data (teacher forcing)

Tacotron 2, loss & learning process

Text: “Hello world”

We predict the next r frames. But for the purposes of our prediction, our previously predicted frames are replaced by the frames from the spectrogram in the training data (teacher forcing)

Tacotron 2, loss & learning process

Text: “Hello world”

We predict the next r frames. But for the purposes of our prediction, our previously predicted frames are replaced by the frames from the spectrogram in the training data (teacher forcing)

Tacotron 2, loss & learning process

Text: “Hello world”

We predict the next r frames. But for the purposes of our prediction, our previously predicted frames are replaced by the frames from the spectrogram in the training data (teacher forcing)

Tacotron 2, loss & learning process

At each step we calculate the difference between our predicted spectrogram frames and our ground truth frames (L2 frame reconstruction loss). This is the key loss we’ll be seeking to minimise during training.

Tacotron 2, attention

Recall that attention will determine the correspondence between our input and output sequences. This means that for TTS attention will control speech pace, rhythm, stress etc.

Tacotron 2, attention

In order to minimise our loss, we’ll need to learn good attention. Learning attention correctly will often represent the bulk of our training effort.

Tacotron 2, attention

We can “swap out” the attention mechanism, giving us a choice of many possible mechanisms. The mechanism we choose can impact training time, robustness and naturalness:

Tacotron 2, attention

• Some attention mechanisms on offer include: Bahdanau attention, location sensitive, location relative/dynamic convolution, forward attention, stepwise monotonic, GMM, windowed, double decoder consistency…
• Too many to explain in detail today!
• I’d recommend Double Decoder Consistency (DDC) or Dynamic Convolution Attention (DCA)

Tacotron 2 - Architecture

Vocoders

• Vocoders are trained for speech audio only (aren’t general mel spectogram → audio converters)
• Can be single-speaker or multi-speaker

Vocoder training loop

Vocoder options

• WaveNet (original vocoder used with Tacotron 2): sounds good, but sloooow
• Several efficient spinoffs of WaveNet: WaveRNN, WaveGrad, WaveGlow (comparable quality but much quicker)
• GAN-based vocoders are beginning to dominate: MelGAN, HifiGAN, VocGAN

Step-by-step guide to training your TTS model

(with Tacotron 2 + vocoder of choice)

Step 0: Choose an open source Tacotron 2 implementation

• There are many good implementations out there. Particular honourable mentions for NVIDIA’s and espnet’s.
• Today, though, we’ll go with the implementation from Coqui (was Mozilla): https://github.com/coqui-ai/TTS
  • High quality implementation which yields good results
  • DDC attention mechanism built-in (good default choice which is fast to train)
  • Easy to use

Step 1: Prepare our data

Target data format

• We ultimately want a dataset of 1-20 second audio clips from a single speaker, with accompanying transcripts. Our transcript file is usually just the .wav filename followed by |, followed by the transcript: LJ002-0026|Hello and good morning!
• Shoot for at least 15 hours of audio (research from NVIDIA found it’s hard to learn robust attention with < 15 hours data)
  • More data (beyond 15 hours) will probably be beneficial!

Possible data source: option 1, premade dataset

• Easiest option!
• The LJSpeech dataset (24 hours) is widely used but sounds a little bland IMO
• I’d recommend the en_UK subset of the M-AILABS Speech Dataset, which is similar to LJSpeech but longer (45 hours) and sounds a little nice

Possible data source: option 2, data from full-length audio + transcript

• For my Obama example I bought a (DRM-free) audiobook + ebook copy of A Promised Land
• Then I had to split the long audio into small chunks and align them with the right part of the book. This problem is known as “forced alignment”, and mature tools exist to tackle it.
  • I’d recommend either gentle or DSAlign.
• I used gentle. With a simple, conservative configuration (to minimise the chance of bad transcripts) I was able to align ~60% of the book: about 18 hours of data.

Possible data source: option 3, DIY transcript

• If I have audio but no transcript, I could:
  • Split the audio into small chunks (use voice activity detection to avoid splitting mid-word/mid-phrase).
  • Send the chunks to a service like Amazon Transcribe.
• I’ve heard of this working well, but be careful: errors in the transcription could propagate to your trained model (junk in, junk out ).

Final preprocessing steps

• Trim silences at the beginning and end of clips.
• Ensure our clips are at the same sample rate.
• Normalise the volume levels (if we’re taking clips from disparate sources).
• Possibly discard outlier clips with a particularly long duration.

Creating train-validation split

• Don’t bother keeping a large validation set (e.g. an 80-20 split), just a few minutes of validation clips is fine (training data is too valuable in a TTS context to waste!)
• Don’t bother making a test set, we’ll ultimately judge a final model with MOS anyway.

Step 2: Training

I just point my config to my data directory (containing my transcripts + .wav files and run):

And then the waiting begins. Fully training a model will typically take 12 hours - several days on a decent GPU (i.e. a V100).

Monitoring training

You can use Tensorboard to monitor the progress of your model’s training:

Speeding up training

• Turn on mixed precision if your GPU supports it.
• Initialize from a pretrained model, rather than a “cold” start.
• Gradual training: begin with a high reduction factor (i.e. r = 7), so we make less granular predictions, yielding a “lower resolution” spectrogram but faster training. Then reduce r (i.e. r = 6) and continue training. Repeat until r = 2.

Step 2b: Optionally train your own vocoder

• You can train a vocoder from scratch if you’d like.
• Alternatively, just use a pretrained vocoder from the Coqui team: they have “universal” MelGAN and WaveGrad vocoders available.

Step 3: Synthesize!

Step 3: Synthesize!

“Once training is complete, you can get your model to say anything you’d like.”

Aims (revisited)

• Leave the talk able to train a near state-of-art TTS system, with a voice of your choice, from scratch.
• Understand the problem domain and common architectures for solutions.
• That the paragraph below won’t be gibberish by the end of the session!

a recurrent sequence-to-sequence feature prediction network with attention which predicts a sequence of mel spectrogram frames from an input character sequence, combined with a vocoder which generates time-domain waveform samples conditioned on the predicted mel spectrogram frames. — Tacotron 2 paper

Hearing is Believing: Generating Realistic Speech with Deep Learning

Thanks for listening! Any questions? (You can also drop me a line: me@alexpeattie.com).

Slides online at alexpeattie.com/talks/tts