Distilling LLMs' Decomposition Abilities into Compact Language Models

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Tarasov, Denis, Shridhar, Kumar
Formato: Preprint
Publicado: 2024
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866909091232743424
author Tarasov, Denis
Shridhar, Kumar
author_facet Tarasov, Denis
Shridhar, Kumar
contents Large Language Models (LLMs) have demonstrated proficiency in their reasoning abilities, yet their large size presents scalability challenges and limits any further customization. In contrast, compact models offer customized training but often fall short in solving complex reasoning tasks. This study focuses on distilling the LLMs' decomposition skills into compact models using offline reinforcement learning. We leverage the advancements in the LLM`s capabilities to provide feedback and generate a specialized task-specific dataset for training compact models. The development of an AI-generated dataset and the establishment of baselines constitute the primary contributions of our work, underscoring the potential of compact models in replicating complex problem-solving skills.
format Preprint
id arxiv_https___arxiv_org_abs_2402_01812
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Distilling LLMs' Decomposition Abilities into Compact Language Models
Tarasov, Denis
Shridhar, Kumar
Computation and Language
Artificial Intelligence
Machine Learning
Large Language Models (LLMs) have demonstrated proficiency in their reasoning abilities, yet their large size presents scalability challenges and limits any further customization. In contrast, compact models offer customized training but often fall short in solving complex reasoning tasks. This study focuses on distilling the LLMs' decomposition skills into compact models using offline reinforcement learning. We leverage the advancements in the LLM`s capabilities to provide feedback and generate a specialized task-specific dataset for training compact models. The development of an AI-generated dataset and the establishment of baselines constitute the primary contributions of our work, underscoring the potential of compact models in replicating complex problem-solving skills.
title Distilling LLMs' Decomposition Abilities into Compact Language Models
topic Computation and Language
Artificial Intelligence
Machine Learning
url https://arxiv.org/abs/2402.01812