TGViewer
Machine learning books and papers Machine learning books and papers @machine_learn · 24.2K subscribers
Post #3809 3.87K
📃 Energy-Based Transformers are Scalable Learners and Thinkers

Inference-time computation techniques, analogous to human System 2 Thinking, have recently become popular for improving model performances. 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.


@Machine_learn
More from @machine_learn
  1. Oct 12, 2026🔥 Prism: Dynamic Sparse Attention for Native 2K Joint Video-Audio Generation Model Traini…
  2. Oct 12, 2026🔥 Spatial Memory Intelligence: Endowing World Models with Understanding-Driven Long-Term…
  3. Oct 5, 2026Machine learning books and papers pinned «High-Impact Research Paper – MedicalRec / GROKRe…
  4. Oct 5, 2026High-Impact Research Paper – MedicalRec / GROKRec Title: MedicalRec Price: $1,000 USD Look…
  5. Oct 5, 2026🔥 FrameMorrow: Future-guided Frame Selection with Prospective Tokens for Long-Horizon Vid…
  6. Oct 5, 2026🔥 TAPNext: Tracking Any Point (TAP) as Next Token Prediction 🔗 Links: GitHub: https://gi…
Threads Profile ViewerView any public Threads profile without an account.Open ThreadLook →Writing with AI? Make it sound human.Metric37 rewrites AI drafts so they read naturally. Free AI detector, 1,500 words free.Try Metric37 →