Search papers, labs, and topics across Lattice.
This paper introduces a Reinforcement Learning (RL) method aimed at optimizing the faithfulness of self-explanations in large language models (LLMs) by directly adjusting model parameters based on faithfulness metrics. The authors demonstrate that fine-tuning Llama3.1-8B and Qwen3-8B using a per-sample reward derived from the Phi-CCT correlation metric leads to significant improvements in faithfulness scores, with in-distribution metrics increasing from near-zero to 0.664. Additionally, the research reveals unexpected cross-intervention generalization, suggesting that models can transfer learned insights across different types of interventions, albeit with variability across model architectures.
Fine-tuning LLMs with RL can boost self-explanation faithfulness from nearly zero to 0.664, revealing surprising cross-intervention generalization capabilities.
We propose a Reinforcement Learning (RL) method to directly optimize the faithfulness of self-explanations - the extent to which a model's generated reasoning accurately reflects its internal decision-making process. While existing work focuses on evaluating faithfulness or using inference-time prompting frameworks to improve an LLM's self-explanation's tractability, these approaches do not provide a mechanism to directly optimize a model's parameters to generate faithful self-explanations. We bridge this gap by modifying existing faithfulness metrics into an RL training objective. We investigate (1) if models can be trained to accurately detect factors that affect their decisions, and (2) whether RL can directly optimize for the disclosure of these factors thereby improving LLM self-explanations' faithfulness. We experiment with two intervention types: random-word insertions and user-bias insertions, using a per-sample reward derived from the Phi-CCT correlation metric. RL fine-tuned Llama3.1-8B and Qwen3-8B show substantial improvements on the Phi-CCT faithfulness metric, with in-distribution scores rising from near-zero to as high as 0.664, and out-of-distribution scores reaching up to 0.691 on held-out tasks such as StrategyQA. Cross-intervention generalization is weaker but more interesting: a priori we would not expect a model trained only on random word insertions to generalize to user-bias phrases, yet Llama3.1-8B shows non-zero transfer in this direction. The reverse direction and Qwen3-8B do not replicate this, indicating model-dependent and setup-dependent effects we cannot yet explain. Lastly we analyze model behavior to rule out reward gaming behaviors that often plague RL training. Ultimately, we show that models can be trained to implicitly identify influential factors and disclose them, offering a scalable path toward reducing unfaithful reasoning in LLMs.
