Loading video...

Video Failed to Load

Go Home

The boundary between trainable and untrainable neural network hyperparameter configurations is *fractal*! And beautiful! Here is a grid search over a different pair of hyperparameters -- this time learning rate and the mean of the parameter initialization distribution.

250,458 views • 2 years ago •via X (Twitter)

10 Comments

Jascha Sohl-Dickstein's profile picture
Jascha Sohl-Dickstein2 years ago

Have you ever done a dense grid search over neural network hyperparameters? Like a *really dense* grid search? It looks like this (!!). Blueish colors correspond to hyperparameters for which training converges, redish colors to hyperparameters for which training diverges.

Jascha Sohl-Dickstein's profile picture
Jascha Sohl-Dickstein2 years ago

There are similarities between the way in which many fractals are generated, and the way in which we train neural networks. Both involve repeatedly applying a function to its own output. In both cases, that function has hyperparameters that control its behavior.

Jascha Sohl-Dickstein's profile picture
Jascha Sohl-Dickstein2 years ago

In both cases the function iteration can produce outputs that either diverge to infinity or remain happily bounded depending on those hyperparameters. Fractals are often defined by the boundary between hyperparameters where function iteration diverges or remains bounded.

Jascha Sohl-Dickstein's profile picture
Jascha Sohl-Dickstein2 years ago

So it shouldn't (post-hoc) be a surprise that hyperparameter landscapes are fractal. This is a general phenomenon: in these panes we see fractal hyperparameter landscapes for every neural network configuration I tried, including deep linear networks.

Jascha Sohl-Dickstein's profile picture
Jascha Sohl-Dickstein2 years ago

The best performing hyperparameters are typically at the edge of stability -- so when you optimize neural network hyperparameters, you are contending with hyperparameter landscapes that look like this.

Jascha Sohl-Dickstein's profile picture
Jascha Sohl-Dickstein2 years ago

Want to learn more? Blog post: 3-page paper:

Jascha Sohl-Dickstein's profile picture
Jascha Sohl-Dickstein2 years ago

I don't have a SoundCloud, but I did join Anthropic last week, and so far it has exceeded my (high) expectations. I would strongly recommend working there (and using Claude). *this project not done at Anthropic -- this was recreational machine learning on my own time.

Kosta Derpanis's profile picture
Kosta Derpanis2 years ago

Just in time to make the cut for my lecture today. At 45 sec mark. Thanks for sharing!

Mihoda's profile picture
Mihoda2 years ago

I'm not sure what I'm looking at, but my guess at interpretation would be instability.

Kenneth Shinozuka's profile picture
Kenneth Shinozuka2 years ago

beautiful result

Related Videos