Browse by author
Lookup NU author(s): Dr Sidharth Maheshwari, Tousif Rahman, Professor Rishad Shafik, Professor Alex Yakovlev, Dr Ashur Rafiev
This work is licensed under a Creative Commons Attribution 4.0 International License (CC BY 4.0).
AuthorInference at-the-edge using embedded machine learning models is associated with challenging trade-offs between resource metrics, such as energy and memory footprint, and the performance metrics, such as computation time and accuracy. In this work, we go beyond the conventional Neural Network based approaches to explore Tsetlin Machine (TM), an emerging machine learning algorithm, that uses learning automata to create propositional logic for classification. We use algorithm-hardware co-design to propose a novel methodology for training and inference of TM. The methodology, called REDRESS, comprises independent TM training and inference techniques to reduce the memory footprint of the resulting automata to target low and ultra-low power applications. The array of Tsetlin Automata (TA) holds learned information in the binary form as bits: ${0,1}$, called excludes and includes, respectively. REDRESS proposes a lossless TA compression method, called the include-encoding, that stores only the information associated with includes to achieve over 99% compression. This is enabled by a novel computationally minimal training procedure, called the Tsetlin Automata Re-profiling, to improve the accuracy and increase the sparsity of TA to reduce the number of includes, hence, the memory footprint. Finally, REDRESS includes an inherently bit-parallel inference algorithm that operates on the optimally trained TA in the compressed domain, that does not require decompression during runtime, to obtain high speedups when compared with the state-of-the-art Binary Neural Network (BNN) models. In this work, we demonstrate that using REDRESS approach, TM outperforms BNN models on all design metrics for five benchmark datasets viz. MNIST, CIFAR2, KWS6, Fashion-MNIST and Kuzushiji-MNIST. When implemented on an STM32F746G-DISCO microcontroller, REDRESS obtained speedups and energy savings ranging 5-5700× compared with different BNN models.
Author(s): Maheshwari S, Rahman T, Shafik R, Yakovlev A, Rafiev A, Jiao L, Granmo O
Publication type: Article
Publication status: Published
Journal: IEEE Transactions on Pattern Analysis and Machine Intelligence
Year: 2023
Volume: 45
Issue: 9
Pages: 11152-11168
Online publication date: 19/04/2023
Acceptance date: 02/04/2023
Date deposited: 09/05/2023
ISSN (print): 0162-8828
ISSN (electronic): 1939-3539
Publisher: IEEE Computer Society
URL: https://doi.org/10.1109/TPAMI.2023.3268415
DOI: 10.1109/TPAMI.2023.3268415
Altmetrics provided by Altmetric