Self-ReSET: Learning to Self-Recover from Unsafe Reasoning Trajectories

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Zhang, Dongcheng, Zhang, Yi, Chen, Yuxin, Zhang, An, Wang, Xiang, Lu, Chaochao
Format: Preprint
Veröffentlicht: 2026
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866913106803818496
author Zhang, Dongcheng
Zhang, Yi
Chen, Yuxin
Zhang, An
Wang, Xiang
Lu, Chaochao
author_facet Zhang, Dongcheng
Zhang, Yi
Chen, Yuxin
Zhang, An
Wang, Xiang
Lu, Chaochao
contents Large Reasoning Models possess remarkable capabilities for self-correction in general domain; however, they frequently struggle to recover from unsafe reasoning trajectories under adversarial attacks. Existing alignment methods attempt to mitigate this vulnerability by fine-tuning the model on expert data including reflection traces or adversarial prefixes. Crucially, these approaches are often hindered by static training data which inevitably deviate from model's dynamic, on-policy reasoning traces, resulting in model hardly covering its vast generation space and learning to recover from its own failures. To bridge this gap, we propose Self-ReSET, a pure reinforcement learning framework designed to equip LRMs with the intrinsic capacity to recover from their own safety error trajectories, which are subsequently reused as an initial state for reinforcement learning. Extensive experiments across various LRMs and benchmarks demonstrate that Self-ReSET significantly enhances robustness against adversarial attacks especially out-of-distribution (OOD) jailbreak prompts while maintaining general utility, along with efficient data utilization. Further analysis reveals that our method effectively fosters self-recovery patterns, enabling models to better identify and recover from unsafe intermediate error states back to benign paths. Our codes and data are available at https://github.com/Ing1024/Self-ReSET.
format Preprint
id arxiv_https___arxiv_org_abs_2605_08936
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle Self-ReSET: Learning to Self-Recover from Unsafe Reasoning Trajectories
Zhang, Dongcheng
Zhang, Yi
Chen, Yuxin
Zhang, An
Wang, Xiang
Lu, Chaochao
Artificial Intelligence
Machine Learning
Large Reasoning Models possess remarkable capabilities for self-correction in general domain; however, they frequently struggle to recover from unsafe reasoning trajectories under adversarial attacks. Existing alignment methods attempt to mitigate this vulnerability by fine-tuning the model on expert data including reflection traces or adversarial prefixes. Crucially, these approaches are often hindered by static training data which inevitably deviate from model's dynamic, on-policy reasoning traces, resulting in model hardly covering its vast generation space and learning to recover from its own failures. To bridge this gap, we propose Self-ReSET, a pure reinforcement learning framework designed to equip LRMs with the intrinsic capacity to recover from their own safety error trajectories, which are subsequently reused as an initial state for reinforcement learning. Extensive experiments across various LRMs and benchmarks demonstrate that Self-ReSET significantly enhances robustness against adversarial attacks especially out-of-distribution (OOD) jailbreak prompts while maintaining general utility, along with efficient data utilization. Further analysis reveals that our method effectively fosters self-recovery patterns, enabling models to better identify and recover from unsafe intermediate error states back to benign paths. Our codes and data are available at https://github.com/Ing1024/Self-ReSET.
title Self-ReSET: Learning to Self-Recover from Unsafe Reasoning Trajectories
topic Artificial Intelligence
Machine Learning
url https://arxiv.org/abs/2605.08936