Dynamic data sampler for cross-language transfer learning in large language models

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Li, Yudong, Feng, Yuhao, Zhou, Wen, Zhao, Zhe, Shen, Linlin, Hou, Cheng, Hou, Xianxu
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866911879321878528
author Li, Yudong
Feng, Yuhao
Zhou, Wen
Zhao, Zhe
Shen, Linlin
Hou, Cheng
Hou, Xianxu
author_facet Li, Yudong
Feng, Yuhao
Zhou, Wen
Zhao, Zhe
Shen, Linlin
Hou, Cheng
Hou, Xianxu
contents Large Language Models (LLMs) have gained significant attention in the field of natural language processing (NLP) due to their wide range of applications. However, training LLMs for languages other than English poses significant challenges, due to the difficulty in acquiring large-scale corpus and the requisite computing resources. In this paper, we propose ChatFlow, a cross-language transfer-based LLM, to address these challenges and train large Chinese language models in a cost-effective manner. We employ a mix of Chinese, English, and parallel corpus to continuously train the LLaMA2 model, aiming to align cross-language representations and facilitate the knowledge transfer specifically to the Chinese language model. In addition, we use a dynamic data sampler to progressively transition the model from unsupervised pre-training to supervised fine-tuning. Experimental results demonstrate that our approach accelerates model convergence and achieves superior performance. We evaluate ChatFlow on popular Chinese and English benchmarks, the results indicate that it outperforms other Chinese models post-trained on LLaMA-2-7B.
format Preprint
id arxiv_https___arxiv_org_abs_2405_10626
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Dynamic data sampler for cross-language transfer learning in large language models
Li, Yudong
Feng, Yuhao
Zhou, Wen
Zhao, Zhe
Shen, Linlin
Hou, Cheng
Hou, Xianxu
Computation and Language
Large Language Models (LLMs) have gained significant attention in the field of natural language processing (NLP) due to their wide range of applications. However, training LLMs for languages other than English poses significant challenges, due to the difficulty in acquiring large-scale corpus and the requisite computing resources. In this paper, we propose ChatFlow, a cross-language transfer-based LLM, to address these challenges and train large Chinese language models in a cost-effective manner. We employ a mix of Chinese, English, and parallel corpus to continuously train the LLaMA2 model, aiming to align cross-language representations and facilitate the knowledge transfer specifically to the Chinese language model. In addition, we use a dynamic data sampler to progressively transition the model from unsupervised pre-training to supervised fine-tuning. Experimental results demonstrate that our approach accelerates model convergence and achieves superior performance. We evaluate ChatFlow on popular Chinese and English benchmarks, the results indicate that it outperforms other Chinese models post-trained on LLaMA-2-7B.
title Dynamic data sampler for cross-language transfer learning in large language models
topic Computation and Language
url https://arxiv.org/abs/2405.10626