能量模型让大模型自学习推理,无需额外训练即可像人一样思考。
Energy-Based Transformers are Scalable Learners and Thinkers
- 用能量函数验证输入与预测的匹配度,通过梯度下降优化预测结果。
- 训练速度比主流方法快35%,推理时语言任务性能提升29%。
- 适合作为通用推理框架,尤其适合追求高效智能的AI系统开发者。
推理时计算技术(类比人类系统2思维)近年流行,但多数方法存在模态或问题局限性,或需额外监督训练。本文提出能量基变压器(EBTs),一种仅通过无监督学习就能学会推理的新模型。EBTs为输入与候选预测对分配能量值,将预测转化为能量最小化的优化问题。在离散(文本)和连续(视觉)模态上,EBTs的训练扩展速率比主流Transformer++高35%,涵盖数据量、批量大小、参数量、浮点运算次数和深度。推理阶段,其在语言任务上的系统2思维提升比Transformer++高出29%,图像去噪性能优于扩散变压器,且前向传播次数更少。此外,在相同或更差预训练表现下,EBTs在多数下游任务中仍取得更好结果,表明其泛化能力更强。因此,EBTs是提升模型学习与推理能力的新型范式。
原文摘要 · Abstract (English)
Inference-time computation techniques, analogous to human System 2 Thinking, have recently become popular for improving model performances. However, most existing approaches suffer from several limitations: they are modality-specific (e.g., working only in text), problem-specific (e.g., verifiable domains like math and coding), or require additional supervision/training on top of unsupervised pretraining (e.g., verifiers or verifiable rewards). In this paper, we ask the question "Is it possible to generalize these System 2 Thinking approaches, and develop models that learn to think solely from unsupervised learning?" Interestingly, we find the answer is yes, by learning to explicitly verify the compatibility between inputs and candidate-predictions, and then re-framing prediction problems as optimization with respect to this verifier. Specifically, we train Energy-Based Transformers (EBTs) -- a new class of Energy-Based Models (EBMs) -- to assign an energy value to every input and candidate-prediction pair, enabling predictions through gradient descent-based energy minimization until convergence. Across both discrete (text) and continuous (visual) modalities, we find EBTs scale faster than the dominant Transformer++ approach during training, achieving an up to 35% higher scaling rate with respect to data, batch size, parameters, FLOPs, and depth. During inference, EBTs improve performance with System 2 Thinking by 29% more than the Transformer++ on language tasks, and EBTs outperform Diffusion Transformers on image denoising while using fewer forward passes. Further, we find that EBTs achieve better results than existing models on most downstream tasks given the same or worse pretraining performance, suggesting that EBTs generalize better than existing approaches. Consequently, EBTs are a promising new paradigm for scaling both the learning and thinking capabilities of models.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。