This is the repo for the following paper:
Lu, Q., Nguyen, T. T., Zhang, Q., Hasson, U., Griffiths, T. L., Zacks, J. M., Gershman, S. J., & Norman, K. A. (2023). Reconciling Shared versus Context-Specific Information in a Neural Network Model of Latent Causes. In arXiv [q-bio.NC]. arXiv. http://arxiv.org/abs/2312.08519
This repo contains the code for Simulation 1 and Simulation 2. The code for simulation 3 is in a separate repo here.
For Simulation 1, run the following code under src
python sim-poly.py
For Simulation 2, run the following code under src
python sim-csw.py
. ├── LICENSE ├── README.md └── src ├── demo │ └── ...... ├── model # model components │ ├── A2C.py │ ├── CGRU.py │ ├── CRPLSTM.py │ ├── CRPNN.py │ ├── ContextRep.py │ ├── GRUA2C.py │ ├── PECRP.py │ ├── PEKDECRP.py │ ├── PEKNN.py │ ├── PEKNNCRP.py │ ├── PETracker.py │ ├── ShortCut.py │ ├── SimpleContext.py │ ├── SimplePETracker.py │ ├── TabularShortCut.py │ ├── TabularShortCutIS.py │ ├── __init__.py │ ├── anderson91.py │ └── utils.py ├── task # definition for the tasks │ ├── _Polynomial.py # task used in Simulation 1 │ ├── _CSW.py # task used in Simulation 2 │ ├── _ContextualBandit.py │ ├── _MixedCSW.py │ ├── _SimpleTwoArmBandit.py │ ├── _Waves.py │ └── __init__.py ├── sim-poly.py # code for Simulation 1 ├── sim-csw.py # code for Simulation 2 - model training ├── sim-mixedcsw.py ├── stats.py ├── utils.py ├── vis-group-poly.py # code for Simulation 2 - result visualization └── vis.py