Replication Files for the paper "Simplicity Bias of Two-Layer Networks beyond Linearly-Separable Data"

This repository presents replication files for the paper "Simplicity Bias of Two-Layer Networks beyond Linearly-Separable Data" (2024) by N. Tsoy and N. Konstantinov, published at ICML 2024.

Replication Instuctions

1. Run code-create-dominos.ipynb to generate domino dataset.
2. Run the uploaded Jupyter notebooks file to generate Figures from the paper.

Packages Used

1. Python 3.11.6
2. Numpy 1.24.2
3. Matplotlib 3.6.3
4. Jupyter Notebook
5. PyTorch 2.1.2+cu121
6. TorchVision 0.16.2+cu121