Learning Evidence Highlighting for Frozen LLMs

arXiv:2604.22565v1 Announce Type: cross Abstract: Large Language Models (LLMs) can reason well, yet often miss decisive evidence when it is buried in long, noisy contexts. We introduce HiLight, an Evidence Emphasis framework that decouples evidence selection from reasoning for frozen LLM solvers. HiLight avoids compressing or rewriting the input, which can discard or distort evidence, by training a lightweight Emphasis Actor to insert minimal highlight tags around pivotal spans in the unaltered context. A frozen Solver then performs downstream reasoning on the emphasized input. We cast highlighting as a weakly supervised decision-making problem and optimize the Actor with reinforcement learning using only the Solver's task reward, requiring no evidence labels and no access to or modification of the Solver. Across sequential recommendation and long-context question answering, HiLight consistently improves performance over strong prompt-based and automated prompt-optimization baselines. The learned emphasis policy transfers zero-shot to both smaller and larger unseen Solver families, including an API-based Solver, suggesting that the Actor captures genuine, reusable evidence structure rather than overfitting to a single backbone.

Related

• ZoomR: Memory Efficient Reasoning through Multi-Granularity Key Value Retrieval
• GRAIL:Learning to Interact with Large Knowledge Graphs for Retrieval Augmented Reasoning
• Deliberative Searcher: Improving LLM Reliability via Reinforcement Learning with constraints
• Rethinking Token-Level Credit Assignment in RLVR: A Polarity-Entropy Analysis
• Revisiting Entropy Regularization: Adaptive Coefficient Unlocks Its Potential for LLM Reinforcement Learning

Source: arXiv cs.AI | 2026-04-27