0% found this document useful (0 votes)
25 views8 pages

Transfer Learning in MNIST Models

The document discusses neural networks and softmax regression models built on the MNIST dataset. Models were built for even and odd digit classifications, obtaining over 95% accuracy. Transfer learning was also explored, showing comparable performance to initial models.

Uploaded by

emailpython6
Copyright
© All Rights Reserved
We take content rights seriously. If you suspect this is your content, claim it here.
Available Formats
Download as PDF, TXT or read online on Scribd
0% found this document useful (0 votes)
25 views8 pages

Transfer Learning in MNIST Models

The document discusses neural networks and softmax regression models built on the MNIST dataset. Models were built for even and odd digit classifications, obtaining over 95% accuracy. Transfer learning was also explored, showing comparable performance to initial models.

Uploaded by

emailpython6
Copyright
© All Rights Reserved
We take content rights seriously. If you suspect this is your content, claim it here.
Available Formats
Download as PDF, TXT or read online on Scribd

Introduction

Softmax Regression
Neural Network
Transfer Learning
Conclusion

NEURAL NETWORKS WITH MNIST


DATASET.

Duaa Alshareif, Tony Chisenga, Ignatius Boadi

African Institute for Mathematical Sciences, AIMS-Senegal

Supervised by Soona Osman


April 2, 2024

Presenters NEURAL NETWORKS WITH MNIST DATASET. 1/8


Introduction
Softmax Regression
Neural Network
Transfer Learning
Conclusion

Outlines

1 Introduction

2 Softmax Regression

3 Neural Network

4 Transfer Learning

5 Conclusion

Presenters NEURAL NETWORKS WITH MNIST DATASET. 2/8


Introduction
Softmax Regression
Neural Network
Transfer Learning
Conclusion

Introduction

The MNIST dataset was read using Pytorch. There were


60000 sets of images with their labels for training and 10000
for testing.
The sets of data for both training and testing were further
divided into even and odd numbers sets, with classification
models built for each dataset.
Softmax regression and neural networks with one hidden layer
(ReLU activation) and softmax on the output layer were built
for these datasets.
A softmax regression model was also built for each of the
datasets using transfer learning with output from the first
layer of the trained neural networks.

Presenters NEURAL NETWORKS WITH MNIST DATASET. 3/8


Introduction
Softmax Regression
Neural Network
Transfer Learning
Conclusion

Softmax Regression
Softmax regression models were trained on the even and odd
MNIST datasets.
These models obtained accuracy levels of 96% and 95% on
the test dataset for the even and odd datasets respectively.

Presenters NEURAL NETWORKS WITH MNIST DATASET. 4/8


Introduction
Softmax Regression
Neural Network
Transfer Learning
Conclusion

Neural Network with one hidden layer


Neural network with one hidden layer (ReLU activation) were
trained on the even and odd MNIST datasets.
These models obtained accuracy levels of 98% and 97.7% on
the test dataset for the even and odd datasets respectively.

Presenters NEURAL NETWORKS WITH MNIST DATASET. 5/8


Introduction
Softmax Regression
Neural Network
Transfer Learning
Conclusion

Transfer Learning
A softmax regression model was trained on the even and odd
MNIST datasets.

Presenters NEURAL NETWORKS WITH MNIST DATASET. 6/8


Introduction
Softmax Regression
Neural Network
Transfer Learning
Conclusion

Comparisons and observations


The models generalize well as their performance were good
and consistent across the test datasets.
The softmax regression models built with transfer learning had
comparative performance with initial softmax regression. It
outperforms the initial softmax regression model on the odd
dataset.
This suggests that features from a model can be used to train
a neural network using similar datasets.
Dataset/Model Softmax Reg NN Trasfer learning
Even Dataset 96% 98% 95%
Odd Dataset 95% 97.7% 97%
Table: Model accuracy comparisons

Presenters NEURAL NETWORKS WITH MNIST DATASET. 7/8


Introduction
Softmax Regression
Neural Network
Transfer Learning
Conclusion

Conclusion

Transfer learning generally leads to better performance.


Explore transfer learning on neural networks.

Presenters NEURAL NETWORKS WITH MNIST DATASET. 8/8

Common questions

Powered by AI

Transfer learning played a role in enhancing model performance by leveraging the features from a trained model on a similar dataset. It improved the accuracy of the softmax regression model for the odd dataset from 95% to 97%. This suggests that transfer learning can help models achieve better performance by reusing learned representations from one model to facilitate the training of another .

The initial softmax regression model for the odd MNIST dataset achieved an accuracy of 95%. In contrast, the model enhanced with transfer learning improved to 97%. This highlights transfer learning's effectiveness in boosting model performance on similar datasets .

Transfer learning might outperform initial softmax regression on the odd dataset specifically because it leverages existing model knowledge to fine-tune learning for particular datasets. This process targets improvements in feature extraction and generalization, possibly addressing specific challenges unique to the odd dataset .

The activation function used in the neural network with one hidden layer was ReLU. The significance of using ReLU lies in its ability to efficiently handle gradient vanishing problems by enabling faster convergence of the model, making it a preferred choice for deep neural networks .

The softmax regression models achieved an accuracy of 96% on the even dataset and 95% on the odd dataset. In contrast, neural networks with one hidden layer reached 98% accuracy on the even dataset and 97.7% on the odd dataset, indicating that the neural networks performed better on both datasets .

Transfer learning offers the advantage of improving model performance by utilizing pre-trained knowledge from existing models. It allows for faster training, reduced computational resources, and improved accuracy by transferring learned features, especially when training on similar datasets .

It was concluded that softmax regression models performed well on the MNIST dataset, reaching 96% accuracy on the even dataset and 95% on the odd. Transfer learning further enhanced performance on the odd dataset, suggesting that previously learned features can improve model accuracy .

The main observation was that both softmax regression and neural networks with one hidden layer generalized well, demonstrating good and consistent performance across the test datasets. The models maintained high accuracy levels, suggesting robust performance on unseen data .

The MNIST dataset, consisting of 60,000 training and 10,000 testing images, was used. It might have been divided into even and odd sets to test model robustness against different data splits and to evaluate model performance under varying conditions. This division also helps understand how well models can generalize across dataset manipulations .

The document's outline, which moves from introduction through different modeling techniques to conclusions, suggests a hierarchical approach to understanding and applying machine learning techniques on the MNIST dataset. It indicates a structured methodology progressing from basic model construction to complex techniques like transfer learning .

You might also like