RESCUE：用强化学习把语言模型错误修复为稀疏电路
RESCUE: Repairing Language Model Errors to Sparse Circuits via Reinforcement Learning
AI 导读
RESCUE 是一个定位并修复错误相关稀疏电路的框架，通过强化学习优化掩码，再用剪枝和微调对错误电路做定向更新。作者指出，多步推理和长文本生成中早期偏差会让前缀偏离监督参考，使基于 SFT 的掩码优化忽略生成阶段的错误电路，因此改用多次掩码模型 rollout 的强化学习来细化掩码。在数学推理上，该方法识别出密度 1.40% 的数学错误电路，修复后准确率从 6.0% 提升到 75.5%；在医疗问答上，密度 1.44% 的电路把修复集准确率从 0% 提升到 81%。代码已公开。