arXiv:2605.20613cs.CL2026-05被引 5

用类脑分层循环结构,用极少数据和算力训练出性能媲美大模型的中文语言模型。

HRM-Text: Efficient Pretraining Beyond Scaling

论文配图:HRM-Text: Efficient Pretraining Beyond Scaling
图 1 · 摘自论文原文
  • 采用分层循环架构,分离长期策略与短期执行计算,提升学习效率。
  • 仅用400亿词、1500美元预算训练10亿参数模型,多项任务得分超60%。
  • 适合关注高效预训练、资源受限研究者,推动小团队参与基础模型研发。

当前大语言模型的预训练依赖海量算力与互联网级原始文本,构成基础研究的高门槛。受生物系统多时标处理机制启发,本文提出HRM-Text,以分层循环模型(HRM)替代标准Transformer,将计算解耦为慢速演化的策略层与快速演化的执行层。为稳定深层循环,引入MagicNorm与暖启动深度信用分配机制。训练不使用原始文本,而是基于指令-响应对,采用任务完成目标与PrefixLM掩码。作为高效预训练的实证范例,一个从零训练的10亿参数模型,仅用400亿唯一标记与1500美元预算,在MMLU上达60.7%,ARC-C为81.9%,DROP为82.2%,GSM8K为84.5%,MATH为56.2%。相比主流基线,训练样本减少100-900倍,算力节省96-432倍,性能媲美2-70亿参数开源模型。结果表明,架构与目标协同设计可显著降低算力-性能比,使从头预训练对更广泛研究群体可及。

原文摘要 · Abstract (English)

The current pretraining paradigm for large language models relies on massive compute and internet-scale raw text, creating a significant barrier to foundational research. In contrast, biological systems demonstrate highly sample-efficient learning through multi-timescale processing, such as the functional organization of the frontoparietal loop. Taking this as inspiration, we introduce HRM-Text, which replaces standard Transformers with a Hierarchical Recurrent Model (HRM) that decouples computation into slow-evolving strategic and fast-evolving execution layers. To stabilize this deep recurrence for language modeling, we introduce MagicNorm and warmup deep credit assignment. Furthermore, instead of standard raw-text pretraining, we train exclusively on instruction-response pairs using a task-completion objective and PrefixLM masking. Serving as an empirical existence proof of efficient pretraining, a 1B-parameter HRM-Text model trained from scratch on only 40 billion unique tokens and $1,500 budget achieves 60.7% on MMLU, 81.9% on ARC-C, 82.2% on DROP, 84.5% on GSM8K, and 56.2% on MATH. Despite utilizing roughly 100-900x fewer training tokens and 96-432x less estimated compute than standard baselines, HRM-Text performs competitively with 2-7B parameter open models. These results demonstrate that co-designing architectures and objectives can radically reduce the compute-to-performance ratio, making pretraining from scratch accessible to the broader research community.

高效预训练分层循环小样本建模算力优化

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。