Hybrid Computing Architecture Based on Analog Phase-Change Memory Chips for Deep Neural Network Training
Deep neural networks (DNNs) have revolutionized fields like image recognition and natural language processing but face limitations with traditional von Neumann architectures due to high energy consumption and limited computing speed. We propose a hybrid architecture for DNN training combining a digital processing unit (DPU) and analog phase-change memory (PCM) chips using 40 nm CMOS technology. The DPU manages precise computations, while the PCM chip handles matrix-vector multiplication (MVM) with a novel nonlinear pulse scheme for accurate conductance tuning. Our architecture successfully trained a 3-layer fully connected neural network, achieving a classification accuracy of 97.26%, on par with software-based training. Simulations confirm the feasibility of extending this approach to more complex convolutional neural networks, demonstrating its adaptability to PCM device characteristics and potential for high-efficiency DNN training.
Authors
- Shunfen Li (ORCID: https://orcid.org/0000-0002-2464-1414)
- Houpeng Chen (ORCID: https://orcid.org/0000-0002-5766-5746)
- Weibang Dai (ORCID: https://orcid.org/0009-0009-3585-8561)
- Tao Hong (ORCID: https://orcid.org/0000-0003-2421-7290)
- Zhitang Song
- Chengcai Tu
- Xiaogang Chen (ORCID: https://orcid.org/0000-0003-1478-040X)
- Z. Jiao
Institutions
- University of Science and Technology of China (CN)
- Donghua University (CN)
- Shanghai Institute of Microsystem and Information Technology (CN)
Publication Details
- Published
- 2024-09-23
- DOI
- https://doi.org/10.1109/hpec62836.2024.10938478
- Primary Topic
- Advanced Memory and Neural Computing
- Type
- article
- Field-Weighted Citation Impact
- 0.00