Towards Optimal Learning of Language Models

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Gu, Yuxian, Dong, Li, Hao, Yaru, Dong, Qingxiu, Huang, Minlie, Wei, Furu
Formato: Preprint
Publicado: 2024
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866911787645927424
author Gu, Yuxian
Dong, Li
Hao, Yaru
Dong, Qingxiu
Huang, Minlie
Wei, Furu
author_facet Gu, Yuxian
Dong, Li
Hao, Yaru
Dong, Qingxiu
Huang, Minlie
Wei, Furu
contents This work studies the general principles of improving the learning of language models (LMs), which aims at reducing the necessary training steps for achieving superior performance. Specifically, we present a theory for the optimal learning of LMs. We first propose an objective that optimizes LM learning by maximizing the data compression ratio in an "LM-training-as-lossless-compression" view. Then, we derive a theorem, named Learning Law, to reveal the properties of the dynamics in the optimal learning process under our objective. The theorem is then validated by experiments on a linear classification and a real-world language modeling task. Finally, we empirically verify that the optimal learning of LMs essentially stems from the improvement of the coefficients in the scaling law of LMs, indicating great promise and significance for designing practical learning acceleration methods. Our code can be found at https://aka.ms/LearningLaw.
format Preprint
id arxiv_https___arxiv_org_abs_2402_17759
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Towards Optimal Learning of Language Models
Gu, Yuxian
Dong, Li
Hao, Yaru
Dong, Qingxiu
Huang, Minlie
Wei, Furu
Computation and Language
This work studies the general principles of improving the learning of language models (LMs), which aims at reducing the necessary training steps for achieving superior performance. Specifically, we present a theory for the optimal learning of LMs. We first propose an objective that optimizes LM learning by maximizing the data compression ratio in an "LM-training-as-lossless-compression" view. Then, we derive a theorem, named Learning Law, to reveal the properties of the dynamics in the optimal learning process under our objective. The theorem is then validated by experiments on a linear classification and a real-world language modeling task. Finally, we empirically verify that the optimal learning of LMs essentially stems from the improvement of the coefficients in the scaling law of LMs, indicating great promise and significance for designing practical learning acceleration methods. Our code can be found at https://aka.ms/LearningLaw.
title Towards Optimal Learning of Language Models
topic Computation and Language
url https://arxiv.org/abs/2402.17759