Artificial Intelligence · 01.09.2026, 11:18 UTC
Better World Models Can Lead to Better Post-Training Performance
| Schweregrad | info |
|---|---|
| Kategorie | Artificial Intelligence |
| Quelle | arXiv cs.AI ↗ |
| Veröffentlicht | 01.09.2026 UTC |
Sicherheitsmeldung mit Schweregrad noch nicht bewertet. Technische Details im Tab „Originaltext“; empfohlene Schritte in der Checkliste.
arXiv:2512.03400v2 Announce Type: replace-cross Abstract: We study how explicit world-modeling objectives affect the internal representations and downstream capability of Transformers, using Rubik's Cubes as our training domain. We ask: (1) how does explicitly pretraining a world model affect a model's latent representations, (2) how does world-model quality affect post-training performance, and (3) how should a finite data budget be split between pretraining and task-specific fine-tuning? We compare standard action-prediction fine-tuning with two strategies that add explicit state-prediction supervision: state pretraining followed by fine-tuning, and joint action and state training. We measure task accuracy after Group Relative Policy Optimization (GRPO). We further find that explicit world-modeling yields better representations in terms of higher probing accuracy and steerability of the model, and that better representations yield larger gains from GRPO, especially on harder cube states. Finally, when the fine-tuning budget is held fixed, probe accuracy strongly predicts the GRPO improvement. Under a fixed total data budget, accuracy is maximized by allocating only a small fraction to pretraining.
Maßnahmen
⬇ Als MarkdownVerwandte Beiträge
- info Relevance as a Vulnerability: How Web Retrieval Degrades Safety Alignment in LLM Agents
- info OISD: On-Policy Internal Self-Distillation of Language Models
- info Skill-Conditioned Gated Self-Distillation for LLM Reasoning
- info Reverse Probing: Supervised Token-level Uncertainty Quantification for Large Language Models in Clinical Text