braincog 0.2.7.19

BrainCog is an open source spiking neural network based brain-inspired cognitive intelligence engine for Brain-inspired Artificial Intelligence and brain simulation. More information on braincog can be found on its homepage http://www.brain-cog.network/

BrainCog

BrainCog is an open source spiking neural network based brain-inspired cognitive intelligence engine for Brain-inspired Artificial Intelligence and brain simulation. More information on BrainCog can be found on its homepage http://www.brain-cog.network/

The current version of BrainCog contains at least 18 functional spiking neural network algorithms (including but not limited to perception and learning, decision making, knowledge representation and reasoning, motor control, social cognition, etc.) built based on BrainCog infrastructures, and BrainCog also provide brain simulations to drosophila, rodent, monkey, and human brains at multiple scales based on spiking neural networks at multiple scales. More detail in http://www.brain-cog.network/docs/

BrainCog is a community based effort for spiking neural network based artificial intelligence, and we welcome any forms of contributions, from contributing to the development of core components, to contributing for applications.

If you use BrainCog in your research, the following paper can be cited as the source for BrainCog.

Yi Zeng, Dongcheng Zhao, Feifei Zhao, Guobin Shen, Yiting Dong, Enmeng Lu, Qian Zhang, Yinqian Sun, Qian Liang, Yuxuan Zhao, Zhuoya Zhao, Hongjian Fang, Yuwei Wang, Yang Li, Xin Liu, Chengcheng Du, Qingqun Kong, Zizhe Ruan, Weida Bi. BrainCog: A Spiking Neural Network based Brain-inspired Cognitive Intelligence Engine for Brain-inspired AI and Brain Simulation. arXiv:2207.08533, 2022. https://arxiv.org/abs/2207.08533

BrainCog provides essential and fundamental components to model biological and artificial intelligence.

Brain-Inspired AI

BrainCog currently provides cognitive functions components that can be classified into five categories:

• Perception and Learning
• Decision Making
• Motor Control
• Knowledge Representation and Reasoning
• Social Cognition

Brain Simulation

BrainCog currently include two parts for brain simulation:

• Brain Cognitive Function Simulation
• Multi-scale Brain Structure Simulation

The anatomical and imaging data is used to support our simulation from various aspects.

Resources

Lectures

• [BrainCog Talk] Beginning BrainCog Lecture 23. The Implement of Object Detection and Semantic Segmentation Based on SNNs with Braincog [English Version, Chinese Version]
• [BrainCog Talk] Beginning BrainCog Lecture 22. BrainCog Data Engine: Spatio-temporal Sequence Data N-Omniglot and Its Applications [English Version, Chinese Version]
• [BrainCog Talk] Beginning BrainCog Lecture 21. Dynamic structural development for SNNs incorporating constraints, pruning and regeneration based on BrainCog [English Version, Chinese Version]
• [BrainCog Talk] Beginning BrainCog Lecture 20. Developmental Plasticity-inspired Adaptive Pruning for SNNs based on BrainCog [English Version, Chinese Version]
• [BrainCog Talk] Beginning BrainCog Lecture 19. Multi-brain areas Coordinated Brain-inspired Affective Empathy Spiking Neural Network Based on Braincog [English Version, Chinese Version]
• [BrainCog Talk] Beginning BrainCog Lecture 18. Application of the Prefrontal Cortex Column Model in Working Memory Task with BrainCog [English Version, Chinese Version]
• [BrainCog Talk] Beginning BrainCog Lecture 17. A Brain-inspired Theory of Mind Model Based on BrainCog for Reducing Other Agents’ Safety Risks [English Version, Chinese Version]
• [BrainCog Talk] Beginning BrainCog Lecture 16. Brain-inspired Bodily Self-perception Model Based on BrainCog [English Version, Chinese Version]
• [BrainCog Talk] Beginning BrainCog Lecture 15. SNN-based Music Memory and Generation Based on BrainCog [English Version, Chinese Version]
• [BrainCog Talk] Beginning BrainCog Lecture 14. The Implement of Multisensory Concept Learning Framework Based on SNNs with Braincog [English Version, Chinese Version]
• [BrainCog Talk] Beginning BrainCog Lecture 13. Symbolic Representation and Reasoning SNN Based on Braincog [English Version, Chinese Version]
• [BrainCog Talk] Beginning BrainCog Lecture 12. Unsupervised STDP-based Spiking Neural Networks Based on BrainCog [English Version, Chinese Version]
• [BrainCog Talk] Beginning BrainCog Lecture 11. Backpropagation with Spatiotemporal Adjustment for Training Deep Spiking Neural Networks through BrainCog [English Version, Chinese Version]
• [BrainCog Talk] Beginning BrainCog Lecture 10. Multi-brain Areas Coordinated Brain-inspired Decision-Making Spiking Neural Network Based on Braincog [English Version, Chinese Version]
• [BrainCog Talk] Beginning BrainCog Lecture 9. Spiking Neural Networks with Global Feedback Connections Based on BrainCog [English Version, Chinese Version]
• [BrainCog Talk] Beginning BrainCog Lecture 8. Converting Artificial Neural Network to Spiking Neural Network through BrainCog [English Version, Chinese Version]
• [BrainCog Talk] Beginning BrainCog Lecture 7. Implementing Quantum Superposition Inspired Spatio-temporal Spike Encoding through BrainCog [English Version, Chinese Version]
• [BrainCog Talk] Beginning BrainCog Lecture 6. Implementing spiking deep Q network through Braincog [English Version, Chinese Version]
• [BrainCog Talk] Beginning BrainCog Lecture 5. Advanced BrainCog System Functions [English Version, Chinese Version]
• [BrainCog Talk] Beginning BrainCog Lecture 4. Creating Cognitive SNNs for Brain Areas [English Version, Chinese Version]
• [BrainCog Talk] Beginning BrainCog Lecture 3. Creating SNNs Easily and Quickly [English Version, Chinese Version]
• [BrainCog Talk] Beginning BrainCog Lecture 2. Computational Modeling of Spiking Neurons [English Version, Chinese Version]
• [BrainCog Talk] Beginning BrainCog Lecture 1. Installing and Deploying BrainCog platform [English Version, Chinese Version]

Tutorial

• How to Install BrainCog [English Version, Chinese Version]
• Overall Introduction of BrainCog [English Version, Chinese Version]
• Spiking Neuron Modeling with BrainCog [English Version, Chinese Version]
• Building Efficient Spiking Neural Networks with BrainCog [English Version, Chinese Version]
• Building Cognitive Networks with BrainCog [English Version, Chinese Version]
• Implementing Deep Reinforcement Learning SNNs with BrainCog [English Version, Chinese Version]
• Quantum Superposition State Inspired Encoding With BrainCog [English Version, Chinese Version]
• Converting ANNs to SNNs with BrainCog [English Version, Chinese Version]
• SNNs with Global Feedback Connections Based on BrainCog [English Version, Chinese Version]
• Multi-brain Areas Coordinated Brain-inspired Decision-Making SNNs Based on Braincog [English Version, Chinese Version]
• Backpropagation with Spatiotemporal Adjustment for Training SNNs Based on BrainCog [English Version, Chinese Version]
• Unsupervised STDP Based SNNs with Multiple Adaptive Mechanisms Based on BrainCog [English Version, Chinese Version]
• Symbolic Representation and Reasoning SNNs Based on BrainCog [English Version, Chinese Version]
• Multisensory Integration Based on BrainCog [English Version, Chinese Version]
• SNN-based Music Memory and Generation Based on BrainCog [English Version, Chinese Version]
• Brain-inspired Bodily Self-perception Model Based on BrainCog [English Version, Chinese Version]
• A Brain-inspired Theory of Mind Model Based on BrainCog for Reducing Other Agents’ Safety Risks [English Version, Chinese Version]
• Application of the Prefrontal Cortex Column Model in Working Memory Task with BrainCog [English Version, Chinese Version]
• Multi-brain areas Coordinated Brain-inspired Affective Empathy Spiking Neural Network Based on Braincog [English Version, Chinese Version]
• Developmental Plasticity-inspired Adaptive Pruning for SNNs based on BrainCog [English Version, Chinese Version]
• Dynamic structural development for SNNs incorporating constraints, pruning and regeneration based on BrainCog [English Version, Chinese Version]
• BrainCog Data Engine: Spatio-temporal Sequence Data N-Omniglot and Its Applications [English Version, Chinese Version]
• The Implement of Object Detection and Semantic Segmentation Based on SNNs with Braincog [English Version, Chinese Version]

BrainCog Data Engine

In addition to the static datasets, BrainCog supports the commonly used neuromorphic datasets, such as DVSGesture, DVSCIFAR10, NCALTECH101, ES-ImageNet. Also, the neuromorphic dataset N-Omniglot for few-shot learning is also integrated into BrainCog.

DVSGesture

This dataset contains 11 hand gestures from 29 subjects under 3 illumination conditions recorded using a DVS128.

DVSCIFAR10

This dataset converts 10,000 frame-based images in the CIFAR10 dataset into 10,000 event streams using a dynamic vision sensor.

NCALTECH101

The NCaltech101 dataset is captured by mounting the ATIS sensor on a motorized pan-tilt unit and having the sensor move while it views Caltech101 examples on an LCD monitor. The "Faces" class has been removed from N-Caltech101, leaving 100 object classes plus a background class

ES-ImageNet

The dataset is converted with Omnidirectional Discrete Gradient (ODG) from 1,300,000 frame-based images in the ImageNet dataset into event-stream samples, which has 1000 categories.

N-Omniglot

This dataset contains 1,623 categories of handwritten characters, with only 20 samples per class. The dataset is acquired with the DVS acquisition platform to shoot videos (generated from the original Omniglot dataset) played on the monitor, and use the Robotic Process Automation (RPA) software to collect the data automatically.

You can easily use them in the braincog/datasets folder, taking DVSCIFAR10 as an example

loader_train, loader_eval,_,_ = get_dvsc10_data(batch_size=128,step=10)

Requirements:

• python == 3.8
• CUDA toolkit == 11.
• numpy >= 1.21.2
• scipy >= 1.8.0
• h5py >= 3.6.0
• torch >= 1.10
• torchvision >= 0.12.0
• torchaudio >= 0.11.0
• timm >= 0.5.4
• matplotlib >= 3.5.1
• einops >= 0.4.1
• thop >= 0.0.31
• pyyaml >= 6.0
• loris >= 0.5.3
• pandas >= 1.4.2
• tonic
• pandas >= 1.4.2
• xlrd == 1.2.0

Install

Install Online

1. You can install braincog by running:

   pip install braincog

2. Also, install from github by running:

   pip install git+https://github.com/braincog-X/Brain-Cog.git

Install locally

1. If you are a developer, it is recommanded to download or clone braincog from github.

   git clone https://github.com/braincog-X/Brain-Cog.git

2. Enter the folder of braincog

   cd Brain-Cog

3. Install braincog locally

   pip install -e .

Example

1. Examples for Image Classification

cd ./examples/Perception_and_Learning/img_cls/bp 
python main.py --model cifar_convnet --dataset cifar10 --node-type LIFNode --step 8 --device 0

2. Examples for Event Classification

cd ./examples/Perception_and_Learning/img_cls/bp 
python main.py --model dvs_convnet --node-type LIFNode --dataset dvsc10 --step 10 --batch-size 128 --act-fun QGateGrad --device 0 

Other BrainCog features and tutorials can be found at http://www.brain-cog.network/docs/

BrainCog Assistant

Please add our BrainCog Assitant via wechat and we will invite you to our wechat developer group.

Publications Using BrainCog

Brain Inspired AI

Perception and Leanring

Papers	Codes	Publisher
Quantum superposition inspired spiking neural network	https://github.com/BrainCog-X/Brain-Cog/tree/main/examples/Perception_and_Learning/QSNN	Cell iScience
Backpropagation with biologically plausible spatiotemporal adjustment for training deep spiking neural networks	https://github.com/BrainCog-X/Brain-Cog/tree/main/examples/Perception_and_Learning/img_cls/bp	Cell Patterns
N-Omniglot, a large-scale neuromorphic dataset for spatio-temporal sparse few-shot learning	https://github.com/BrainCog-X/Brain-Cog/tree/main/braincog/datasets/NOmniglot	Scientific Data
Efficient and Accurate Conversion of Spiking Neural Network with Burst Spikes	https://github.com/BrainCog-X/Brain-Cog/blob/main/examples/Perception_and_Learning/Conversion/converted_CIFAR10.py	IJCAI 2022
BackEISNN: A deep spiking neural network with adaptive self-feedback and balanced excitatory–inhibitory neurons	https://github.com/BrainCog-X/Brain-Cog/blob/main/examples/Perception_and_Learning/img_cls/bp/main_backei.py	Neural Networks
Spiking CapsNet: A spiking neural network with a biologically plausible routing rule between capsules	https://github.com/BrainCog-X/Brain-Cog/tree/main/examples/Perception_and_Learning/img_cls/spiking_capsnet	Information Sciences
Multisensory Concept Learning Framework Based on Spiking Neural Networks	https://github.com/BrainCog-X/Brain-Cog/tree/main/examples/Perception_and_Learning/MultisensoryIntegration	Frontiers in Systems Neuroscience
GLSNN: A Multi-Layer Spiking Neural Network Based on Global Feedback Alignment and Local STDP Plasticity	https://github.com/BrainCog-X/Brain-Cog/tree/main/examples/Perception_and_Learning/img_cls/glsnn	Fontiers in Computational Neuroscience
EventMix: An Efficient Augmentation Strategy for Event-Based Data	https://github.com/BrainCog-X/Brain-Cog/blob/main/braincog/datasets/cut_mix.py	Arxiv
Spike Calibration: Fast and Accurate Conversion of Spiking Neural Network for Object Detection and Segmentation	https://github.com/BrainCog-X/Brain-Cog/blob/main/examples/Perception_and_Learning/Conversion/converted_CIFAR10.py	Arxiv
An Unsupervised Spiking Neural Network Inspired By Biologically Plausible Learning Rules and Connections	https://github.com/BrainCog-X/Brain-Cog/tree/main/examples/Perception_and_Learning/UnsupervisedSTDP	Arxiv

Social Cognition

Papers	Codes	Publisher
Toward Robot Self-Consciousness (II): Brain-Inspired Robot Bodily Self Model for Self-Recognition	https://github.com/BrainCog-X/Brain-Cog/tree/main/examples/Social_Cognition/mirror_test	Cognitive Computation
A brain-inspired intention prediction model and its applications to humanoid robot	https://github.com/BrainCog-X/Brain-Cog/tree/main/examples/Social_Cognition/Intention_Prediction	Frontiers in Neuroscience
A Brain-Inspired Theory of Mind Spiking Neural Network for Reducing Safety Risks of Other Agents	https://github.com/BrainCog-X/Brain-Cog/tree/main/examples/Social_Cognition/ToM	Frontiers in Neuroscience
Brain-Inspired Affective Empathy Computational Model and Its Application on Altruistic Rescue Task	https://github.com/BrainCog-X/Brain-Cog/tree/main/examples/Social_Cognition/affective_empathy/BAE-SNN	Frontiers in Computational Neuroscience
A brain-inspired robot pain model based on a spiking neural network	https://github.com/BrainCog-X/Brain-Cog/tree/main/examples/Social_Cognition/affective_empathy/BRP-SNN	Frontiers in Neurorobotics
Brain-Inspired Theory of Mind Spiking Neural Network Elevates Multi-Agent Cooperation and Competition	https://github.com/BrainCog-X/Brain-Cog/tree/main/examples/Social_Cognition/MAToM-SNN	SSRN

Knowledge Representation and Reasoning

Papers	Codes	Publisher
A Brain-Inspired Causal Reasoning Model Based on Spiking Neural Networks	https://github.com/BrainCog-X/Brain-Cog/tree/main/examples/Knowledge_Representation_and_Reasoning/CRSNN	IJCNN2021
Brain Inspired Sequences Production by Spiking Neural Networks With Reward-Modulated STDP	https://github.com/BrainCog-X/Brain-Cog/tree/main/examples/Knowledge_Representation_and_Reasoning/SPSNN	Frontiers in Computational Neuroscience
Temporal-Sequential Learning With a Brain-Inspired Spiking Neural Network and Its Application to Musical Memory	https://github.com/BrainCog-X/Brain-Cog/tree/main/examples/Knowledge_Representation_and_Reasoning/musicMemory	Frontiers in Computational Neuroscience
Brain-Inspired Affective Empathy Computational Model and Its Application on Altruistic Rescue Task	https://github.com/BrainCog-X/Brain-Cog/tree/main/examples/Knowledge_Representation_and_Reasoning/musicMemory	Frontiers in System Neuroscience
Stylistic Composition of Melodies Based on a Brain-Inspired Spiking Neural Network	https://github.com/BrainCog-X/Brain-Cog/tree/main/examples/Social_Cognition/affective_empathy/BRP-SNN	Frontiers in Neurorobotics
Brain-inspired Graph Spiking Neural Networks for Commonsense Knowledge Representation and Reasoning	https://github.com/BrainCog-X/Brain-Cog/tree/main/examples/Knowledge_Representation_and_Reasoning/CKRGSNN	Arxiv

Decision Making

Papers	Codes	Publisher
Nature-inspired self-organizing collision avoidance for drone swarm based on reward-modulated spiking neural network	https://github.com/BrainCog-X/Brain-Cog/blob/main/examples/decision_making/swarm/Collision-Avoidance.py	Cell Patterns
Solving the spike feature information vanishing problem in spiking deep Q network with potential based normalization	https://github.com/BrainCog-X/Brain-Cog/tree/main/examples/decision_making/RL/sdqn	Frontiers in Neuroscience
A Brain-Inspired Decision-Making Spiking Neural Network and Its Application in Unmanned Aerial Vehicle	https://github.com/BrainCog-X/Brain-Cog/blob/main/examples/decision_making/BDM-SNN/BDM-SNN-hh.py	Frontiers in Neurorobotics
Multi-compartment Neuron and Population Encoding improved Spiking Neural Network for Deep Distributional Reinforcement Learning	https://github.com/BrainCog-X/Brain-Cog/tree/main/examples/decision_making/RL/mcs-fqf	Arxiv

Motor Control

Papers	Codes	Publisher
	https://github.com/BrainCog-X/Brain-Cog/tree/main/examples/MotorControl/experimental

SNN Safety

Papers	Codes	Publisher
DPSNN	https://github.com/BrainCog-X/Brain-Cog/tree/main/examples/Snn_safety/DPSNN	Arxiv

Development and Evolution

Papers	Codes	Publisher
Developmental Plasticity-inspired Adaptive Pruning for Deep Spiking and Artificial Neural Networks	https://github.com/BrainCog-X/Brain-Cog/tree/main/examples/Structural_Development/DPAP	Arxiv
Adaptive Sparse Structure Development with Pruning and Regeneration for Spiking Neural Networks	https://github.com/BrainCog-X/Brain-Cog/tree/main/examples/Structural_Development/SD-SNN	Arxiv

Hardware Acceleration

Papers	Codes	Publisher
FireFly: A High-Throughput and Reconfigurable Hardware Accelerator for Spiking Neural Networks	https://github.com/BrainCog-X/Brain-Cog/tree/main/examples/Hardware_acceleration	Arxiv

Brain Simulation

Funtion

Papers	Codes	Publisher
A neural algorithm for Drosophila linear and nonlinear decision-making	https://github.com/BrainCog-X/Brain-Cog/blob/main/examples/Brain_Cognitive_Function_Simulation/drosophila/drosophila.py	Scientific Reports
Comparison Between Human and Rodent Neurons for Persistent Activity Performance: A Biologically Plausible Computational Investigation	https://github.com/BrainCog-X/Brain-Cog/tree/main/examples/Multiscale_Brain_Structure_Simulation/Human_PFC_Model	Frontiers in System Neuroscience

Structure

Papers	Codes	Publisher
Corticothalamic minicolumn	https://github.com/BrainCog-X/Brain-Cog/tree/main/examples/Multiscale_Brain_Structure_Simulation/CorticothalamicColumn
Human Brain	https://github.com/BrainCog-X/Brain-Cog/tree/main/examples/Multiscale_Brain_Structure_Simulation/HumanBrain
Macaque Brain	https://github.com/BrainCog-X/Brain-Cog/tree/main/examples/Multiscale_Brain_Structure_Simulation/MacaqueBrain
Mouse Brain	https://github.com/BrainCog-X/Brain-Cog/tree/main/examples/Multiscale_Brain_Structure_Simulation/Mouse_brain