├── .flake8 ├── .github └── workflows │ └── python-app.yml ├── .gitignore ├── .idea └── .gitignore ├── LICENSE ├── README.md ├── SCRATCH └── celeb-dict-mse │ ├── dict_hist │ └── top100atoms_e3.png │ └── loss_history.png ├── TODO.md ├── UNIT_TESTS ├── unit_test_images │ ├── ISTA-test-visualization.png │ └── fista-test-visualization.png └── unrolled_algo_tests.py ├── bin └── run.py ├── experiments ├── convDictLearning.yml ├── train-linear-dict.yml └── vae-ia-train.yml ├── legacy-code ├── paramSearchResults │ ├── ASIRRA16_0000.png │ ├── CIFAR1010_0210.png │ ├── FashionMNIST10_0220.png │ ├── MNIST32_0000.png │ ├── lossHist.png │ ├── reconErrHist.png │ └── sparsityHist.png ├── results │ ├── MNIST10dictAtoms.png │ ├── MNIST32dictAtoms.png │ ├── data_atoms_comparison.pdf │ ├── lossHist.png │ ├── npl1.png │ ├── reconErrHist.png │ ├── sparsityHist.png │ └── tmp │ │ └── no-name.pt └── sanity_ims │ ├── fixedEncoders │ ├── fista_loss.png │ ├── fista_x.png │ ├── ista_loss.png │ ├── ista_x.png │ ├── salsa_loss.png │ └── salsa_x.png │ └── trainedEncoders │ ├── fista_loss.png │ ├── fista_x.png │ ├── ista_loss.png │ ├── ista_x.png │ ├── salsa_loss.png │ └── salsa_x.png ├── lib ├── UTILS │ ├── image_transforms.py │ ├── path_support.py │ ├── rng_control.py │ └── visualizer.py ├── core │ ├── backbone.py │ └── task.py ├── data_loaders │ ├── SP.png │ ├── dataloader_wesad.py │ ├── glitch_datasets.py │ └── pytorch_datasets.py ├── dataset_generators │ ├── dataset_abs_class.py │ ├── populate_inaccessible_worlds.py │ └── sql_tools.py ├── model_blocks │ ├── AlgorithmBlock.py │ ├── ISTA.py │ ├── ModelGenerator.py │ ├── custom_batchnorm.py │ ├── custom_loss_functions.py │ ├── dictionary.py │ └── experimental │ │ ├── altmin-analyzer.py │ │ ├── class_encoder.py │ │ ├── generate_dataset.py │ │ ├── parallelAltMin_skeletonCode.py │ │ ├── trainers │ │ ├── AlternatingMinimizer.py │ │ ├── encode_decode_trainer.py │ │ └── ssl_trainer.py │ │ └── variational_autoencoders.py └── tasks │ ├── dictionary_learning.py │ ├── model_builders.py │ ├── return_object.py │ ├── trainer_abc.py │ └── tsne_encode.py ├── requirements.txt └── test_all_unit_tests.py /.flake8: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BenCowen/SparseCoding/HEAD/.flake8 -------------------------------------------------------------------------------- /.github/workflows/python-app.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BenCowen/SparseCoding/HEAD/.github/workflows/python-app.yml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BenCowen/SparseCoding/HEAD/.gitignore -------------------------------------------------------------------------------- /.idea/.gitignore: -------------------------------------------------------------------------------- 1 | # Default ignored files 2 | /shelf/ 3 | /workspace.xml 4 | -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BenCowen/SparseCoding/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BenCowen/SparseCoding/HEAD/README.md -------------------------------------------------------------------------------- /SCRATCH/celeb-dict-mse/dict_hist/top100atoms_e3.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BenCowen/SparseCoding/HEAD/SCRATCH/celeb-dict-mse/dict_hist/top100atoms_e3.png -------------------------------------------------------------------------------- /SCRATCH/celeb-dict-mse/loss_history.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BenCowen/SparseCoding/HEAD/SCRATCH/celeb-dict-mse/loss_history.png -------------------------------------------------------------------------------- /TODO.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BenCowen/SparseCoding/HEAD/TODO.md -------------------------------------------------------------------------------- /UNIT_TESTS/unit_test_images/ISTA-test-visualization.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BenCowen/SparseCoding/HEAD/UNIT_TESTS/unit_test_images/ISTA-test-visualization.png -------------------------------------------------------------------------------- /UNIT_TESTS/unit_test_images/fista-test-visualization.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BenCowen/SparseCoding/HEAD/UNIT_TESTS/unit_test_images/fista-test-visualization.png -------------------------------------------------------------------------------- /UNIT_TESTS/unrolled_algo_tests.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BenCowen/SparseCoding/HEAD/UNIT_TESTS/unrolled_algo_tests.py -------------------------------------------------------------------------------- /bin/run.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BenCowen/SparseCoding/HEAD/bin/run.py -------------------------------------------------------------------------------- /experiments/convDictLearning.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BenCowen/SparseCoding/HEAD/experiments/convDictLearning.yml -------------------------------------------------------------------------------- /experiments/train-linear-dict.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BenCowen/SparseCoding/HEAD/experiments/train-linear-dict.yml -------------------------------------------------------------------------------- /experiments/vae-ia-train.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BenCowen/SparseCoding/HEAD/experiments/vae-ia-train.yml -------------------------------------------------------------------------------- /legacy-code/paramSearchResults/ASIRRA16_0000.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BenCowen/SparseCoding/HEAD/legacy-code/paramSearchResults/ASIRRA16_0000.png -------------------------------------------------------------------------------- /legacy-code/paramSearchResults/CIFAR1010_0210.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BenCowen/SparseCoding/HEAD/legacy-code/paramSearchResults/CIFAR1010_0210.png -------------------------------------------------------------------------------- /legacy-code/paramSearchResults/FashionMNIST10_0220.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BenCowen/SparseCoding/HEAD/legacy-code/paramSearchResults/FashionMNIST10_0220.png -------------------------------------------------------------------------------- /legacy-code/paramSearchResults/MNIST32_0000.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BenCowen/SparseCoding/HEAD/legacy-code/paramSearchResults/MNIST32_0000.png -------------------------------------------------------------------------------- /legacy-code/paramSearchResults/lossHist.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BenCowen/SparseCoding/HEAD/legacy-code/paramSearchResults/lossHist.png -------------------------------------------------------------------------------- /legacy-code/paramSearchResults/reconErrHist.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BenCowen/SparseCoding/HEAD/legacy-code/paramSearchResults/reconErrHist.png -------------------------------------------------------------------------------- /legacy-code/paramSearchResults/sparsityHist.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BenCowen/SparseCoding/HEAD/legacy-code/paramSearchResults/sparsityHist.png -------------------------------------------------------------------------------- /legacy-code/results/MNIST10dictAtoms.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BenCowen/SparseCoding/HEAD/legacy-code/results/MNIST10dictAtoms.png -------------------------------------------------------------------------------- /legacy-code/results/MNIST32dictAtoms.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BenCowen/SparseCoding/HEAD/legacy-code/results/MNIST32dictAtoms.png -------------------------------------------------------------------------------- /legacy-code/results/data_atoms_comparison.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BenCowen/SparseCoding/HEAD/legacy-code/results/data_atoms_comparison.pdf -------------------------------------------------------------------------------- /legacy-code/results/lossHist.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BenCowen/SparseCoding/HEAD/legacy-code/results/lossHist.png -------------------------------------------------------------------------------- /legacy-code/results/npl1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BenCowen/SparseCoding/HEAD/legacy-code/results/npl1.png -------------------------------------------------------------------------------- /legacy-code/results/reconErrHist.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BenCowen/SparseCoding/HEAD/legacy-code/results/reconErrHist.png -------------------------------------------------------------------------------- /legacy-code/results/sparsityHist.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BenCowen/SparseCoding/HEAD/legacy-code/results/sparsityHist.png -------------------------------------------------------------------------------- /legacy-code/results/tmp/no-name.pt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BenCowen/SparseCoding/HEAD/legacy-code/results/tmp/no-name.pt -------------------------------------------------------------------------------- /legacy-code/sanity_ims/fixedEncoders/fista_loss.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BenCowen/SparseCoding/HEAD/legacy-code/sanity_ims/fixedEncoders/fista_loss.png -------------------------------------------------------------------------------- /legacy-code/sanity_ims/fixedEncoders/fista_x.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BenCowen/SparseCoding/HEAD/legacy-code/sanity_ims/fixedEncoders/fista_x.png -------------------------------------------------------------------------------- /legacy-code/sanity_ims/fixedEncoders/ista_loss.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BenCowen/SparseCoding/HEAD/legacy-code/sanity_ims/fixedEncoders/ista_loss.png -------------------------------------------------------------------------------- /legacy-code/sanity_ims/fixedEncoders/ista_x.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BenCowen/SparseCoding/HEAD/legacy-code/sanity_ims/fixedEncoders/ista_x.png -------------------------------------------------------------------------------- /legacy-code/sanity_ims/fixedEncoders/salsa_loss.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BenCowen/SparseCoding/HEAD/legacy-code/sanity_ims/fixedEncoders/salsa_loss.png -------------------------------------------------------------------------------- /legacy-code/sanity_ims/fixedEncoders/salsa_x.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BenCowen/SparseCoding/HEAD/legacy-code/sanity_ims/fixedEncoders/salsa_x.png -------------------------------------------------------------------------------- /legacy-code/sanity_ims/trainedEncoders/fista_loss.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BenCowen/SparseCoding/HEAD/legacy-code/sanity_ims/trainedEncoders/fista_loss.png -------------------------------------------------------------------------------- /legacy-code/sanity_ims/trainedEncoders/fista_x.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BenCowen/SparseCoding/HEAD/legacy-code/sanity_ims/trainedEncoders/fista_x.png -------------------------------------------------------------------------------- /legacy-code/sanity_ims/trainedEncoders/ista_loss.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BenCowen/SparseCoding/HEAD/legacy-code/sanity_ims/trainedEncoders/ista_loss.png -------------------------------------------------------------------------------- /legacy-code/sanity_ims/trainedEncoders/ista_x.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BenCowen/SparseCoding/HEAD/legacy-code/sanity_ims/trainedEncoders/ista_x.png -------------------------------------------------------------------------------- /legacy-code/sanity_ims/trainedEncoders/salsa_loss.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BenCowen/SparseCoding/HEAD/legacy-code/sanity_ims/trainedEncoders/salsa_loss.png -------------------------------------------------------------------------------- /legacy-code/sanity_ims/trainedEncoders/salsa_x.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BenCowen/SparseCoding/HEAD/legacy-code/sanity_ims/trainedEncoders/salsa_x.png -------------------------------------------------------------------------------- /lib/UTILS/image_transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BenCowen/SparseCoding/HEAD/lib/UTILS/image_transforms.py -------------------------------------------------------------------------------- /lib/UTILS/path_support.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BenCowen/SparseCoding/HEAD/lib/UTILS/path_support.py -------------------------------------------------------------------------------- /lib/UTILS/rng_control.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BenCowen/SparseCoding/HEAD/lib/UTILS/rng_control.py -------------------------------------------------------------------------------- /lib/UTILS/visualizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BenCowen/SparseCoding/HEAD/lib/UTILS/visualizer.py -------------------------------------------------------------------------------- /lib/core/backbone.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BenCowen/SparseCoding/HEAD/lib/core/backbone.py -------------------------------------------------------------------------------- /lib/core/task.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BenCowen/SparseCoding/HEAD/lib/core/task.py -------------------------------------------------------------------------------- /lib/data_loaders/SP.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BenCowen/SparseCoding/HEAD/lib/data_loaders/SP.png -------------------------------------------------------------------------------- /lib/data_loaders/dataloader_wesad.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BenCowen/SparseCoding/HEAD/lib/data_loaders/dataloader_wesad.py -------------------------------------------------------------------------------- /lib/data_loaders/glitch_datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BenCowen/SparseCoding/HEAD/lib/data_loaders/glitch_datasets.py -------------------------------------------------------------------------------- /lib/data_loaders/pytorch_datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BenCowen/SparseCoding/HEAD/lib/data_loaders/pytorch_datasets.py -------------------------------------------------------------------------------- /lib/dataset_generators/dataset_abs_class.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BenCowen/SparseCoding/HEAD/lib/dataset_generators/dataset_abs_class.py -------------------------------------------------------------------------------- /lib/dataset_generators/populate_inaccessible_worlds.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BenCowen/SparseCoding/HEAD/lib/dataset_generators/populate_inaccessible_worlds.py -------------------------------------------------------------------------------- /lib/dataset_generators/sql_tools.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BenCowen/SparseCoding/HEAD/lib/dataset_generators/sql_tools.py -------------------------------------------------------------------------------- /lib/model_blocks/AlgorithmBlock.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BenCowen/SparseCoding/HEAD/lib/model_blocks/AlgorithmBlock.py -------------------------------------------------------------------------------- /lib/model_blocks/ISTA.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BenCowen/SparseCoding/HEAD/lib/model_blocks/ISTA.py -------------------------------------------------------------------------------- /lib/model_blocks/ModelGenerator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BenCowen/SparseCoding/HEAD/lib/model_blocks/ModelGenerator.py -------------------------------------------------------------------------------- /lib/model_blocks/custom_batchnorm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BenCowen/SparseCoding/HEAD/lib/model_blocks/custom_batchnorm.py -------------------------------------------------------------------------------- /lib/model_blocks/custom_loss_functions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BenCowen/SparseCoding/HEAD/lib/model_blocks/custom_loss_functions.py -------------------------------------------------------------------------------- /lib/model_blocks/dictionary.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BenCowen/SparseCoding/HEAD/lib/model_blocks/dictionary.py -------------------------------------------------------------------------------- /lib/model_blocks/experimental/altmin-analyzer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BenCowen/SparseCoding/HEAD/lib/model_blocks/experimental/altmin-analyzer.py -------------------------------------------------------------------------------- /lib/model_blocks/experimental/class_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BenCowen/SparseCoding/HEAD/lib/model_blocks/experimental/class_encoder.py -------------------------------------------------------------------------------- /lib/model_blocks/experimental/generate_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BenCowen/SparseCoding/HEAD/lib/model_blocks/experimental/generate_dataset.py -------------------------------------------------------------------------------- /lib/model_blocks/experimental/parallelAltMin_skeletonCode.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BenCowen/SparseCoding/HEAD/lib/model_blocks/experimental/parallelAltMin_skeletonCode.py -------------------------------------------------------------------------------- /lib/model_blocks/experimental/trainers/AlternatingMinimizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BenCowen/SparseCoding/HEAD/lib/model_blocks/experimental/trainers/AlternatingMinimizer.py -------------------------------------------------------------------------------- /lib/model_blocks/experimental/trainers/encode_decode_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BenCowen/SparseCoding/HEAD/lib/model_blocks/experimental/trainers/encode_decode_trainer.py -------------------------------------------------------------------------------- /lib/model_blocks/experimental/trainers/ssl_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BenCowen/SparseCoding/HEAD/lib/model_blocks/experimental/trainers/ssl_trainer.py -------------------------------------------------------------------------------- /lib/model_blocks/experimental/variational_autoencoders.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BenCowen/SparseCoding/HEAD/lib/model_blocks/experimental/variational_autoencoders.py -------------------------------------------------------------------------------- /lib/tasks/dictionary_learning.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BenCowen/SparseCoding/HEAD/lib/tasks/dictionary_learning.py -------------------------------------------------------------------------------- /lib/tasks/model_builders.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BenCowen/SparseCoding/HEAD/lib/tasks/model_builders.py -------------------------------------------------------------------------------- /lib/tasks/return_object.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BenCowen/SparseCoding/HEAD/lib/tasks/return_object.py -------------------------------------------------------------------------------- /lib/tasks/trainer_abc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BenCowen/SparseCoding/HEAD/lib/tasks/trainer_abc.py -------------------------------------------------------------------------------- /lib/tasks/tsne_encode.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BenCowen/SparseCoding/HEAD/lib/tasks/tsne_encode.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BenCowen/SparseCoding/HEAD/requirements.txt -------------------------------------------------------------------------------- /test_all_unit_tests.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BenCowen/SparseCoding/HEAD/test_all_unit_tests.py --------------------------------------------------------------------------------