alecruces/fashiondbn

Uncover the secrets of deep learning with FashionDBN - implementing PyTorch's Deep Belief Network for accurate image classification and beyond.

Implementing a Deep Belief Network in PyTorch using the FashionMNIST dataset

---

Description

The project focuses on the training of a Deep Belief Network using various hyperparameters such as hidden layers, nodes, and optimizer choices for optimizing performance in image classification. It analyzes the internal representations developed by the model through hierarchical clustering and feature visualization techniques. The goal is to understand the accuracy and psychometric curves for the DBN. Finally, it compares the ability of two models—the feedforward network and the DBN—to resist adversarial attacks.

Keywords

Image Classification, Deep Belief Network (DBN), FashionMNIST, Adversarial Attacks

Data

• FashionMNIST dataset

Methods

• Deep Belief Network
• Feedforward networks

Software

• Python (Pytorch)

Files

• Code:
  • code.ipynb: Jupyter notebook that run al the project.
  • DBN.py and RBN.py: Have the functions called by code.ipynb.