Design of a Highly Flexible Hybrid Neural Network Inference Platform with 10 Million Neurons
A large-scale neural network platform that explores brain functions while deploying artificial networks has become an urgent research objective. However, large-scale platforms struggle to balance versatility, flexibility, and performance. This paper proposes a hybrid network platform comprising 1280 cores, supporting the deployment of 10 million neurons and 80 billion synapses. Functionally, the cores utilize a hybrid neuron model, enabling neuromorphic and traditional artificial intelligence computation and supporting the mapping of many structures, especially the Transformer. In terms of performance, the proposed architecture features a bandwidth-matching design, employing near-memory computing and multi-channel high-speed interfaces to alleviate bottlenecks, achieving a 9.0 times speed improvement compared to off-chip computation. For flexibility, the architecture uses flexible core cluster designs, incorporating variable-scale deployment and intra-cluster fusion, providing up to a 7.1 times increase in computational utilization and a maximum 341 times reduction in latency for the same network deployment. The architecture successfully deployed multiple networks for recognition on an emulator. This architecture supports flexible hybrid inference, providing comprehensive hardware conditions for deploying large-scale neural networks.
Authors
- Kai Cao (ORCID: https://orcid.org/0000-0003-2468-0381)
- Shiao Wang (ORCID: https://orcid.org/0009-0001-6149-7503)
- Guanchao Qiao (ORCID: https://orcid.org/0000-0003-4982-5938)
- Ye Liu (ORCID: https://orcid.org/0000-0002-2686-3002)
- Rui Zhang (ORCID: https://orcid.org/0000-0002-8132-6250)
- Ning Ning (ORCID: https://orcid.org/0000-0001-7893-1428)
- Yue Zuo (ORCID: https://orcid.org/0000-0001-5791-3634)
- Lin Meng (ORCID: https://orcid.org/0009-0002-0842-1224)
- S. G. Hu
Institutions
- University of Electronic Science and Technology of China (CN)
- National Engineering Research Center of Electromagnetic Radiation Control Materials (CN)
- State Key Laboratory of Electronic Thin Films and Integrated Devices
Publication Details
- Published
- 2024-10-24
- DOI
- https://doi.org/10.1109/biocas61083.2024.10798124
- Citations
- 1
- Primary Topic
- Neural Networks and Applications
- Type
- article
- Field-Weighted Citation Impact
- 0.35