Manhattan Rule for Robust In-Situ Training of Memristive Deep Neural Network Accelerators
In this work, we propose an alternative training approach for memristive circuits - the Manhattan rule training - which utilizes only sign information for weight updates. We present an in-depth analysis in both in-situ and ex-situ settings and show that not only does our method simplify circuit design but it also improves neural network robustness against device non-idealities. Using the MemTorch and our custom in-situ training framework, we implemented the Manhattan rule for MNIST classification and ECG signal detection tasks and achieved close to state-of-the-art performance under noise. Our work also provides a thorough comparison of Manhattan and conventional training methods under the effects of various device non-idealities, giving a crucial benchmark useful for the design of biomedical neural circuits.
Authors
- Mostafa Rahimi Azghadi (ORCID: https://orcid.org/0000-0001-7975-3985)
- Amirali Amirsoleimani (ORCID: https://orcid.org/0000-0001-5760-6861)
- Majid Ahmadi (ORCID: https://orcid.org/0000-0001-5781-6754)
- Jack Cai (ORCID: https://orcid.org/0009-0008-1892-1875)
- Roman Genov
- Ellina Zhang
Institutions
- University of Toronto (CA)
- York University (CA)
- University of Windsor (CA)
- James Cook University (AU)
Publication Details
- Published
- 2024-08-11
- DOI
- https://doi.org/10.1109/mwscas60917.2024.10658688
- Primary Topic
- Advanced Memory and Neural Computing
- Type
- article
- Field-Weighted Citation Impact
- 0.00