- Notifications
You must be signed in to change notification settings - Fork0
The official repository for TMLR paper "A Study of Biologically Plausible Neural Network: The Role and Interactions of Brain-Inspired Mechanisms in Continual Learning""
License
NotificationsYou must be signed in to change notification settings
NeurAI-Lab/Bio-ANN
Folders and files
| Name | Name | Last commit message | Last commit date | |
|---|---|---|---|---|
Repository files navigation
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 theCLS-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}}