Adaptive Engram Memory System for Indonesian Language Model: Generative AI Based on TOBA LM for Batak and Minang Language
Organizations: AI Research Center IT Del & Bandung Fe Institute · Bandung Fe Institute · Bandung Fe Institute & Adjunct Science Fellow in InaAI
Abstract
This study presents TOBA-LM, a trilingual language model based on GPT-2 architecture with 1.2 billion parameters, trained on a corpus encompassing Indonesian, Batak, and Minangkabau using syllabic-agglutinative tokenization. The architecture integrates an Engram Memory mechanism, an adaptive n-gram-based memory system with a 500,000 x 768 embedding table that captures morphological dependencies through bigram and trigram pathways. Empirical results demonstrate a training efficiency of 80%, with the loss value dropping from 6.4 to 1.7996 in only 12,973 steps -- significantly faster than the conventional transformer architecture, which required over 70,000 steps to achieve comparable convergence. These findings confirm that the integration of external statistical memory substantially reduces computational requirements for developing regional language models under limited resources.
Figures & tables
| Category | Main Parameter | Technical Specification |
|---|---|---|
| Backbone Architecture | Layers, Dim, Heads | 36 Blocks, d = 1280, 20 Heads |
| Context Length | 1,024 tokens | |
| Engram Module | Table & Embed Dim | 500,000 768 |
| Internal Heads | 8 Heads | |
| Tokenization | Type | Tokenizer |
| Vocab Size | 2,843 Units |
Language Model Pretraining
Language model pretraining teaches models language patterns and knowledge from large text corpora before they are adapted to specific tasks.
See all 440 papers