loading page

BITLITE: Light Bit-wise Operative Vector Matrix Multiplication for Low-Resolution Platforms
  • +2
  • Vince Tran,
  • Demeng Chen,
  • Roman Genov,
  • Mostafa Rahimi Azghadi,
  • Amirali Amirsoleimani
Vince Tran
Department of Electrical and Computer Engineering, University of Toronto
Demeng Chen
Department of Electrical and Computer Engineering, University of Toronto
Roman Genov
Department of Electrical and Computer Engineering, University of Toronto
Mostafa Rahimi Azghadi
College of Science and Engineering, James Cook University
Amirali Amirsoleimani
Department of Electrical Engineering and Computer Science, York University

Corresponding Author:[email protected]

Author Profile

Abstract

As machine learning (ML) algorithms, particularly neural networks (NN), expand in popularity and capacity, the quest for more efficient computation methods gains momentum. Memristor crossbar technology emerges as a promising alternative to traditional computing units, aiming to address traditional computing challenges. However, conventional matrixvector multiplication (MVM) methods on these platforms are often plagued by device imperfections and drift. In this work, we introduce an innovative lightweight calculation approach leveraging bit-transformation for MVM, significantly enhancing operation precision and, consequently, the performance of ML algorithms on memristor crossbar platforms. We provide details of the core algorithm and its extensions, furnish digital validation, and simulate its efficacy using an autoencoder (AE) neural network with an extended VTEAM model. Our tests demonstrate an average reconstruction precision improvement of approximately 53.5%. This work's applicability extends beyond NNs, offering a foundational method for conducting more precise analog MVM operations.