Search papers, labs, and topics across Lattice.
This paper explores the potential for large language models to learn online from their reasoning traces by converting transient computation into persistent knowledge through lightweight experiential latent memories. The authors demonstrate that traditional In-Context Learning (ICL) struggles with generalization due to the lack of abstraction in individual reasoning traces, but their novel approach using self-generated test-time signals as rewards leads to significant performance improvements. Their method, which utilizes compact modular memories that require minimal parameters and training, achieves competitive results against full-dataset offline training across various mathematical reasoning benchmarks.
Transforming fleeting reasoning traces into durable knowledge can elevate LLM performance beyond traditional training paradigms.
Large language models achieve strong reasoning performance by scaling inference-time compute, yet remain fundamentally stateless, discarding the rich, self-produced reasoning traces generated during this process. We investigate whether models can instead learn online from this experience, converting transient computation (reasoning traces) into persistent reusable knowledge, and without external supervision or access to future data. We show that In-Context Learning (ICL) over raw reasoning traces fails to generalize, reflecting a fundamental limitation of token-level reuse: individual traces lack the abstraction needed for transfer, even after refinement (e.g. self-reflection). In contrast, drawing inspiration from recent works on unsupervised reinforcement learning, we find that lightweight per-instance training with self-generated test-time signals (majority voting) as rewards yields substantial gains, often surpassing full-dataset offline training, motivating a shift from raw traces to learned latent representations. Building on this insight, we propose an online method that distills inference-time compute spent on encountered problems into compact modular latent memories capturing the underlying reasoning structure. These memories are stored and retrieved for future inputs, enabling continual improvement while avoiding catastrophic forgetting through modular design. Importantly, our method is highly efficient, parametrized as extremely lightweight soft prompt memories (~0.001% of model parameters) and trained with only a few gradient steps, yet achieving performance competitive with full parametric updates and offline training. Across challenging mathematical reasoning benchmarks, our approach significantly outperforms zero-shot and raw data ICL baselines, while transferring effectively across datasets.
