How Does Reasoning Flow? Tracing Attention-Induced Information Flow for Targeted RL in LLMs
信息流推理：用注意力图追溯回答导向的推理骨干，实现精准token级信用分配
🧭 推理信息流引导的token级RL信用分配 ★ 深度解读
Shanghai Jiao Tong University · Alibaba Group · Shanghai Artificial Intelligence Laboratory
将LLM推理过程建模为注意力引导的有向无环图，通过回答条件化的流守恒变换识别关键推理骨干token，用于强化学习的细粒度信用分配
📌 工作参考价值（4 点）
- 利用内部注意力图构建回答导向的信息流骨干，对token进行重要度排序，可借鉴到对话生成、Agent轨迹的细粒度信用分配，强化关键决策步骤而抑制无关填充。
- Doob‑h变换实现流守恒，保证中间节点不因路径长度而稀释或膨胀影响力，该归一化思想可直接移植到其他基于图的影响力传播任务（如推荐中的用户行为传导、物品关联）。
- 固定top‑40%高流量token进行GRPO loss重加权，超参简单且表现稳定，工程上仅需一次额外前向传播（2‑4%时间开销），适合在线RL训练环境。
- 中间层注意力包含最丰富的推理结构，避免使用全部层导致信号稀释，这对注意力分析的应用（如可解释性、模型诊断）提供了实用的层选择经验。