├── .gitignore ├── 1. Consolidation simulation.ipynb ├── 2. Exploring prototypical distortions.ipynb ├── 3. Memory distortions in the extended model.ipynb ├── 4. Boundary extension and contraction.ipynb ├── 5. Imagination continued.ipynb ├── 6. Few-shot learning testing.ipynb ├── README.md ├── config.py ├── distortions_utils.py ├── drm_modelling ├── 300ld_100eps_0.001beta_128batch_0.01l1.h5 ├── DRM_VAE_stories_version.ipynb ├── config.py ├── data_preparation.py ├── generative_model.py ├── hopfield_models.py └── lure_words_fraction.pdf ├── end_to_end.py ├── extended_model.py ├── extended_model_utils.py ├── generative_model.py ├── generative_tests.py ├── hopfield_models.py ├── hopfield_utils.py ├── hybrid_model ├── .keep ├── error_thresholds_shapes3d.pdf ├── extended_version_shapes3d_0.03.pdf └── extended_version_shapes3d_0.15.pdf ├── mhn_memories └── .keep ├── misc_plots ├── BE_boxplot.pdf ├── few_shot_testing.png ├── lure_words_fraction.png ├── mhn_drm.png └── mnist_variance_boxplot.pdf ├── model_weights ├── mnist_decoder.h5 ├── mnist_encoder.h5 ├── shapes3d_decoder.h5 └── shapes3d_encoder.h5 ├── outputs ├── decoding_accs.csv └── output_shapes3d_10000items_50eps_20lv_0.001lr_1kl_backup.pdf ├── requirements.txt └── utils.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ellie-as/generative-memory/HEAD/.gitignore -------------------------------------------------------------------------------- /1. Consolidation simulation.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ellie-as/generative-memory/HEAD/1. Consolidation simulation.ipynb -------------------------------------------------------------------------------- /2. Exploring prototypical distortions.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ellie-as/generative-memory/HEAD/2. Exploring prototypical distortions.ipynb -------------------------------------------------------------------------------- /3. Memory distortions in the extended model.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ellie-as/generative-memory/HEAD/3. Memory distortions in the extended model.ipynb -------------------------------------------------------------------------------- /4. Boundary extension and contraction.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ellie-as/generative-memory/HEAD/4. Boundary extension and contraction.ipynb -------------------------------------------------------------------------------- /5. Imagination continued.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ellie-as/generative-memory/HEAD/5. Imagination continued.ipynb -------------------------------------------------------------------------------- /6. Few-shot learning testing.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ellie-as/generative-memory/HEAD/6. Few-shot learning testing.ipynb -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ellie-as/generative-memory/HEAD/README.md -------------------------------------------------------------------------------- /config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ellie-as/generative-memory/HEAD/config.py -------------------------------------------------------------------------------- /distortions_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ellie-as/generative-memory/HEAD/distortions_utils.py -------------------------------------------------------------------------------- /drm_modelling/300ld_100eps_0.001beta_128batch_0.01l1.h5: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ellie-as/generative-memory/HEAD/drm_modelling/300ld_100eps_0.001beta_128batch_0.01l1.h5 -------------------------------------------------------------------------------- /drm_modelling/DRM_VAE_stories_version.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ellie-as/generative-memory/HEAD/drm_modelling/DRM_VAE_stories_version.ipynb -------------------------------------------------------------------------------- /drm_modelling/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ellie-as/generative-memory/HEAD/drm_modelling/config.py -------------------------------------------------------------------------------- /drm_modelling/data_preparation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ellie-as/generative-memory/HEAD/drm_modelling/data_preparation.py -------------------------------------------------------------------------------- /drm_modelling/generative_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ellie-as/generative-memory/HEAD/drm_modelling/generative_model.py -------------------------------------------------------------------------------- /drm_modelling/hopfield_models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ellie-as/generative-memory/HEAD/drm_modelling/hopfield_models.py -------------------------------------------------------------------------------- /drm_modelling/lure_words_fraction.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ellie-as/generative-memory/HEAD/drm_modelling/lure_words_fraction.pdf -------------------------------------------------------------------------------- /end_to_end.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ellie-as/generative-memory/HEAD/end_to_end.py -------------------------------------------------------------------------------- /extended_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ellie-as/generative-memory/HEAD/extended_model.py -------------------------------------------------------------------------------- /extended_model_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ellie-as/generative-memory/HEAD/extended_model_utils.py -------------------------------------------------------------------------------- /generative_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ellie-as/generative-memory/HEAD/generative_model.py -------------------------------------------------------------------------------- /generative_tests.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ellie-as/generative-memory/HEAD/generative_tests.py -------------------------------------------------------------------------------- /hopfield_models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ellie-as/generative-memory/HEAD/hopfield_models.py -------------------------------------------------------------------------------- /hopfield_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ellie-as/generative-memory/HEAD/hopfield_utils.py -------------------------------------------------------------------------------- /hybrid_model/.keep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /hybrid_model/error_thresholds_shapes3d.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ellie-as/generative-memory/HEAD/hybrid_model/error_thresholds_shapes3d.pdf -------------------------------------------------------------------------------- /hybrid_model/extended_version_shapes3d_0.03.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ellie-as/generative-memory/HEAD/hybrid_model/extended_version_shapes3d_0.03.pdf -------------------------------------------------------------------------------- /hybrid_model/extended_version_shapes3d_0.15.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ellie-as/generative-memory/HEAD/hybrid_model/extended_version_shapes3d_0.15.pdf -------------------------------------------------------------------------------- /mhn_memories/.keep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /misc_plots/BE_boxplot.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ellie-as/generative-memory/HEAD/misc_plots/BE_boxplot.pdf -------------------------------------------------------------------------------- /misc_plots/few_shot_testing.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ellie-as/generative-memory/HEAD/misc_plots/few_shot_testing.png -------------------------------------------------------------------------------- /misc_plots/lure_words_fraction.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ellie-as/generative-memory/HEAD/misc_plots/lure_words_fraction.png -------------------------------------------------------------------------------- /misc_plots/mhn_drm.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ellie-as/generative-memory/HEAD/misc_plots/mhn_drm.png -------------------------------------------------------------------------------- /misc_plots/mnist_variance_boxplot.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ellie-as/generative-memory/HEAD/misc_plots/mnist_variance_boxplot.pdf -------------------------------------------------------------------------------- /model_weights/mnist_decoder.h5: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ellie-as/generative-memory/HEAD/model_weights/mnist_decoder.h5 -------------------------------------------------------------------------------- /model_weights/mnist_encoder.h5: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ellie-as/generative-memory/HEAD/model_weights/mnist_encoder.h5 -------------------------------------------------------------------------------- /model_weights/shapes3d_decoder.h5: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ellie-as/generative-memory/HEAD/model_weights/shapes3d_decoder.h5 -------------------------------------------------------------------------------- /model_weights/shapes3d_encoder.h5: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ellie-as/generative-memory/HEAD/model_weights/shapes3d_encoder.h5 -------------------------------------------------------------------------------- /outputs/decoding_accs.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ellie-as/generative-memory/HEAD/outputs/decoding_accs.csv -------------------------------------------------------------------------------- /outputs/output_shapes3d_10000items_50eps_20lv_0.001lr_1kl_backup.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ellie-as/generative-memory/HEAD/outputs/output_shapes3d_10000items_50eps_20lv_0.001lr_1kl_backup.pdf -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ellie-as/generative-memory/HEAD/requirements.txt -------------------------------------------------------------------------------- /utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ellie-as/generative-memory/HEAD/utils.py --------------------------------------------------------------------------------