FLEx: Language Modeling with Few-shot Language Explanations

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Avsian, Adar, Richardson, Christopher, Sundar, Anirudh, Heck, Larry
Formato: Preprint
Publicado: 2026
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866917216523386880
author Avsian, Adar
Richardson, Christopher
Sundar, Anirudh
Heck, Larry
author_facet Avsian, Adar
Richardson, Christopher
Sundar, Anirudh
Heck, Larry
contents Language models have become effective at a wide range of tasks, from math problem solving to open-domain question answering. However, they still make mistakes, and these mistakes are often repeated across related queries. Natural language explanations can help correct these errors, but collecting them at scale may be infeasible, particularly in domains where expert annotators are required. To address this issue, we introduce FLEx ($\textbf{F}$ew-shot $\textbf{L}$anguage $\textbf{Ex}$planations), a method for improving model behavior using a small number of explanatory examples. FLEx selects representative model errors using embedding-based clustering, verifies that the associated explanations correct those errors, and summarizes them into a prompt prefix that is prepended at inference-time. This summary guides the model to avoid similar errors on new inputs, without modifying model weights. We evaluate FLEx on CounterBench, GSM8K, and ReasonIF. We find that FLEx consistently outperforms chain-of-thought (CoT) prompting across all three datasets and reduces up to 83\% of CoT's remaining errors.
format Preprint
id arxiv_https___arxiv_org_abs_2601_04157
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle FLEx: Language Modeling with Few-shot Language Explanations
Avsian, Adar
Richardson, Christopher
Sundar, Anirudh
Heck, Larry
Computation and Language
Machine Learning
Language models have become effective at a wide range of tasks, from math problem solving to open-domain question answering. However, they still make mistakes, and these mistakes are often repeated across related queries. Natural language explanations can help correct these errors, but collecting them at scale may be infeasible, particularly in domains where expert annotators are required. To address this issue, we introduce FLEx ($\textbf{F}$ew-shot $\textbf{L}$anguage $\textbf{Ex}$planations), a method for improving model behavior using a small number of explanatory examples. FLEx selects representative model errors using embedding-based clustering, verifies that the associated explanations correct those errors, and summarizes them into a prompt prefix that is prepended at inference-time. This summary guides the model to avoid similar errors on new inputs, without modifying model weights. We evaluate FLEx on CounterBench, GSM8K, and ReasonIF. We find that FLEx consistently outperforms chain-of-thought (CoT) prompting across all three datasets and reduces up to 83\% of CoT's remaining errors.
title FLEx: Language Modeling with Few-shot Language Explanations
topic Computation and Language
Machine Learning
url https://arxiv.org/abs/2601.04157