The official repository for TMLR and COLLAs'23 paper "A Study of Biologically Plausible Neural Network: The Role and Interactions of Brain-Inspired Mechanisms in Continual Learning" We extended the CLS-ER Repo framework with our method (Bio-ANN)
Please see the scripts folder for specific commands
```
python main.py \
--experiment_id {exp_id} \
--model bio_ann \
--backbone sparse_dale_active_dendrites \
--buffer_size {buffer_size} \
--alpha {alpha} \
--beta {beta} \
--dataset {dataset} \
--num_tasks {n_tasks} \
--num_segments {n_tasks} \
--hebbian_update 1 \
--hebbian_lr {hebbian_lr} \
--apply_heterogeneous_dropout 1 \
--dropout_alpha {dropout_alpha} \
--sc_weight {sc_weight} \
--wix_si_weight {wix_si_weight} \
--wei_si_weight {wei_si_weight} \
--output_dir {/path/to/output/folder} \
--tensorboard \
--csv_log \
--mnist_seed 0 \
-
torch==1.7.0
-
torchvision==0.9.0
-
quadprog==0.1.7
If you find the code useful in your research, please consider citing our paper:
@article{sarfraz2023study,
title={A Study of Biologically Plausible Neural Network: The Role and Interactions of Brain-Inspired Mechanisms in Continual Learning},
author={Sarfraz, Fahad and Arani, Elahe and Zonooz, Bahram},
journal={Transactions on Machine Learning Research},
year={2023}
}