This project is a playground for experimenting with various deep learning techniques, particularly focusing on neural network layers and transformations. It provides implementations of several advanced techniques and includes experiments to demonstrate their applications.
- src/dl_techniques/layers: Contains implementations of various deep learning layers and techniques, such as convolutional transformers, differentiable KMeans, Gaussian filters, and more.
- src/dl_techniques/regularizers: Includes regularization techniques to improve model generalization.
- src/dl_techniques/utils: Utility functions for logging, tensor operations, and visualization.
- src/experiments: Scripts demonstrating the application of the implemented techniques, such as KMeans clustering and logit normalization experiments.
- tests: Unit tests for the implemented layers, regularizers, and utilities.
To install the required dependencies, run:
pip install -r requirements.txtTo run the KMeans clustering demo, execute:
python src/experiments/basic.pyTo run the logit normalization experiments, execute:
python src/experiments/coupled_logit_norm.py- numpy
- pytest
- pytest-cov
- matplotlib
- scikit-learn
- keras~=3.8.0
- tensorflow==2.18.0
This project is licensed under the MIT License. See the LICENSE file for more details.
Contributions are welcome! Please open an issue or submit a pull request for any improvements or new features.
For any questions or inquiries, please contact the project maintainer.