Less is More: Recursive Reasoning with Tiny Networks

AK's Picks LLM Latent Reasoning Agent Agent Planning ML NN LN
分层推理模型(HRM)是一种新颖的方法,使用两个小型神经网络以不同频率进行递归。这种受生物启发的方法在数独、迷宫和ARC-AGI等复杂谜题任务上超越了大语言模型(LLMs),而其训练仅依赖于小型模型(2700万参数)和少量数据(约1000个示例)。HRM在利用小型网络解决难题方面展现出巨大潜力,但目前对其理解尚不充分,且可能存在优化不足的问题。我们提出了微型递归模型(TRM),这是一种更为简单的递归推理方法,仅使用一个仅有两层的微型网络,却实现了远超HRM的泛化能力。TRM仅有700万参数,在ARC-AGI-1上达到了45%的测试准确率,在ARC-AGI-2上达到8%,表现优于大多数大语言模型(例如Deepseek R1、o3-mini、Gemini 2.5 Pro),而参数量却不到这些模型的0.01%。
Hierarchical Reasoning Model (HRM) is a novel approach using two small neural networks recursing at different frequencies. This biologically inspired method beats Large Language models (LLMs) on hard puzzle tasks such as Sudoku, Maze, and ARC-AGI while trained with small models (27M parameters) on small data (around 1000 examples). HRM holds great promise for solving hard problems with small networks, but it is not yet well understood and may be suboptimal. We propose Tiny Recursive Model (TRM), a much simpler recursive reasoning approach that achieves significantly higher generalization than HRM, while using a single tiny network with only 2 layers. With only 7M parameters, TRM obtains 45% test-accuracy on ARC-AGI-1 and 8% on ARC-AGI-2, higher than most LLMs (e.g., Deepseek R1, o3-mini, Gemini 2.5 Pro) with less than 0.01% of the parameters.
许愿