Tsetlin machine

Short description: Artificial intelligence algorithm

A simple block diagram of the Tsetlin machine

A Tsetlin machine is an Artificial Intelligence algorithm based on propositional logic.

Background

A Tsetlin machine is a form of learning automaton collective for learning patterns using propositional logic. Ole-Christoffer Granmo gave the method its name after Michael Lvovitch Tsetlin, who invented the Tsetlin automaton and worked on Tsetlin automata collectives and games. Collectives of Tsetlin automata were originally constructed, implemented, and studied theoretically by Vadim Stefanuk in 1962.

The Tsetlin machine uses computationally simpler and more efficient primitives compared to more ordinary artificial neural networks.

As of April 2018 it has shown promising results on a number of test sets.

Types

• Original Tsetlin machine
• Convolutional Tsetlin machine
• Regression Tsetlin machine
• Relational Tsetlin machine
• Weighted Tsetlin machine
• Arbitrarily deterministic Tsetlin machine
• Parallel asynchronous Tsetlin machine
• Coalesced multi-output Tsetlin machine
• Tsetlin machine for contextual bandit problems
• Tsetlin machine autoencoder
• Tsetlin machine composites: plug-and-play collaboration between specialized Tsetlin machines

Applications

• Keyword spotting
• Aspect-based sentiment analysis
• Word-sense disambiguation
• Novelty detection
• Intrusion detection
• Semantic relation analysis
• Image analysis
• Text categorization
• Fake news detection
• Game playing
• Batteryless sensing
• Recommendation systems
• Word embedding
• ECG analysis
• Edge computing
• Bayesian network learning

Original Tsetlin machine

A detailed block diagram of the original Tsetlin machine

List of hyperparameters

Description	Symbol
Number of binary inputs	[math]\displaystyle{ N_\text{Inputs} }[/math]
Number of classes	[math]\displaystyle{ N_\text{Classes} }[/math]
Number of clauses per class	[math]\displaystyle{ N_\text{Clauses} }[/math]
Number of automaton states	[math]\displaystyle{ 2n }[/math]
Automaton decision boundary	n
Automaton initialization state	[math]\displaystyle{ \varnothing_\text{Init} }[/math]
Feedback threshold	T
Learning sensitivity	s

Tsetlin automaton

The Tsetlin automaton is the fundamental learning unit of the Tsetlin machine. It tackles the multi-armed bandit problem, learning the optimal action in an environment from penalties and rewards. Computationally, it can be seen as a finite-state machine (FSM) that changes its states based on the inputs. The FSM will generate its outputs based on the current states.

• A quintuple describes a two-action Tsetlin automaton:

[math]\displaystyle{ \{\underline{\Phi}, \underline{\alpha}, \underline{\beta}, F(\cdot,\cdot), G(\cdot)\}. }[/math]

• A Tsetlin automaton has [math]\displaystyle{ 2n }[/math] states, here 6:

[math]\displaystyle{ \underline{\Phi} = \{\phi_1, \phi_2, \phi_3, \phi_4, \phi_5, \phi_6\} }[/math]

• The FSM can be triggered by two input events

[math]\displaystyle{ \underline{\beta} = \{\beta_{\mathrm{Penalty}}, \beta_{\mathrm{Reward}}\} }[/math]

• The rules of state migration of the FSM are stated as

[math]\displaystyle{ F(\phi_u, \beta_v) = \begin{cases} \phi_{u+1},& \text{if}~ 1 \le u \le 3 ~\text{and}~ v = \text{Penalty}\\ \phi_{u-1},& \text{if}~ 4 \le u \le 6 ~\text{and}~ v = \text{Penalty}\\ \phi_{u-1},& \text{if}~ 1 \lt u \le 3 ~\text{and}~ v = \text{Reward}\\ \phi_{u+1},& \text{if}~ 4 \le u \lt 6 ~\text{and}~ v = \text{Reward}\\ \phi_{u},& \text{otherwise}. \end{cases} }[/math]

• It includes two output actions

[math]\displaystyle{ \underline{\alpha} = \{\alpha_1, \alpha_2\} }[/math]

• Which can be generated by the algorithm

[math]\displaystyle{ G(\phi_u) = \begin{cases} \alpha_1, & \text{if}~ 1 \le u \le 3\\ \alpha_2, & \text{if}~ 4 \le u \le 6. \end{cases} }[/math]

Boolean input

A basic Tsetlin machine takes a vector [math]\displaystyle{ X=[x_1,\ldots,x_o] }[/math] of o Boolean features as input, to be classified into one of two classes, [math]\displaystyle{ y=0 }[/math] or [math]\displaystyle{ y=1 }[/math]. Together with their negated counterparts, [math]\displaystyle{ \bar{x}_k = {\lnot} {x}_k = 1-x_k }[/math], the features form a literal set [math]\displaystyle{ L = \{x_1,\ldots,x_o,\bar{x}_1,\ldots,\bar{x}_o\} }[/math].

Clause computing module

A Tsetlin machine pattern is formulated as a conjunctive clause [math]\displaystyle{ C_j }[/math], formed by ANDing a subset [math]\displaystyle{ L_j {\subseteq} L }[/math] of the literal set:

[math]\displaystyle{ C_j (X)=\bigwedge_{{{l}} {\in} L_j} l = \prod_{{{l}} {\in} L_j} l }[/math].

For example, the clause [math]\displaystyle{ C_j(X)=x_1\land{\lnot}x_2=x_1 \bar{x}_2 }[/math] consists of the literals [math]\displaystyle{ L_j = \{x_1,\bar{x}_2\} }[/math] and outputs 1 iff [math]\displaystyle{ x_1 = 1 }[/math] and [math]\displaystyle{ x_2 = 0 }[/math].

Summation and thresholding module

The number of clauses employed is a user-configurable parameter n. Half of the clauses are assigned positive polarity. The other half is assigned negative polarity. The clause outputs, in turn, are combined into a classification decision through summation and thresholding using the unit step function [math]\displaystyle{ u(v) = 1 ~\text{if}~ v \ge 0 ~\text{else}~ 0 }[/math]:

[math]\displaystyle{ \hat{y} = u\left(\sum_{j=1}^{n/2} C_j^+(X) - \sum_{j=1}^{n/2} C_j^-(X)\right). }[/math] In other words, classification is based on a majority vote, with the positive clauses voting for [math]\displaystyle{ y=1 }[/math] and the negative for [math]\displaystyle{ y=0 }[/math]. The classifier

[math]\displaystyle{ \hat{y} = u\left(x_1 \bar{x}_2 + \bar{x}_1 x_2 - x_1 x_2 - \bar{x}_1 \bar{x}_2\right) }[/math],

for instance, captures the XOR-relation.

Feedback module

Type I feedback

Type I Feedback

Action	Clause	1		0
	Literal	1	0	1	0
Include literal	P(reward)	[math]\displaystyle{ \frac{s-1}{s} }[/math]	N/A	0	0
	P(inaction)	[math]\displaystyle{ \frac{1}{s} }[/math]	N/A	[math]\displaystyle{ \frac{s-1}{s} }[/math]	[math]\displaystyle{ \frac{s-1}{s} }[/math]
	P(penalty)	0	N/A	[math]\displaystyle{ \frac{1}{s} }[/math]	[math]\displaystyle{ \frac{1}{s} }[/math]
Exclude literal	P(reward)	0	[math]\displaystyle{ \frac{1}{s} }[/math]	[math]\displaystyle{ \frac{1}{s} }[/math]	[math]\displaystyle{ \frac{1}{s} }[/math]
	P(inaction)	[math]\displaystyle{ \frac{1}{s} }[/math]	[math]\displaystyle{ \frac{s-1}{s} }[/math]	[math]\displaystyle{ \frac{s-1}{s} }[/math]	[math]\displaystyle{ \frac{s-1}{s} }[/math]
	P(penalty)	[math]\displaystyle{ \frac{s-1}{s} }[/math]	0	0	0

Type II feedback

Type II Feedback

Action	Clause	1		0
	Literal	1	0	1	0
Include literal	P(reward)	0	N/A	0	0
	P(inaction)	1.0	N/A	1.0	1.0
	P(penalty)	0	N/A	0	0
Exclude literal	P(reward)	0	0	0	0
	P(inaction)	1.0	0	1.0	1.0
	P(penalty)	0	1.0	0	0

Resource allocation

Resource allocation dynamics ensure that clauses distribute themselves across the frequent patterns, rather than missing some and overconcentrating on others. That is, for any input X, the probability of reinforcing a clause gradually drops to zero as the clause output sum

[math]\displaystyle{ v = \sum_{j=1}^{n/2} C_j^+(X) - \sum_{j=1}^{n/2} C_j^-(X) }[/math] approaches a user-set target T for [math]\displaystyle{ y=1 }[/math] ([math]\displaystyle{ -T }[/math] for [math]\displaystyle{ y=0 }[/math]).

If a clause is not reinforced, it does not give feedback to its Tsetlin automata, and these are thus left unchanged. In the extreme, when the voting sum v equals or exceeds the target T (the Tsetlin Machine has successfully recognized the input X), no clauses are reinforced. Accordingly, they are free to learn new patterns, naturally balancing the pattern representation resources.

Implementations

Software

• Tsetlin Machine in C, Python, multithreaded Python, CUDA
• Convolutional Tsetlin Machine
• Weighted Tsetlin Machine in C++

Hardware

• One of the first FPGA-based hardware implementation of the Tsetlin Machine on the Iris dataset was developed by the µSystems (microSystems) Research Group at Newcastle University.
• They also presented the first ASIC implementation of the Tsetlin Machine focusing on energy frugality, claiming it could deliver 10 trillion operation per Joule. The ASIC design had demoed on DATA2020.

Additional Read

Books

• An Introduction to Tsetlin Machines

Conferences

• International Symposium on the Tsetlin Machine (ISTM)

Videos

• Tsetlin Machine—A new paradigm for pervasive AI
• Keyword Spotting Using Tsetlin Machines
• IOLTS Presentation: Explainability and Dependability Analysis of Learning Automata based AI hardware
• FPGA and uC co-design: Tsetlin Machine on Iris demo
• The-Ruler-of-Tsetlin-Automaton
• Interpretable clustering and dimension reduction with Tsetlin automata machine learning.
• Predicting and explaining economic growth using real-time interpretable learning
• Early detection of breast cancer from a simple blood test
• Recent advances in Tsetlin Machines

Papers

• On the Convergence of Tsetlin Machines for the XOR Operator
• Learning Automata based Energy-efficient AI Hardware Design for IoT Applications
• On the Convergence of Tsetlin Machines for the IDENTITY- and NOT Operators
• The Tsetlin Machine - A Game Theoretic Bandit Driven Approach to Optimal Pattern Recognition with Propositional Logic

Publications/news/articles

• A low-power AI alternative to neural networks