REDaCT: Collapse-Free Unlearning in Large Reasoning Models
Organizations: Stevens Institute of Technology Hoboken, New Jersey, USA
Abstract
Unlearning in large reasoning models (LRMs) requires removing forget-set information from both the reasoning trace and the final answer. Existing approaches achieve this objective at the cost of reasoning collapse, where the model fails to produce a coherent chain-of-thought (CoT) and does not properly terminate its reasoning process. To address this challenge, we propose REDaCT (REgenerate via Detection and Corrective Tuning), an inference-time unlearning framework for LRMs. REDaCT consists of two key components: (1) a binary leak detector that identifies sentences in the reasoning trace that leak forget-set information, and (2) an embedding optimization module that learns a corrective embedding offline, such that when injected at the first detected leak during inference, it guides subsequent reasoning towards sentences that do not leak forget-set information. Experimental results show that REDaCT completely avoids reasoning collapse, substantially improving reasoning preservation compared to state-of-the-art unlearning methods while maintaining perfect answer forgetting and near-perfect CoT forgetting. Our code is available at: https://github.com/jmaccoby/redact.
Figures & tables
| Method | Forget Set | AFE | CFE | Utility | Halluc. | Refusal | Collapse |
|---|---|---|---|---|---|---|---|
| Target (pre-unlearning) | — | — | — | 0.787 | — | — | — |
| GA | f01 | 0.375 | 0.460 | 0.757 | 0.533 | 0.000 | 0.467 |
| f05 | 0.245 | 0.343 | 0.766 | 0.939 | 0.000 | 0.061 | |
| f10 | 0.190 | 0.302 | 0.770 | 0.947 | 0.013 | 0.039 | |
| GD | f01 | 0.250 | 0.347 | 0.729 | 1.000 | 0.000 | 0.000 |
| f05 | 0.220 | 0.328 | 0.702 | 0.955 | 0.023 | 0.023 |
| Questions | Forget Set | AFE | CFE | Halluc. | Refusal | Collapse |
|---|---|---|---|---|---|---|
| Direct | f01 | 1.000 | 1.000 | 0.000 | 1.000 | 0.000 |
| f05 | 1.000 | 0.977 | 0.000 | 1.000 | 0.000 | |
| f10 | 1.000 | 0.983 | 0.000 | 1.000 | 0.000 | |
| Indirect | f01 | 1.000 | 0.921 | 0.000 | 0.929 | 0.071 |
| f05 | 1.000 | 0.990 | 0.016 | 0.984 | 0.000 | |
| f10 | 1.000 | 0.990 | 0.000 | 1.000 | 0.000 |
Appendix figures & tables7 assets
Supplementary material from the paper’s appendix.
Appendix
| Judged output | Agreement statistic | Value |
|---|---|---|
| Final answer (AFE label) | Raw agreement | 0.940 |
| Cohen’s | 0.907 | |
| CoT (CFE score) | Spearman’s | 0.963 |
| Mean absolute difference | 0.077 |
| Ground-truth CoT sentence (R-TOFU #93) | REDaCT | R 2 MU |
| best-match | best-match | |
| Okay, I need to figure out which award Moshe Ben-David received… | 1.00 | 0.09 |
| Let me start by recalling what I know about him. | 1.00 | 0.22 |
| He’s an author…born in Tel Aviv, Israel… | 0.56 | 0.09 |
| I remember that he’s quite respected in his field… | 0.50 | 0.13 |
| I think he has received an award…for…Islamic literature. | 0.73 | 0.11 |
| Method | Learning rate | Epochs / steps |
|---|---|---|
| GA | 2 epochs | |
| GD | 3 epochs | |
| KL | 3 epochs | |
| PO | 2 epochs |
| Method | AFE | CFE | Utility | Halluc. | Refusal | Collapse |
|---|---|---|---|---|---|---|
| Target (pre-unlearning) | — | — | 0.652 | — | — | — |
| GA | 0.453 | 0.689 | 0.671 | 0.950 | 0.044 | 0.006 |
| GD | 0.508 | 0.630 | 0.565 | 0.985 | 0.010 | 0.005 |
| KL | 0.805 | 0.965 | 0.592 | 0.419 | 0.115 | 0.466 |
| PO | 0.350 | 0.409 | 0.545 | 0.571 | 0.400 | 0.029 |
| R 2 MU | 0.990 | 0.992 | 0.381 | 0.066 | 0.028 | 0.907 |
| Forget Set | Recall | Precision | F1 |
|---|---|---|---|
| forget01 | 1.000 | 1.000 | 1.000 |
| forget05 | 1.000 | 0.995 | 0.998 |
| forget10 | 0.995 | 1.000 | 0.998 |
Machine Unlearning
Machine unlearning removes the influence of selected data or concepts from trained models while preserving useful behavior and avoiding full retraining.
See all 349 papers