Path-Consistency with Prefix Enhancement for Efficient Inference in LLMs

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Zhu, Jiace, Huang, Yuanzhe, Shen, Yingtao, Zhao, Jie, Zou, An
Formato: Preprint
Publicado: 2024
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866917057319141376
author Zhu, Jiace
Huang, Yuanzhe
Shen, Yingtao
Zhao, Jie
Zou, An
author_facet Zhu, Jiace
Huang, Yuanzhe
Shen, Yingtao
Zhao, Jie
Zou, An
contents To enhance the reasoning capabilities of large language models (LLMs), self-consistency has become a popular approach, combining multiple samplings with majority voting. However, current methods are computationally expensive and time-consuming due to the need for numerous samplings. To address this, this paper introduces path-consistency, which leverages the confidence of earlier-generated answers to identify the most promising prefix and guide the generation of subsequent branches. By dynamically guiding the generation of subsequent branches based on this prefix, path-consistency mitigates both the errors and redundancies from random or less useful sampling in self-consistency. This approach reduces errors and redundancies from random sampling, significantly accelerating inference by minimizing token consumption. Our extensive empirical results demonstrate that path-consistency improves inference latency by up to 40.5\%, while maintaining task accuracy across various tasks, including mathematical reasoning, commonsense reasoning, and symbolic reasoning.
format Preprint
id arxiv_https___arxiv_org_abs_2409_01281
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Path-Consistency with Prefix Enhancement for Efficient Inference in LLMs
Zhu, Jiace
Huang, Yuanzhe
Shen, Yingtao
Zhao, Jie
Zou, An
Computation and Language
Artificial Intelligence
To enhance the reasoning capabilities of large language models (LLMs), self-consistency has become a popular approach, combining multiple samplings with majority voting. However, current methods are computationally expensive and time-consuming due to the need for numerous samplings. To address this, this paper introduces path-consistency, which leverages the confidence of earlier-generated answers to identify the most promising prefix and guide the generation of subsequent branches. By dynamically guiding the generation of subsequent branches based on this prefix, path-consistency mitigates both the errors and redundancies from random or less useful sampling in self-consistency. This approach reduces errors and redundancies from random sampling, significantly accelerating inference by minimizing token consumption. Our extensive empirical results demonstrate that path-consistency improves inference latency by up to 40.5\%, while maintaining task accuracy across various tasks, including mathematical reasoning, commonsense reasoning, and symbolic reasoning.
title Path-Consistency with Prefix Enhancement for Efficient Inference in LLMs
topic Computation and Language
Artificial Intelligence
url https://arxiv.org/abs/2409.01281