This is a repo for the individual course assignments of DD2424 Deep Learning in Data Science at KTH 2020. The code in this repo is mainly done in Matlab, and manipulations involved in the training process like gradient calculations and parameter updates are implemented in a trival way (low level).
In this project, A k-layers neural networks is designed with the implemention of He Initialisation, Cyclical Learning Rates, Batch Normalisation. The objective is to use these neural network models to classfy images from the CIFAR-10 dataset.
- Test Accuracy: 36.96%
Simple 1-layer neural network
- Test Accuracy: 51.44%
2-layer neural network + Cyclical Learning Rates
- Test Accuracy: 53.33%
k-layer neural network + Cyclical Learning Rates + Batch Normalisation
Chieh-Ju Wu (Jeremy) - jeremy.cjwukth@gmail.com