Skip to content

vickshan001/CIFAR-10-CNN-Enhancer-Neural-Networks

Folders and files

NameName
Last commit message
Last commit date

Latest commit

Β 

History

4 Commits
Β 
Β 
Β 
Β 

Repository files navigation

🧠 CIFAR-10 CNN Enhancer – Neural Networks

The aim was to improve the classification accuracy of a CNN model on the CIFAR-10 dataset through architecture tuning, data augmentation, and dropout regularization.


πŸ“¦ Dataset

CIFAR-10 – a 60,000-image dataset across 10 classes like airplane, bird, cat, deer, dog, etc.


🧠 Key Enhancements

πŸ” Data Augmentation

  • RandomHorizontalFlip()
  • RandomCrop(32, padding=4)

🧱 Model Architecture

  • Intermediate block improvements:
    • Dropout for regularization
    • Adapted fully connected layers
  • Output block:
    • Multiple FC layers with ReLU activation
    • Final FC layer outputs raw logits

πŸ§ͺ Initialization & Optimisation

  • Xavier (Glorot) initialization for weights
  • Adam Optimizer with CrossEntropy Loss

πŸ“Š Training Results

  • Accuracy increased gradually across epochs
  • Final Test Accuracy: 62%
  • Visualization of loss and accuracy over epochs


πŸ“‚ Project Structure

  • Final_Score.ipynb – Full notebook including architecture, training, and evaluation

πŸš€ How to Run

  1. Clone the repository
  2. Run Final_Score.ipynb in Jupyter Notebook
  3. Required Libraries:
    • torch, torchvision, numpy, matplotlib

🏫 Module Info

  • πŸ“… Year: 2023/24
  • 🏫 University: Queen Mary University of London
  • πŸ‘¨β€πŸ’» Author: Vickshan Vicknakumaran

πŸ“œ License

For educational and research purposes only.

About

CNN classifier for CIFAR-10 with enhanced architecture, dropout, and data augmentation.

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published