This repository explores 3 deep learning architectures for classification tasks:
- ResNet: Implemented from scratch.
- Xception: Transfer learning with fine-tuning.
- DenseNet: Transfer learning with fine-tuning.
- Introduced in "Deep Residual Learning for Image Classification" by He et al. (2015).
- Key Idea: Skip connections to combat vanishing gradients.
- Implemented from scratch in this project.
- Presented in "Xception: Deep Learning with Depthwise Separable Convolutions" by Chollet (2017).
- Focuses on depthwise separable convolutions to reduce computation.
- Proposed in "Densely Connected Convolutional Networks" by Huang et al. (2017).
- Ensures feature reuse by connecting each layer to all others.
- Accuracy: Measures the percentage of correctly classified samples.
- Confusion Matrix: Visualized to analyze true/false positives and negatives.
- Recall, Precision, F1-Score: Highlights the model's performance balance.
- ROC and AUC: Visualizes the trade-off between sensitivity and specificity.
References (references updated and the actual references for the project are mentioned in the documentation file)
- ResNet: He et al., 2015 Deep Residual Learning for Image Recognition
- Xception: Chollet, 2017 Xception: Deep Learning with Depthwise Separable Convolutions
- DenseNet: Huang et al., 2017 Densely Connected Convolutional Networks