arXiv:2412.17548cs.CLcs.AI2024-12被引 1

4GB显存下用QLoRA微调1.5B模型,让阿拉伯语LLM更高效精准。

Resource-Aware Arabic LLM Creation: Model Adaptation, Integration, and Multi-Domain Testing

  • 用QLoRA量化低秩适配,4GB显存完成阿拉伯语模型微调。
  • 训练10,000步后损失降至0.1083,多任务表现显著提升。
  • 适合资源受限场景的阿拉伯语NLP研究与应用开发者。

本文提出一种新方法,使用量化低秩适配(QLoRA)在仅4GB显存的系统上微调Qwen2-1.5B模型以处理阿拉伯语。通过Bactrian、OpenAssistant及维基百科阿拉伯语语料等多样化数据集,结合定制化数据预处理、模型配置与梯度累积、混合精度训练等优化技术,解决阿拉伯语的形态复杂性、方言差异和符号标记处理难题。实验显示,在10,000次训练迭代后,模型损失收敛至0.1083,显著提升文本分类、问答与方言识别等任务性能。我们还分析了显存占用、训练动态与抗扰性,证明该模型对阿拉伯语语言现象有更强适应能力。本研究为多语言AI提供了资源高效的专用模型构建方案,推动低资源语言NLP的发展。

原文摘要 · Abstract (English)

This paper presents a novel approach to fine-tuning the Qwen2-1.5B model for Arabic language processing using Quantized Low-Rank Adaptation (QLoRA) on a system with only 4GB VRAM. We detail the process of adapting this large language model to the Arabic domain, using diverse datasets including Bactrian, OpenAssistant, and Wikipedia Arabic corpora. Our methodology involves custom data preprocessing, model configuration, and training optimization techniques such as gradient accumulation and mixed-precision training. We address specific challenges in Arabic NLP, including morphological complexity, dialectal variations, and diacritical mark handling. Experimental results over 10,000 training steps show significant performance improvements, with the final loss converging to 0.1083. We provide comprehensive analysis of GPU memory usage, training dynamics, and model evaluation across various Arabic language tasks, including text classification, question answering, and dialect identification. The fine-tuned model demonstrates robustness to input perturbations and improved handling of Arabic-specific linguistic phenomena. This research contributes to multilingual AI by demonstrating a resource-efficient approach for creating specialized language models, potentially democratizing access to advanced NLP technologies for diverse linguistic communities. Our work paves the way for future research in low-resource language adaptation and efficient fine-tuning of large language models.

阿拉伯语QLoRA低资源微调

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。