We introduce a large language model UMamba specifically trained for the Ukrainian language using the novel state-space Mamba architecture. This paper addresses the underrepresentation of Ukrainian in current multilingual models and aims to enhance performance on Ukrainian text. We start by pretraining a state-space model (SSM) on the extensive Malyuk Ukrainian text dataset. Evaluations on perplexity and cross-entropy suggest that the model could be very useful for different downstream tasks, demonstrating significant improvements over comparably sized multilingual models and achieving performance on par with much larger models. Notably, the model is highly capable of handling extremely long contexts, making it well-suited for tasks requiring extended reasoning or document-level understanding. Additionally, due to the efficiency of the SSM, it requires significantly less computational power compared to standard transformer-based models, enabling more cost-effective deployment. Building on this foundation, we perform a simple instruction tuning alignment using a translated version of the Alpaca dataset, allowing the model to better follow instructions. These results underscore the benefits of dedicated monolingual pretraining and targeted alignment, demonstrating that our approach effectively enhances language-specific performance while maintaining efficiency.

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

Linear-Time Sequence Modeling with Selective State Spaces for the Ukrainian Language

  • Anton Bazdyrev

摘要

We introduce a large language model UMamba specifically trained for the Ukrainian language using the novel state-space Mamba architecture. This paper addresses the underrepresentation of Ukrainian in current multilingual models and aims to enhance performance on Ukrainian text. We start by pretraining a state-space model (SSM) on the extensive Malyuk Ukrainian text dataset. Evaluations on perplexity and cross-entropy suggest that the model could be very useful for different downstream tasks, demonstrating significant improvements over comparably sized multilingual models and achieving performance on par with much larger models. Notably, the model is highly capable of handling extremely long contexts, making it well-suited for tasks requiring extended reasoning or document-level understanding. Additionally, due to the efficiency of the SSM, it requires significantly less computational power compared to standard transformer-based models, enabling more cost-effective deployment. Building on this foundation, we perform a simple instruction tuning alignment using a translated version of the Alpaca dataset, allowing the model to better follow instructions. These results underscore the benefits of dedicated monolingual pretraining and targeted alignment, demonstrating that our approach effectively enhances language-specific performance while maintaining efficiency.