feldberlin / wavenet

An unconditioned Wavenet implementation with fast generation.
3 stars 0 forks source link

Reproduce track overfitting #27

Closed purzelrakete closed 3 years ago

purzelrakete commented 3 years ago

What

We would like to reproduce perfect track overfitting, as in #4. We expect the generated tracks to perfectly reproduce the training data.

Results

Write up the results of your experiment once it has completed and has been analysed. Include links to the treatment run, and also to the baseline if appropriate.

Acceptance Criteria

purzelrakete commented 3 years ago

Experiments

Baseline experiment notebook. Launched with:

bin/train track -p batch_norm False

We don't have a notebook for the baseline we are reproducing, so re-running without batch norm. This gets the same or slightly better results than the previous baseline for track. Rendered tracks sound like the input ✅

Treatment experiment notebook. Launched with:

bin/train track -p batch_norm True -p learning_rate 0.01
purzelrakete commented 3 years ago

Results

This turned into a bit of a batch norm experiment. That's because when we use the standard batch norm setup, the rendered audio no longer sounds like piano.

train nll test nll
baseline 0.0006 7.385
treatment 0.075 1.322

We can say that

Insights

These results aren't very surprising. Batch norm has a regularising effect, which explains the better test likelihood, and also the worse training error.

Open questions