1
|
Chen T, She C, Wang L, Duan S. Memristive leaky integrate-and-fire neuron and learnable straight-through estimator in spiking neural networks. Cogn Neurodyn 2024; 18:3075-3091. [PMID: 39555273 PMCID: PMC11564454 DOI: 10.1007/s11571-024-10133-w] [Citation(s) in RCA: 0] [Impact Index Per Article: 0] [Reference Citation Analysis] [Abstract] [Key Words] [Grants] [Track Full Text] [Journal Information] [Subscribe] [Scholar Register] [Received: 12/18/2023] [Revised: 04/28/2024] [Accepted: 05/15/2024] [Indexed: 11/19/2024] Open
Abstract
Compared to artificial neural networks (ANNs), spiking neural networks (SNNs) present a more biologically plausible model of neural system dynamics. They rely on sparse binary spikes to communicate information and operate in an asynchronous, event-driven manner. Despite the high heterogeneity of the neural system at the neuronal level, most current SNNs employ the widely used leaky integrate-and-fire (LIF) neuron model, which assumes uniform membrane-related parameters throughout the entire network. This approach hampers the expressiveness of spiking neurons and restricts the diversity of neural dynamics. In this paper, we propose replacing the resistor in the LIF model with a discrete memristor to obtain the heterogeneous memristive LIF (MLIF) model. The memristance of the discrete memristor is determined by the voltage and flux at its terminals, leading to dynamic changes in the membrane time parameter of the MLIF model. SNNs composed of MLIF neurons can not only learn synaptic weights but also adaptively change membrane time parameters according to the membrane potential of the neuron, enhancing the learning ability and expression of SNNs. Furthermore, since the proper threshold of spiking neurons can improve the information capacity of SNNs, a learnable straight-through estimator (LSTE) is proposed. The LSTE, based on the straight-through estimator (STE) surrogate function, features a learnable threshold that facilitates the backward propagation of gradients through neurons firing spikes. Extensive experiments on several popular static and neuromorphic benchmark datasets demonstrate the effectiveness of the proposed MLIF and LSTE, especially on the DVS-CIFAR10 dataset, where we achieved the top-1 accuracy of 84.40 % .
Collapse
Affiliation(s)
- Tao Chen
- College of Artificial Intelligence, Southwest University, Chongqing, 400715 China
| | - Chunyan She
- College of Artificial Intelligence, Southwest University, Chongqing, 400715 China
| | - Lidan Wang
- College of Artificial Intelligence, Southwest University, Chongqing, 400715 China
- National and Local Joint Engineering Research Center of Intelligent Transmission and Control Technology, Chongqing, 400715 China
- Chongqing Key Laboratory of Brain-inspired Computing and Intelligent Chips, Chongqing, 400715 China
| | - Shukai Duan
- College of Artificial Intelligence, Southwest University, Chongqing, 400715 China
- National and Local Joint Engineering Research Center of Intelligent Transmission and Control Technology, Chongqing, 400715 China
- Chongqing Key Laboratory of Brain-inspired Computing and Intelligent Chips, Chongqing, 400715 China
| |
Collapse
|
2
|
Tan J, Zhang F, Wu J, Luo L, Duan S, Wang L. Enhancing in-situ updates of quantized memristor neural networks: a Siamese network learning approach. Cogn Neurodyn 2024; 18:2047-2059. [PMID: 39534792 PMCID: PMC11551091 DOI: 10.1007/s11571-024-10069-1] [Citation(s) in RCA: 0] [Impact Index Per Article: 0] [Reference Citation Analysis] [Abstract] [Key Words] [Grants] [Track Full Text] [Journal Information] [Subscribe] [Scholar Register] [Received: 10/31/2023] [Revised: 12/22/2023] [Accepted: 01/16/2024] [Indexed: 11/16/2024] Open
Abstract
Brain-inspired neuromorphic computing has emerged as a promising solution to overcome the energy and speed limitations of conventional von Neumann architectures. In this context, in-memory computing utilizing memristors has gained attention as a key technology, harnessing their non-volatile characteristics to replicate synaptic behavior akin to the human brain. However, challenges arise from non-linearities, asymmetries, and device variations in memristive devices during synaptic weight updates, leading to inaccurate weight adjustments and diminished recognition accuracy. Moreover, the repetitive weight updates pose endurance challenges for these devices, adversely affecting latency and energy consumption. To address these issues, we propose a Siamese network learning approach to optimize the training of multi-level memristor neural networks. During neural inference, forward propagation takes place within the memristor neural network, enabling error and noise detection in the memristive devices and hardware circuits. Simultaneously, high-precision gradient computation occurs on the software side, initially updating the floating-point weights within the Siamese network with gradients. Subsequently, weight quantization is performed, and the memristor conductance values requiring updates are modified using a sparse update strategy. Additionally, we introduce gradient accumulation and weight quantization error compensation to further enhance network performance. The experimental results of MNIST data recognition, whether based on a MLP or a CNN model, demonstrate the rapid convergence of our network model. Moreover, our method successfully eliminates over 98% of weight updates for memristor conductance weights within a single epoch. This substantial reduction in weight updates leads to a significant decrease in energy consumption and time delay by more than 98% when compared to the basic closed-loop update method. Consequently, this approach effectively addresses the durability requirements of memristive devices.
Collapse
Affiliation(s)
- Jinpei Tan
- College of Artificial Intelligence, Southwest University, Chongqing, 400715 China
- Brain-inspired Computing & Intelligent Control of Chongqing Key Lab, Chongqing, 400715 China
| | - Fengyun Zhang
- College of Artificial Intelligence, Southwest University, Chongqing, 400715 China
| | - Jiening Wu
- College of Artificial Intelligence, Southwest University, Chongqing, 400715 China
- Brain-inspired Computing & Intelligent Control of Chongqing Key Lab, Chongqing, 400715 China
| | - Li Luo
- College of Artificial Intelligence, Southwest University, Chongqing, 400715 China
- Brain-inspired Computing & Intelligent Control of Chongqing Key Lab, Chongqing, 400715 China
| | - Shukai Duan
- College of Artificial Intelligence, Southwest University, Chongqing, 400715 China
- Brain-inspired Computing & Intelligent Control of Chongqing Key Lab, Chongqing, 400715 China
| | - Lidan Wang
- College of Artificial Intelligence, Southwest University, Chongqing, 400715 China
- Brain-inspired Computing & Intelligent Control of Chongqing Key Lab, Chongqing, 400715 China
| |
Collapse
|
3
|
Zhou G, Li J, Song Q, Wang L, Ren Z, Sun B, Hu X, Wang W, Xu G, Chen X, Cheng L, Zhou F, Duan S. Full hardware implementation of neuromorphic visual system based on multimodal optoelectronic resistive memory arrays for versatile image processing. Nat Commun 2023; 14:8489. [PMID: 38123562 PMCID: PMC10733375 DOI: 10.1038/s41467-023-43944-2] [Citation(s) in RCA: 23] [Impact Index Per Article: 11.5] [Reference Citation Analysis] [Abstract] [Key Words] [Grants] [Track Full Text] [Journal Information] [Subscribe] [Scholar Register] [Received: 06/07/2023] [Accepted: 11/22/2023] [Indexed: 12/23/2023] Open
Abstract
In-sensor and near-sensor computing are becoming the next-generation computing paradigm for high-density and low-power sensory processing. To fulfil a high-density and efficient neuromorphic visual system with fully hierarchical emulation of the retina and visual cortex, emerging multimodal neuromorphic devices for multi-stage processing and a fully hardware-implemented system with versatile image processing functions are still lacking and highly desirable. Here we demonstrate an emerging multimodal-multifunctional resistive random-access memory (RRAM) device array based on modified silk fibroin protein (MSFP), exhibiting both optoelectronic RRAM (ORRAM) mode featured by unique negative and positive photoconductance memory and electrical RRAM (ERRAM) mode featured by analogue resistive switching. A full hardware implementation of the artificial visual system with versatile image processing functions is realised for the first time, including ORRAM mode array for the in-sensor image pre-processing (contrast enhancement, background denoising, feature extraction) and ERRAM mode array for near-sensor high-level image recognition, which hugely improves the integration density, and simply the circuit design and the fabrication and integration complexity.
Collapse
Affiliation(s)
- Guangdong Zhou
- College of Artificial Intelligence, Chongqing Key Laboratory of Brain-inspired Computing and Intelligent Chips, Key Laboratory of Luminescence Analysis and Molecular Sensors (Ministry of Education), Southwest University, Chongqing, 400715, China
| | - Jie Li
- School of Microelectronics, Southern University of Science and Technology, Shenzhen, 518055, China
| | - Qunliang Song
- Faculty of Materials and Energy, Southwest University, Chongqing, 400715, China
| | - Lidan Wang
- College of Artificial Intelligence, Chongqing Key Laboratory of Brain-inspired Computing and Intelligent Chips, Key Laboratory of Luminescence Analysis and Molecular Sensors (Ministry of Education), Southwest University, Chongqing, 400715, China
| | - Zhijun Ren
- College of Artificial Intelligence, Chongqing Key Laboratory of Brain-inspired Computing and Intelligent Chips, Key Laboratory of Luminescence Analysis and Molecular Sensors (Ministry of Education), Southwest University, Chongqing, 400715, China
| | - Bai Sun
- Frontier Institute of Science and Technology, Xi'an Jiaotong University, Shanxi, 710049, China
| | - Xiaofang Hu
- College of Artificial Intelligence, Chongqing Key Laboratory of Brain-inspired Computing and Intelligent Chips, Key Laboratory of Luminescence Analysis and Molecular Sensors (Ministry of Education), Southwest University, Chongqing, 400715, China
| | - Wenhua Wang
- Faculty of Materials and Energy, Southwest University, Chongqing, 400715, China
| | - Gaobo Xu
- Faculty of Materials and Energy, Southwest University, Chongqing, 400715, China
| | - Xiaodie Chen
- Department of Electrical and Electronic Engineering, The University of Hong Kong, Hong Kong, 999077, China
| | - Lan Cheng
- State Key Laboratory of Silkworm Genome, College of Sericulture, Textile and Biomass Sciences, Southwest University, Chongqing, 400715, China
| | - Feichi Zhou
- School of Microelectronics, Southern University of Science and Technology, Shenzhen, 518055, China.
| | - Shukai Duan
- College of Artificial Intelligence, Chongqing Key Laboratory of Brain-inspired Computing and Intelligent Chips, Key Laboratory of Luminescence Analysis and Molecular Sensors (Ministry of Education), Southwest University, Chongqing, 400715, China.
| |
Collapse
|
4
|
Kim M, Rehman MA, Lee D, Wang Y, Lim DH, Khan MF, Choi H, Shao QY, Suh J, Lee HS, Park HH. Filamentary and Interface-Type Memristors Based on Tantalum Oxide for Energy-Efficient Neuromorphic Hardware. ACS APPLIED MATERIALS & INTERFACES 2022; 14:44561-44571. [PMID: 36164762 DOI: 10.1021/acsami.2c12296] [Citation(s) in RCA: 20] [Impact Index Per Article: 6.7] [Reference Citation Analysis] [Abstract] [Key Words] [Track Full Text] [Subscribe] [Scholar Register] [Indexed: 06/16/2023]
Abstract
To implement artificial neural networks (ANNs) based on memristor devices, it is essential to secure the linearity and symmetry in weight update characteristics of the memristor, and reliability in the cycle-to-cycle and device-to-device variations. This study experimentally demonstrated and compared the filamentary and interface-type resistive switching (RS) behaviors of tantalum oxide (Ta2O5 and TaO2)-based devices grown by atomic layer deposition (ALD) to propose a suitable RS type in terms of reliability and weight update characteristics. Although Ta2O5 is a strong candidate for memristor, the filament-type RS behavior of Ta2O5 does not fit well with ANNs demanding analog memory characteristics. Therefore, this study newly designed an interface-type TaO2 memristor and compared it to a filament type of Ta2O5 memristor to secure the weight update characteristics and reliability. The TaO2-based interface-type memristor exhibited gradual RS characteristics and area dependency in both high- and low-resistance states. In addition, compared to the filamentary memristor, the RS behaviors of the TaO2-based interface-type device exhibited higher suitability for the neuromorphic, symmetric, and linear long-term potentiation (LTP) and long-term depression (LTD). These findings suggest better types of memristors for implementing ionic memristor-based ANNs among the two types of RS mechanisms.
Collapse
Affiliation(s)
- Minjae Kim
- Department of Materials Science and Engineering, Yonsei University, Seoul 03722, South Korea
| | - Malik Abdul Rehman
- Department of Materials Science and Engineering, Yonsei University, Seoul 03722, South Korea
| | - Donghyun Lee
- Department of Materials Science and Engineering, Ulsan National Institute of Science and Technology, Ulsan 44919, Korea
| | - Yue Wang
- Department of Materials Science and Engineering, Yonsei University, Seoul 03722, South Korea
| | - Dong-Hyeok Lim
- Department of Materials Science and Engineering, Ulsan National Institute of Science and Technology, Ulsan 44919, Korea
| | - Muhammad Farooq Khan
- Department of Electrical Engineering, Sejong University, Seoul 05006, South Korea
| | - Haryeong Choi
- Department of Materials Science and Engineering, Yonsei University, Seoul 03722, South Korea
| | - Qing Yi Shao
- Provincial Key Laboratory of Nuclear Science, Institute of Quantum Matter, South China Normal University, Guangzhou 510006, China
| | - Joonki Suh
- Department of Materials Science and Engineering, Ulsan National Institute of Science and Technology, Ulsan 44919, Korea
| | - Hong-Sub Lee
- Department of Advanced Materials Engineering for Information and Electronics, Kyung Hee University, Yongin, Gyeonggi-do 17104, Korea
| | - Hyung-Ho Park
- Department of Materials Science and Engineering, Yonsei University, Seoul 03722, South Korea
| |
Collapse
|