Reasoning Fine-Tuning Induces Persistent Latent Policy States

· Source: Computation and Language · Field: Technology & Digital — Artificial Intelligence & Machine Learning, Natural Language Processing · Depth: Expert, quick

Summary

A study investigates the internal changes in reasoning-specialized language models, specifically whether reasoning fine-tuning improves local token-level competence or globally reorganizes inference structure. Researchers modeled Chain-of-Thought reasoning as a switching dynamical system (SDS), combining time-aware contrastive representation learning with discrete regime discovery to recover latent policies from activation trajectories. Across four benchmarks and model scales from 1.5B to 32B parameters, reasoning-fine-tuned models exhibited richer latent-policy organization, characterized by more differentiated transition structure and changes in state utilization, persistence, and mixing. The recovered regimes showed functional specialization aligned with distinct reasoning stages. Causal interventions confirmed these regimes are functionally meaningful, with state-swap ablations reducing predictive fit and transplanting dynamics improving base model performance. Furthermore, SDS-guided pruning of failure-prone reasoning prefixes outperformed self-consistency in 11 of 12 settings, achieving gains up to 12.5 percentage points. This suggests reasoning fine-tuning globally reorganizes latent dynamics.

Key takeaway

For NLP Engineers optimizing large language models for complex reasoning tasks, understanding that fine-tuning globally reorganizes latent dynamics is crucial. This insight suggests you should explore switching dynamical systems (SDS) for mechanistic analysis, rather than solely focusing on token-level competence. Consider applying SDS-guided pruning to improve inference reliability, potentially achieving significant performance gains, such as the observed 12.5 percentage points, by identifying and mitigating failure-prone reasoning paths.

Key insights

Reasoning fine-tuning globally reorganizes language model latent dynamics, enabling process-level control and performance gains.

Principles

Method

Model Chain-of-Thought reasoning as a switching dynamical system (SDS), combining time-aware contrastive representation learning with discrete regime discovery to recover latent policies from activation trajectories.

In practice

Topics

Best for: Research Scientist, AI Scientist, NLP Engineer, Machine Learning Engineer

Related on AIssential

Open in AIssential →

Editorial summary, takeaway, and curation by AIssential. Original article published by Computation and Language.