Hierarchical nucleation in deep neural networks

This repository contains the code to reproduce the results of the paper Hierarchical nucleation in deep neural networks. This work has been included in the NeurIPS 2020 Proceedings.

Requirements

We will use Anaconda to create the environment. You can install it by following:

https://docs.conda.io/projects/conda/en/latest/user-guide/install/

You can create the environment with the following command:

conda env create -f environment.yml

Reproduce the paper plots.

You can reproduce the paper results following the examples in the jupyter notebook:

conda activate hier_nucl
jupyter notebook hier_nucl Hierarchical_nucleation_in_deep_networks.ipynb