Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Omni TM-AE: A Scalable and Interpretable Embedding Model Using the Full Tsetlin Machine State Space

About

The increasing complexity of large-scale language models has amplified concerns regarding their interpretability and reusability. While traditional embedding models like Word2Vec and GloVe offer scalability, they lack transparency and often behave as black boxes. Conversely, interpretable models such as the Tsetlin Machine (TM) have shown promise in constructing explainable learning systems, though they previously faced limitations in scalability and reusability. In this paper, we introduce Omni Tsetlin Machine AutoEncoder (Omni TM-AE), a novel embedding model that fully exploits the information contained in the TM's state matrix, including literals previously excluded from clause formation. This method enables the construction of reusable, interpretable embeddings through a single training phase. Extensive experiments across semantic similarity, sentiment classification, and document clustering tasks show that Omni TM-AE performs competitively with and often surpasses mainstream embedding models. These results demonstrate that it is possible to balance performance, scalability, and interpretability in modern Natural Language Processing (NLP) systems without resorting to opaque architectures.

Ahmed K. Kadhim, Lei Jiao, Rishad Shafik, Ole-Christoffer Granmo• 2025

Related benchmarks

TaskDatasetResultRank
Word SimilarityMEN
Spearman Rho0.596
74
Word SimilarityMechanical Turk-771
Spearman ρ0.487
14
Word SimilarityMTurk-287
Spearman Correlation0.519
12
Word SimilarityWS-353 SIM
Spearman Correlation0.478
12
Word SimilarityWord-similarity Suite Avg.
Spearman Correlation (ρ)0.543
6
Showing 5 of 5 rows

Other info

Follow for update