International Journal of Computer Applications |
Foundation of Computer Science (FCS), NY, USA |
Volume 186 - Number 49 |
Year of Publication: 2024 |
Authors: Wei Zhang, Qingtian Zhang, Huaqiang Wu |
10.5120/ijca2024924139 |
Wei Zhang, Qingtian Zhang, Huaqiang Wu . Genetic Algorithm-based Training Method for Memristor Neural Networks. International Journal of Computer Applications. 186, 49 ( Nov 2024), 7-13. DOI=10.5120/ijca2024924139
In recent years, deep learning and large models have significantly advanced artificial intelligence applications in areas such as natural language processing and computer vision. However, as model scales grow, the demand for computing power increases, revealing the limitations of traditional von Neumann architectures. Memristor-based in-memory computing offers a promising alternative, yet neural networks deployed on memristor arrays suffer from accuracy loss due to device non-idealities. To address this, authors introduce a novel genetic algorithm (GA)- based training methodology specifically designed for memristor arrays to enhance neural network performance. authors detail the framework and strategic operations of this approach, supported by empirical validation using a series of lightweight models and demonstrate substantial accuracy improvements. Additionally, authors explore the impact of various hyperparameter settings on model precision. Overall, this approach significantly enhances the accuracy of lightweight neural networks on memristor arrays, with important implications for edge computing environments.