Epoch wise double descent based on paper https://arxiv.org/abs/2108.12006
These are the results of a project we did in ML2 course (097209) at the 'The Technion – Israel Institute of Technology'.
- Based on Running Resnet18 on CIFAR10
- code based on https://github.com/mohammadpz/Epoch_wise_Double_Descent
- In contrast to above, we took a subset of categories and preformed PCA to remove EDD which above showed.
- We implemented the second method mentioned in the paper using a simple Fully connected net and a synthetic dataset.
- This method was supposed to eliminate the double descent by replacing the weights of the last layer with converged weights, and continue training.
- We show that for a specific input, it indeed eliminated the double descent, but these results vary from run to run, where for some runs there is no convergence at all (not show here).