Researchers have developed REACT, a method to intervene in the chain-of-thought (CoT) process of language models to improve safety. REACT learns a safety direction from paired safe and unsafe trajectories and applies it to an early CoT hidden state, freezing model weights thereafter. Studies indicate that the most effective control window is within the first two to three tokens of the CoT, with effectiveness diminishing significantly by the fourth or fifth token. This approach successfully increased safe CoT-answer outputs on the DeepSeek-R1-Distill-Llama-8B model from 29.06% to 82.05% while maintaining baseline levels of benign refusal and usefulness. AI
IMPACT Offers a novel technique for enhancing AI safety by intervening early in the reasoning process, potentially improving control over model outputs.
RANK_REASON Research paper detailing a new method for controlling AI model behavior. [lever_c_demoted from research: ic=1 ai=1.0]
- alphaXiv
- arXiv
- CatalyzeX
- DagsHub
- DeepSeek-R1-Distill-Llama-8B
- Gotit.pub
- Hugging Face
- IArxiv
- REACT
- ScienceCast
- StrongREJECT-273
AI-generated summary · Google Gemini · from 1 sources. How we write summaries →