Taming CoT Obfuscation in VLMs: From Mechanistic Evidence to Activation-Level Enforcement
Organizations: City University of Hong Kong · Hong Kong Baptist University · University of Toronto · The University of Sydney ·
Abstract
Reinforcement learning (RL) has substantially improved the reasoning capabilities of vision-language models (VLMs), yet it can induce chain-of-thought (CoT) obfuscation, an operational, non-intentional regime where models achieve high accuracy while producing reasoning traces that are less grounded and harder to monitor. While prior work documents this decay at the behavioral level, the underlying representational correlates of such drift remain poorly understood. In this work, we find that obfuscation is accompanied by a representational shift, where RL-induced optimization makes task-agnostic template features less separable from visually grounded content in the model's activation space. Based on this insight, we propose TAME, an activation-level intervention framework that leverages Sparse Autoencoders (SAEs) to regularize VLM internal features during RL training. Specifically, TAME utilizes an anti-obfuscation feedback loop to reduce behavioral obfuscation and applies an asymmetric SAE constraint to specific internal directions, penalizing template activations while preserving visual evidence. By directly intervening on the features associated with obfuscation, our method improves the transparency and visual grounding of reasoning traces. Experiments on the VIRL-39k and SPA-VL benchmarks across two model families show that TAME improves CoT monitorability (by up to +30.9 points on VIRL and +16.7 points on SPA-VL over GRPO), while largely preserving task accuracy and general-capability performance across four standard VLM benchmarks, with task-specific trade-offs. Our code is available at https://github.com/henrymao2004/TAME.
Figures & tables
| Stage | Question | Tool | Metric / evidence |
|---|---|---|---|
| Phenomenon | Does RL improve reward while reducing trace monitorability? | GPT-4o predictor; two-gate labels | ; monitor accuracy |
| Localization | Which layers and token types change? | Attention; Integrated Gradients (IG); patching | Ground / Entity / Template attention mass; normalized Cheating Index (CI); IG |
| Representation | How does template/ground separability change? | Frozen TopK sparse autoencoder (SAE) | Conditional enrichment score (CES); Template precision/recall; Jensen–Shannon divergence (JSD) |
| Causal test | Do selected features contribute beyond matched random features? | SAE ablation/injection; human audit | ; diagnostic pass-rate differences |
| Mitigation | Can training improve monitorability with bounded trade-offs? | Tame and component ablations | Monitorability; task metrics; benchmark accuracy |
| Trained on VIRL | Trained on SPA-VL | ||||||||||||||||
| Mitigation | General capability | Mitigation | General capability | ||||||||||||||
| Model | Method | Acc | Mon. | MMMU | MVis | MVrs | MEGA | HS | Win | Mon. | MMMU | MVis | MVrs | MEGA | |||
| Qwen | Base | – | – | – | 48.6 | 72.7 | 39.6 | 37.6 | – | – | – | – | 48.6 | 72.7 | 39.6 | 37.6 | |
| GRPO | .646 | .302 | .57 | 44.6 | 77.8 | 49.3 | 33.3 | 6.8 | 63.7 | .394 | .63 | 46.9 | 71.9 | 38.5 | 34.7 | ||
| DAPO | .643 | .318 | .57 | 45.1 | 77.1 | 48.7 | 33.2 | 7.2 | 64.2 | .412 | .64 | 46.6 | 71.8 | 39.2 | 34.5 | ||
| + CoT-mon. reward | .634 | .557 | .70 | 44.2 | 76.2 | 47.7 | 34.5 | 6.6 | 65.8 | .463 | .69 | 47.2 | 72.4 | 38.8 | 36.1 | ||
| Metric | VIRL | SPA-VL |
|---|---|---|
| Inter-annotator | .74 (400 traces) | |
| Ref.–human agreement | 87.5% | 84% |
| Paired Fleiss’ | .61 | .69 |
| Human | .265 .611 | .356 .636 |
| Human accuracy | .59 .74 | .60 .81 |
| Grounding pref. | +6.25 pp | +7.50 pp |
Appendix figures & tables18 assets
Supplementary material from the paper’s appendix.
Appendix
| Notation | Meaning | Notation | Meaning |
|---|---|---|---|
| RL; VLM; CoT | Reinforcement learning; vision-language model; chain-of-thought | LLM; SAE | Large language model; sparse autoencoder |
| GRPO; DAPO | Group Relative Policy Optimization; RL recipe with decoupled clipping and dynamic sampling | Tame | Targeted Anti-obfuscation with Mechanistic Enforcement |
| A ; B | Monitorable and non-monitorable two-gate reference classes | Correct A predictions, correct B predictions, and number of traces | |
| Trace from method at checkpoint and its reference label | Evaluator and its predicted label for the trace | ||
| Recall on labels A and B, respectively | Headline monitorability metric, | ||
| monitor accuracy | HS; Win | Harm Score (%) on SPA-VL test-harm, lower is better; win rate vs. GPT-4o on test-help, higher is better |
| Parameter | GQA | VIRL-39k | SPA-VL |
|---|---|---|---|
| GRPO Stage | |||
| Learning rate | 1e-6 | 2e-6 | 2e-6 |
| Total steps | 155 | 200 | 200 |
| Group size | 6 | 8 | 8 |
| SAE Training | |||
| Num latents | 4096 | 4096 | 4096 |
| Data | Ref. P/N (G;T) |
|---|---|
| VIRL | 44/36; 54/26 |
| SPA-VL | 49/31; 49/31 |
| Dataset | MSE ( ) | Cosine Sim. | (active) | Dead Latents (%) |
|---|---|---|---|---|
| VIRL | 2.14 | 0.987 | 31.8 | 3.2 |
| SPA-VL | 2.41 | 0.984 | 31.6 | 4.1 |
| Source | Recon. cos. (0/200) | CES | Prec. | Rec. | JSD |
|---|---|---|---|---|---|
| Step 0 | .987/.971 | +.970 | .244 | .109 | .02764 |
| Step 200 | .949/.988 | +.425 | .117 | .113 | .02345 |
| Mixed 50/50 | .973/.971 | +.361 | .077 | .079 | .02449 |
RL for VLMs
Reinforcement learning for vision-language models aligns multimodal reasoning, tool use, and action with task-specific feedback.
See all 472 papers