English

Auto-Patching: Enhancing Multi-Hop Reasoning in Language Models

Computation and Language 2025-06-03 v1 Machine Learning

Abstract

Multi-hop questions still stump large language models (LLMs), which struggle to link information across multiple reasoning steps. We introduce Auto-Patch, a novel method that dynamically patches hidden states during inference to enhance multi-hop reasoning in LLMs. Building on the PatchScopes framework, Auto-Patch selectively modifies internal representations using a learned classifier. Evaluated on the MuSiQue dataset, Auto-Patch improves the solve rate from 18.45\% (baseline) to 23.63~±\pm~0.7\% (3 runs), narrowing the gap to Chain-of-Thought prompting (27.44\%). Our results highlight the potential of dynamic hidden state interventions for advancing complex reasoning in LLMs.

Keywords

Cite

@article{arxiv.2506.00483,
  title  = {Auto-Patching: Enhancing Multi-Hop Reasoning in Language Models},
  author = {Aviv Jan and Dean Tahory and Omer Talmi and Omar Abo Mokh},
  journal= {arXiv preprint arXiv:2506.00483},
  year   = {2025}
}

Comments

8 pages, 5 figures

R2 v1 2026-07-01T02:52:11.596Z