Stop Rewarding Hallucinated Steps: Faithfulness-Aware Step-Level Reinforcement Learning for Small Reasoning Models

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Nie, Shuo, Deng, Hexuan, Wang, Chao, Fang, Ruiyu, Liu, Xuebo, Song, Shuangyong, Li, Yu, Zhang, Min, Li, Xuelong
Natura: Preprint
Pubblicazione: 2026
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866914606231846912
author Nie, Shuo
Deng, Hexuan
Wang, Chao
Fang, Ruiyu
Liu, Xuebo
Song, Shuangyong
Li, Yu
Zhang, Min
Li, Xuelong
author_facet Nie, Shuo
Deng, Hexuan
Wang, Chao
Fang, Ruiyu
Liu, Xuebo
Song, Shuangyong
Li, Yu
Zhang, Min
Li, Xuelong
contents As large language models become smaller and more efficient, small reasoning models (SRMs) are crucial for enabling chain-of-thought (CoT) reasoning in resource-constrained settings. However, they are prone to faithfulness hallucinations, especially in intermediate reasoning steps. Existing mitigation methods based on online reinforcement learning rely on outcome-based rewards or coarse-grained CoT evaluation, which can inadvertently reinforce unfaithful reasoning when the final answer is correct. To address these limitations, we propose Faithfulness-Aware Step-Level Reinforcement Learning (FaithRL), introducing step-level supervision via explicit faithfulness rewards from a process reward model, together with an implicit truncated resampling strategy that generates contrastive signals from faithful prefixes, while also mitigating reward hacking from step-level rewards. Experiments across multiple SRMs and Open-Book QA benchmarks demonstrate that FaithRL consistently reduces hallucinations in both the CoT and final answers, leading to more faithful and reliable reasoning. Code is available at https://github.com/Easy195/FaithRL.
format Preprint
id arxiv_https___arxiv_org_abs_2602_05897
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle Stop Rewarding Hallucinated Steps: Faithfulness-Aware Step-Level Reinforcement Learning for Small Reasoning Models
Nie, Shuo
Deng, Hexuan
Wang, Chao
Fang, Ruiyu
Liu, Xuebo
Song, Shuangyong
Li, Yu
Zhang, Min
Li, Xuelong
Computation and Language
As large language models become smaller and more efficient, small reasoning models (SRMs) are crucial for enabling chain-of-thought (CoT) reasoning in resource-constrained settings. However, they are prone to faithfulness hallucinations, especially in intermediate reasoning steps. Existing mitigation methods based on online reinforcement learning rely on outcome-based rewards or coarse-grained CoT evaluation, which can inadvertently reinforce unfaithful reasoning when the final answer is correct. To address these limitations, we propose Faithfulness-Aware Step-Level Reinforcement Learning (FaithRL), introducing step-level supervision via explicit faithfulness rewards from a process reward model, together with an implicit truncated resampling strategy that generates contrastive signals from faithful prefixes, while also mitigating reward hacking from step-level rewards. Experiments across multiple SRMs and Open-Book QA benchmarks demonstrate that FaithRL consistently reduces hallucinations in both the CoT and final answers, leading to more faithful and reliable reasoning. Code is available at https://github.com/Easy195/FaithRL.
title Stop Rewarding Hallucinated Steps: Faithfulness-Aware Step-Level Reinforcement Learning for Small Reasoning Models
topic Computation and Language
url https://arxiv.org/abs/2602.05897