braincog

This is a link for braincog. See the original repository https://github.com/BrainCog-X/Brain-Cog.

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.

Resources

Lecture

The current version of the lectures are in Chinese, and the English version will come soon. Stay tuned...

• [BrainCog Talk] Begining BrainCog Lecture 10. Multi-brain Areas Coordinated Brain-inspired Decision-Making Spiking Neural Network Based on Braincog
• [BrainCog Talk] Begining BrainCog Lecture 9. Spikin Neural Networks with Global Feedback Connections Based on BrainCog
• [BrainCog Talk] Begining BrainCog Lecture 8. Converting Artificial Neural Network to Spiking Neural Network through BrainCog
• [BrainCog Talk] Begining BrainCog Lecture 7. Implementing quantum superposition inspired spatio-temporal spike encoding through BrainCog
• [BrainCog Talk] Begining BrainCog Lecture 6. Implementing spiking deep Q network through Braincog
• [BrainCog Talk] Begining BrainCog Lecture 5. Advanced BrainCog System Functions
• [BrainCog Talk] Begining BrainCog Lecture 4. Creating Cognitive SNNs for Brain Areas
• [BrainCog Talk] Begining BrainCog Lecture 3. Creating SNNs Easily and Quickly
• [BrainCog Talk] Begining BrainCog Lecture 2. Computational Modeling of Spiking Neurons
• [BrainCog Talk] Begining BrainCog Lecture 1. Installing and Deploying BrainCog platform

Tutorial

• How to install braincog
• Overall introduction of BrainCog
• Tutorial of Spiking Neuron Modeling
• Building Efficient Spiking Neural Networks with BrainCog
• How to build a cognitive network

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.

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 (special)
• 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 .

Install datasets (optional)

If you use datasets in your code, especially neuromorphic datasets, you have to install another package

pip install git+https://github.com/BrainCog-X/tonic_braincog.git

You can download this package and install locally as well.

git clone https://github.com/BrainCog-X/tonic_braincog.git
cd tonic
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/