Norwegian version of this page

Logic-based Artificial Intelligence Everywhere: Tsetlin Machines in Hardware

Duration
–
Project type
Research project

Project description

Tsetlin Machines (TMs) are a new machine learning (ML) approach founded on the Tsetlin Automaton. TMs use frequent pattern mining and resource allocation to extract common patterns in the data, rather than relying on minimizing output error, which is prone to overfitting. Unlike the intertwined nature of pattern representation in neural networks (NNs), TMs decompose problems into self-contained patterns, each represented as a conjunctive clause. The clause outputs, in turn, are combined into a classification decision through summation and thresholding, akin to a logistic regression function, however, with binary weights and a unit step output function. TM hardware (HW) has demonstrated up to three orders of magnitude reduced energy and faster learning, compared to NNs alike. Logic-driven fundamental blocks, organized in lean parallel processing units, are major contributors to this comparative advantage over NNs that are heavily arithmetic-based. The TM further outperforms vanilla CNNs and LSTMs accuracy-wise on well-established benchmarks. While the reported results on TMs are promising in terms of scalability, training time, accuracy, and energy, we here address three major obstacles. 1) Current FPGA and ASIC prototypes lack scalable memory elements, constraining them to small-scale ML problems. 2) Reinforcement learning (RL) is key to many ML problems, such as playing board games, however, it is unclear how to model RL in the TM framework. 3) State-of-the-art deep learning models support pre-training on unlabelled data, which significantly improves the accuracy of following supervised learning, dealing with shortage of labelled data. It is unclear how to pre-train TMs from unlabelled data. By overcoming these three obstacles we aim to architect a new TM HW/SW ecosystem that outperforms state-of-the-art ML in terms of energy efficiency and scalability, parametrised by accuracy. This will enable powerful logic-based ML applications at the edge and in the cloud.

Participants

Publications

  • Bhattarai, Bimal; Granmo, Ole-Christoffer; Jiao, Lei; Yadav, Rohan Kumar & Sharma, Jivitesh (2024). Tsetlin Machine Embedding: Representing Words Using Logical Expressions, Findings of the Association for Computational Linguistics: EACL 2024. Association for Computational Linguistics (ACL). ISSN 9798891760936. p. 1512–1520. Full text in Research Archive
  • Bhattarai, Bimal; Granmo, Ole-Christoffer & Lei, Jiao (2023). An Interpretable Knowledge Representation Framework for Natural Language Processing with Cross-Domain Application, Advances in Information Retrieval. ECIR 2023. Springer. ISSN 9783031282447. p. 167–181. doi: 10.1007/978-3-031-28244-7_11. Full text in Research Archive
  • Tunheim, Svein Anders; Lei, Jiao & Granmo, Ole-Christoffer (2023). Model Export for the Convolutional Coalesced Tsetlin Machine, 2023 International Symposium on the Tsetlin Machine (ISTM). IEEE (Institute of Electrical and Electronics Engineers). ISSN 9798350344776. doi: 10.1109/ISTM58889.2023.10455048. Full text in Research Archive
  • Przybysz, Emilia Katarzyna; Bhattarai, Bimal; Persia, Cosimo Damiano; Ozaki, Ana; Granmo, Ole-Christoffer & Sharma, Jivitesh (2023). Verifying Properties of Tsetlin Machines, 2023 International Symposium on the Tsetlin Machine (ISTM). IEEE (Institute of Electrical and Electronics Engineers). ISSN 9798350344776. doi: 10.1109/ISTM58889.2023.10454997. Full text in Research Archive
  • Abeyrathna, Kuruge Darshana; Abouzeid, Ahmed Abdulrahem Othman; Bhattarai, Bimal; Giri, Charul; Glimsdal, Sondre & Granmo, Ole-Christoffer [Show all 11 contributors for this article] (2023). Building Concise Logical Patterns by Constraining Tsetlin Machine Clause Size. In Elkind, Edith (Eds.), Proceedings of the Thirty-Second International Joint Conference on Artificial Intelligence. AAAI Press. ISSN 9781956792034. p. 3395–3403. doi: 10.24963/ijcai.2023/378. Full text in Research Archive
  • Bhattarai, Bimal; Granmo, Ole-Christoffer; Lei, Jiao; Andersen, Per-Arne; Tunheim, Svein Anders & Shafik, Rishad Ahmed [Show all 7 contributors for this article] (2023). Contracting Tsetlin Machine with Absorbing Automata, 2023 International Symposium on the Tsetlin Machine (ISTM). IEEE (Institute of Electrical and Electronics Engineers). ISSN 9798350344776. doi: 10.1109/ISTM58889.2023.10455040. Full text in Research Archive
  • Liland, Kristian Hovde; Tomic, Oliver; Indahl, Ulf Geir; Futsæther, Cecilia Marie; Lei, Jiao & Granmo, Ole-Christoffer [Show all 7 contributors for this article] (2023). Tsetlin Machine in DNA sequence classification : Application to prokaryote gene prediction / A match made in silico, 2023 International Symposium on the Tsetlin Machine (ISTM). IEEE (Institute of Electrical and Electronics Engineers). ISSN 9798350344776. doi: 10.1109/ISTM58889.2023.10454960. Full text in Research Archive
  • Sahu, Prajwal Kumar; Boppu, Srinivas; Shafik, Rishad Ahmed; Tunheim, Svein Anders; Granmo, Ole-Christoffer & Cenkeramaddi, Linga Reddy (2023). Enhancing Inference Performance through Include only Literal Incorporation in Tsetlin Machine, 2023 International Symposium on the Tsetlin Machine (ISTM). IEEE (Institute of Electrical and Electronics Engineers). ISSN 9798350344776. doi: 10.1109/ISTM58889.2023.10455016. Full text in Research Archive
  • Drøsdal, Didrik Kallhovd; Grimsmo, Andreas; Andersen, Per-Arne; Granmo, Ole-Christoffer & Goodwin, Morten (2023). Exploring the Potential of Model-Free Reinforcement Learning using Tsetlin Machines, 2023 International Symposium on the Tsetlin Machine (ISTM). IEEE (Institute of Electrical and Electronics Engineers). ISSN 9798350344776. doi: 10.1109/ISTM58889.2023.10455080. Full text in Research Archive
  • Tarasyuk, Olga; Rahman, Tousif; Shafik, Rishad Ahmed; Yakovlev, Alexandre; Gorbenko, Anatoliy & Granmo, Ole-Christoffer [Show all 7 contributors for this article] (2023). Systematic Search for Optimal Hyper-parameters of the Tsetlin Machine on MNIST Dataset, 2023 International Symposium on the Tsetlin Machine (ISTM). IEEE (Institute of Electrical and Electronics Engineers). ISSN 9798350344776. doi: 10.1109/ISTM58889.2023.10454969. Full text in Research Archive

View all works in NVA

Funding

Funded by The Research Council of Norway

Project number: 312434

Published Feb. 28, 2025 10:29 AM - Last modified Aug. 8, 2025 11:25 AM