Can Computation from Earlier Problems Help LLMs Solve New Ones?

Hugging Face Daily Papers Papers

Summary

This paper introduces STAIR (Stale-Token Attention for Inter-query Reuse), a method that stores keys and values from earlier response generations in a fixed bank and trains only 12,288 parameters to direct current queries, improving later-turn accuracy by up to 11.67 percentage points on Qwen models across four benchmarks.

Large language models often solve independent problems in the same conversation. Can computation from earlier problems help them solve new ones? To answer this question, we first conduct preliminary experiments showing that retained history can raise or lower later-turn accuracy, even within the same domain. To understand these effects, we use controlled replay to isolate internal state changes specific to each problem-history pairing. Across different histories, these changes preserve similar relationships among current problems. To improve reasoning under retained history, we introduce STAIR (Stale-Token Attention for Inter-query Reuse). STAIR captures keys and values from earlier response generation in a fixed bank. It learns to redirect current queries when they read this bank during prompt processing. The base model remains frozen; only 12,288 parameters are trained. Across three Qwen models and four benchmarks, STAIR improves average later-turn accuracy by up to 11.67 percentage points over the unmodified model with history.
Original Article
View Cached Full Text

Cached at: 10/05/26, 12:47 PM

Paper page - Can Computation from Earlier Problems Help LLMs Solve New Ones?

Source: https://huggingface.co/papers/2609.39394

Abstract

Largelanguagemodelsoftensolveindependentproblemsinthesameconversation.Cancomputationfromearlierproblemshelpthemsolvenewones?Toanswerthisquestion,wefirstconductpreliminaryexperimentsshowingthatretainedhistorycanraiseorlowerlater-turnaccuracy,evenwithinthesamedomain.Tounderstandtheseeffects,weusecontrolledreplaytoisolateinternalstatechangesspecifictoeachproblem-historypairing.Acrossdifferenthistories,thesechangespreservesimilarrelationshipsamongcurrentproblems.Toimprovereasoningunderretainedhistory,weintroduceSTAIR(Stale-TokenAttentionforInter-queryReuse).STAIRcaptureskeysandvaluesfromearlierresponsegenerationinafixedbank.Itlearnstoredirectcurrentquerieswhentheyreadthisbankduringpromptprocessing.Thebasemodelremainsfrozen;only12,288parametersaretrained.AcrossthreeQwenmodelsandfourbenchmarks,STAIRimprovesaveragelater-turnaccuracybyupto11.67percentagepointsovertheunmodifiedmodelwithhistory.

View arXiv pageView PDFAdd to collection

Get this paper in your agent:

hf papers read 2609\.39394

Don’t have the latest CLI?curl \-LsSf https://hf\.co/cli/install\.sh \| bash

Models citing this paper0

No model linking this paper

Cite arxiv.org/abs/2609.39394 in a model README.md to link it from this page.

Datasets citing this paper0

No dataset linking this paper

Cite arxiv.org/abs/2609.39394 in a dataset README.md to link it from this page.

Spaces citing this paper0

No Space linking this paper

Cite arxiv.org/abs/2609.39394 in a Space README.md to link it from this page.

Collections including this paper0

No Collection including this paper

Add this paper to acollectionto link it from this page.

Similar Articles