Skip to content

New complexity measure for neural networks in the double descent context

License

Notifications You must be signed in to change notification settings

VictorBaillet/double-descent

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

17 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Kernel-Based Complexity Measure for the Study of Double Descent in Neural Networks

We investigate double descent phenomena in neural networks, using a novel kernel-based complexity measure. Building upon the work of Curth et al. 2023

Installation

  1. Clone the repository:

    git clone https://github.com/VictorBaillet/double-descent
    cd double-descent
  2. Set up the environment:

    python -m venv venv
    source venv/bin/activate  # For Windows, use `venv\Scripts\activate`
    pip install -r requirements.txt
  3. Launch the Jupyter notebook for a hands-on experience with the results.

Run experiments

To run the experiments on your machine :

```bash
python main.py config/[experiment_name].json
```

Datasets Employed

The study employs two datasets :

  • MNIST
  • CIFAR-10

About

New complexity measure for neural networks in the double descent context

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published