TituLLMs: A Family of Bangla LLMs with Comprehensive Benchmarking

TL;DR

TituLLMs, a pioneering Bangla LLM with 1B and 3B parameters, uses extended tokenizer and 37B token dataset, outperforming multilingual versions in key tasks.

cs.CL 🔴 Advanced 2025-02-17 56 views
Shahriar Kabir Nahin Rabindra Nath Nandi Sagor Sarker Quazi Sarwar Muhtaseem Md Kowsher Apu Chandraw Shill Md Ibrahim Mehadi Hasan Menon Tareq Al Muntasir Firoj Alam
NLP Low-resource Languages Pretraining Vocabulary Extension Benchmarking

Key Findings

Methodology

This work employs an extended Llama-3.2 tokenizer, integrating language-specific vocabulary, combined with multi-source data including web, books, and synthetic datasets. Vocabulary expansion from 32K to 96K tokens was systematically tested, optimizing token efficiency (TPW). The models were trained on approximately 37 billion tokens using the LlamaFactory platform, with a context length of 4096 tokens, and fine-tuned through continual pretraining. Five benchmark datasets covering world knowledge, commonsense reasoning, and reading comprehension were developed and used for evaluation. The training process involved incremental vocabulary merging, synthetic data augmentation, and multi-task evaluation, with hyperparameters tuned for low-resource adaptation.

Key Results

  • TituLLMs-3B achieved 0.25 accuracy on Bangla MMLU in 0-shot, surpassing prior multilingual models, and scored 0.53 on BoolQ in 0-shot, indicating strong question-answering ability. In reasoning tasks like PIQA, it scored 0.60, demonstrating solid physical commonsense understanding. Vocabulary expansion reduced tokens per word (TPW), improving comprehension efficiency. Synthetic data, including translation and transcription, significantly enhanced knowledge coverage, especially in low-resource settings. Despite these advances, the models still show limitations in world knowledge, suggesting further data and training are needed.

Significance

This research addresses the critical gap in NLP for low-resource languages, demonstrating that with targeted vocabulary expansion and synthetic data, high-quality language models can be developed even with limited resources. The approach offers a scalable pathway for other low-resource languages, fostering inclusivity in AI. The open release of models and datasets provides valuable tools for academia and industry, enabling localized applications such as automated translation, content generation, and educational tools, ultimately broadening AI’s reach in underserved linguistic communities.

Technical Contribution

The study introduces a novel vocabulary merging strategy, combining incremental token set expansion with continual pretraining, tailored for low-resource languages. It innovates by integrating culture-specific vocabulary and high-quality synthetic datasets generated via expressive semantic translation (EST). The multi-source data pipeline and evaluation framework set new standards for low-resource NLP, providing a replicable methodology for similar languages. The models demonstrate improved token efficiency and reasoning capabilities, validated through comprehensive benchmarks, establishing a new baseline for low-resource language modeling.

Novelty

This is the first systematic effort to build a large-scale pretrained Bangla language model from scratch, employing vocabulary expansion and synthetic data generation tailored to cultural context. The EST technique for high-quality dataset creation and the multi-task benchmarking suite are innovative contributions that significantly advance low-resource NLP research, setting a precedent for future work in similar languages.

Limitations

  • The models were trained on a single epoch with 37 billion tokens, limiting their full knowledge capture. The dataset, while diverse, still lacks the breadth of high-resource languages, affecting performance on complex world knowledge tasks. Additionally, the models’ reasoning and long-text comprehension need further improvement, requiring multi-stage fine-tuning and larger datasets in future iterations.

Future Work

Future directions include multi-epoch training on larger, more diverse datasets, integrating multi-modal data such as images and audio, and exploring cross-lingual transfer techniques. Enhancing the models’ reasoning, memory, and contextual understanding through multi-stage fine-tuning and reinforcement learning will be prioritized. Moreover, expanding benchmark datasets to cover more real-world scenarios will further validate and improve model robustness.

AI Executive Summary

The rapid growth of large language models (LLMs) has revolutionized NLP, yet most high-performing models are tailored to resource-rich languages like English. Low-resource languages such as Bangla face significant challenges due to limited digital content, hindering the development of effective models. This study introduces TituLLMs, the first dedicated Bangla pretrained language models with 1B and 3B parameters, designed to overcome these barriers.

Building upon the Transformer architecture, the researchers extended the Llama-3.2 tokenizer to incorporate language-specific vocabulary, enabling more efficient tokenization. They compiled a comprehensive pretraining dataset of approximately 37 billion tokens from diverse sources, including web pages, books, synthetic translations, and transcriptions. The data collection pipeline involved querying the Common Crawl dataset, applying language-specific filtering rules, deduplication, and synthetic data generation using translation and speech recognition technologies.

To optimize vocabulary efficiency, multiple vocabularies ranging from 32K to 96K tokens were trained and merged into the base model. The models were trained with a context length of 4096 tokens, utilizing the LlamaFactory platform, with one epoch requiring 1750 GPU hours. The models were evaluated on five newly developed benchmark datasets covering world knowledge, commonsense reasoning, and reading comprehension. Results showed that TituLLMs outperformed existing multilingual models, especially in reasoning tasks, with a 0.25 accuracy on Bangla MMLU and 0.53 on BoolQ in zero-shot settings.

This work demonstrates that targeted vocabulary expansion and synthetic data augmentation can significantly improve low-resource language models. The models’ performance indicates promising potential for applications in automated translation, content creation, and educational tools tailored to Bangla speakers. Despite current limitations, such as knowledge coverage and reasoning depth, the research sets a foundation for future enhancements through multi-epoch training, multi-modal data integration, and broader benchmarking.

The models and datasets are publicly available, fostering further research and development. Overall, this study marks a significant step toward inclusive AI, enabling more equitable access to NLP technology for underserved languages and cultures, and inspiring similar efforts for other low-resource languages worldwide.

Deep Analysis

Background

Recent advances in Transformer-based models like GPT, BERT, and Llama have significantly advanced NLP, primarily for high-resource languages. However, low-resource languages such as Bangla lack sufficient digital corpora, limiting the development of effective models. Prior efforts include Zehady et al. (2024), who fine-tuned Llama on 12.4 million news articles, but their dataset remains small. The challenge lies in data scarcity, vocabulary limitations, and cultural knowledge gaps. Addressing these issues requires innovative data collection, synthetic data augmentation, and vocabulary expansion strategies. This research aims to build a dedicated Bangla language model, leveraging multi-source data and novel translation techniques, to bridge the gap between high-resource and low-resource NLP.

Core Problem

The core problem is the scarcity of large-scale, high-quality digital corpora for Bangla, which hampers the training of robust language models. Existing models rely heavily on transfer learning from multilingual models, but their performance remains suboptimal due to limited vocabulary coverage and cultural knowledge. Additionally, there is a lack of comprehensive benchmarks to evaluate model capabilities specifically in Bangla. Overcoming these bottlenecks involves developing extensive, culturally relevant datasets, optimizing vocabulary efficiency, and designing evaluation benchmarks that reflect real-world tasks. The challenge is further compounded by the need for models to understand complex reasoning and world knowledge within limited data environments.

Innovation

This work introduces several innovations: 1) Vocabulary expansion through training multiple tokenizers with different sizes and merging them into the base model, significantly reducing tokens per word (TPW). 2) Multi-source data collection pipeline, including web scraping, book digitization with OCR, and synthetic data generation via translation and speech recognition, to create a diverse corpus of 37 billion tokens. 3) The Expressive Semantic Translation (EST) method, which refines translation quality for benchmark data, ensuring cultural and contextual relevance. 4) Development of five comprehensive benchmark datasets covering educational, commonsense, and world knowledge tasks, tailored for Bangla. These innovations collectively enable the creation of a high-performing, culturally adapted low-resource language model.

Methodology

  • �� Data collection: Query Common Crawl via Amazon Athena to extract Bangla web data, followed by filtering based on language and domain-specific rules, then deduplication using MinHash.
  • �� Book digitization: Collect open-source Bangla books, apply OCR (Google OCR and Tesseract), and train document segmentation models (YOLO) to remove non-text elements.
  • �� Synthetic data: Generate conversational data via crawled topics, translate English texts using NLLB-600M, and transcribe speech data with Bangla ASR.
  • �� Vocabulary expansion: Train multiple tokenizers (32K-96K tokens), merge with Llama-3.2, optimize for token efficiency.
  • �� Model training: Fine-tune Llama-3.2 models on the assembled dataset using the LlamaFactory platform, with context length 4096, one epoch (~1750 GPU hours).
  • �� Evaluation: Develop five benchmark datasets, including Bangla MMLU, BoolQ, CommonsenseQA, OpenBookQA, and PIQA, for comprehensive performance assessment.

Experiments

The experiments involve training two models (1B and 3B parameters) with different vocabulary sizes, then evaluating on five benchmark datasets. The datasets cover multiple domains: education (Bangla MMLU), commonsense reasoning (CommonsenseQA), factual knowledge (OpenBookQA), and physical reasoning (PIQA). Performance metrics include accuracy in zero-shot and few-shot settings, with comparisons against baseline models like Llama-3.2, Qwen, and Gemma. Ablation studies analyze the impact of vocabulary size, synthetic data, and training epochs. The evaluation setup ensures robustness across diverse tasks, with hyperparameters tuned for low-resource adaptation.

Results

TituLLMs-3B achieved 0.25 accuracy on Bangla MMLU in 0-shot, outperforming prior multilingual models. In BoolQ, it scored 0.53 in 0-shot, indicating strong question-answering ability. The PIQA score reached 0.60, demonstrating effective physical commonsense reasoning. Vocabulary expansion reduced tokens per word (TPW), improving comprehension efficiency. Synthetic data, especially translation and transcription, significantly enhanced knowledge coverage. Although performance is promising, the models still lag in complex world knowledge tasks, indicating the need for larger datasets and multi-epoch training.

Applications

The models can be used for automated question answering, content generation, translation, and educational tools in Bengali-speaking regions. Their cultural and linguistic adaptation makes them suitable for local content creation, virtual assistants, and language learning applications. Future integration with multimodal data (images, speech) could further enhance their utility, enabling more natural human-computer interactions in low-resource settings.

Limitations & Outlook

The models were trained for only one epoch, limiting knowledge depth. Data scarcity restricts coverage of complex world facts. The models show room for improvement in reasoning and long-text understanding. Future work should include multi-epoch training, larger datasets, and multi-modal integration to address these limitations.

Plain Language Accessible to non-experts

想象你在一家工厂里,工厂里有很多工人(模型),他们需要学习如何制造各种产品(理解和生成语言)。如果工厂只有少量的工具和材料(数据),工人们就很难学会所有的技能。这个研究就像给工厂带来了新工具和新材料,让工人们可以更快学会孟加拉语的各种技能。通过不断练习和改进工具,工人们变得更聪明、更懂得文化和习俗,能帮人们写文章、回答问题,就像一个懂得很多的助手一样。这些工人还学会了用不同的语言交流(翻译、转写),让工厂的产品更丰富、更贴近实际需求。最终,这个工厂的工人们变得更厉害,也让孟加拉语的应用变得更广泛,大家都能用上这些新工具。

ELI14 Explained like you're 14

想象你在学校里学习新东西,但你只知道一些基础知识,没有很多书可以帮你学习。这个研究就像是给你带来了很多新书和学习工具,让你能更快理解孟加拉语。科学家们用特别的方法,把不同的资料(网页、书、对话、翻译)整理在一起,像拼拼图一样,把所有信息都放到一个大脑里。然后,他们用这些资料训练一个超级聪明的机器人,让它学会读、写、理解孟加拉语。这个机器人还学会了用孟加拉语翻译成英语,或者用英语翻译成孟加拉语,就像你用翻译软件一样。它还能回答各种问题,比如“孟加拉的首都是哪里?”或者“为什么天空是蓝色的?”虽然它还不完美,但已经比以前更聪明了。未来,这个机器人会变得更厉害,能帮人们写作文、做作业,甚至帮老师讲课。这就像给你带来了一个超级聪明的朋友,随时帮你解答问题,学习变得更轻松有趣!

Glossary

Tokenizer

A method to split text into smaller units like words or subwords, helping models understand language structure. It uses Byte Pair Encoding (BPE).

Used to expand and optimize vocabulary for Bangla.

Pretraining

Training a model on large unlabeled data to learn basic language patterns, typically using Transformer architecture.

Foundation phase for language understanding.

Synthetic Data

Artificially generated data via translation, transcription, or paraphrasing to supplement scarce real data.

Enhances low-resource language training.

Vocabulary Expansion

Adding new tokens to the model's vocabulary to better cover language-specific words and cultural terms.

Improves comprehension and generation in Bangla.

Benchmark Dataset

A standard set of tasks and questions used to evaluate model performance.

Assesses understanding in knowledge, reasoning, and comprehension.

Open Questions Unanswered questions from this research

  • 1 Despite improvements, the models still lack deep world knowledge and reasoning capabilities, especially in complex scenarios. Larger, more diverse datasets and multi-epoch training are needed to address these gaps.

Applications

Immediate Applications

Bengali Question-Answering System

Deploy TituLLMs for local info retrieval, customer support, and educational tools, improving access to information.

Cultural Content Generation

Assist content creators in generating stories, articles, and educational materials in Bengali, enriching local culture.

Long-term Vision

Multimodal Multilingual AI Assistants

Integrate images, speech, and text to create versatile AI helpers for diverse industries, promoting AI accessibility.

Abstract

In this paper, we present TituLLMs, the first large pretrained Bangla LLMs, available in 1b and 3b parameter sizes. Due to computational constraints during both training and inference, we focused on smaller models. To train TituLLMs, we collected a pretraining dataset of approximately ~37 billion tokens. We extended the Llama-3.2 tokenizer to incorporate language- and culture-specific knowledge, which also enables faster training and inference. There was a lack of benchmarking datasets to benchmark LLMs for Bangla. To address this gap, we developed five benchmarking datasets. We benchmarked various LLMs, including TituLLMs, and demonstrated that TituLLMs outperforms its initial multilingual versions. However, this is not always the case, highlighting the complexities of language adaptation. Our work lays the groundwork for adapting existing multilingual open models to other low-resource languages. To facilitate broader adoption and further research, we have made the TituLLMs models and benchmarking datasets publicly available (https://huggingface.co/collections/hishab/titulm-llama-family-6718d31fc1b83529276f490a).

cs.CL cs.AI