Higher Embedding Dimension Creates a Stronger World Model for a Simple Sorting Task

Brady Bhalla, Honglu Fan, Nancy Chen +1

The paper studies how the size of embedding vectors influences the development of internal world models in transformers trained via reinforcement learning to perform bubble‑sort‑st…