/ToM

A Brain-Inspired Theory of Mind Spiking Neural Network for Reducing Safety Risks of Other Agents

Primary LanguagePython

ToM

This code accompanies the paper the paper "A Brain-Inspired Theory of Mind Spiking Neural Network for Reducing Safety Risks of Other Agents"

Requirments

  • numpy
  • scipy
  • pytorch >= 1.7.0
  • torchvision
  • pygame

Run

Train

  • the file to be run: main_both.py
  • args:
    • the path to save net_NPC: --save_net_N
    • the path to save net_a: --save_net_a
    • time steps: --T
python main_both.py --save_net_N=net_NPC.pth --save_net_a=net_agent.pth --episodes=45 --trajectories=30 --T=50 --mode=train --task=both

Test

You can use the weigts saved by taining in the test environment.

python main_ToM.py --save_net_N=net_NPC.pth --save_net_a=net_agent.pth --episodes=45 --trajectories=30 --T=50 --mode=train --task=both

Citation

@article{zhao2022brain,
    title     = {A Brain-Inspired Theory of Mind Spiking Neural Network for Reducing Safety Risks of Other Agents},
    author    = {Zhao, Zhuoya and Lu, Enmeng and Zhao, Feifei and Zeng, Yi and Zhao, Yuxuan},
    journal   = {Frontiers in neuroscience},
    pages     = {446},
    year      = {2022},
    publisher = {Frontiers}
}