Can Generative Models Finally Be Trained End-to-End? The Core Is a For Loop

marsbitPublished on 2026-08-03Last updated on 2026-08-03

Abstract

This article introduces a novel training paradigm for generative models called Explorative Modeling (XM), which enables true end-to-end training. Traditionally, powerful generative models like autoregressive and diffusion models are not trained end-to-end. They are trained to predict a single small step but require iterative multi-step sampling for inference. This "exposure bias" leads to error accumulation and limits performance. The core challenge XM addresses is "mode blurring." In generative tasks, a single input (e.g., "generate a dog") corresponds to many valid outputs (multiple modes). Standard training objectives like reconstruction loss force the model to average these modes, producing unrealistic, blurry outputs. To avoid this, existing models break generation into many small, almost deterministic steps, sacrificing end-to-end training. XM tackles this by restructuring the training loop itself. Its key insight is to amplify "generative expressivity." For each training input, instead of generating one sample, the model generates K candidate outputs. Only the candidate closest to the real data is used for computing the loss and updating the model via backpropagation. This simple "best-of-K" mechanism is implemented as a short for-loop. By exploring multiple possibilities, the model learns to distribute its guesses across different modes rather than collapsing to their uninformative average. The paper demonstrates that "exploration" acts as a new, powerful scaling a...

In 2012, AlexNet decisively ended an era with an overwhelming victory. Before that, image recognition relied on manually designed, multi-stage feature extraction pipelines. AlexNet proved something that would be repeatedly validated later: handing the entire task to the model to learn end-to-end almost always outperforms human-designed, staged workflows.

From image classification to object detection to image segmentation, behind every leap in deep learning was the same mantra: let it learn everything in one go.

There has been only one consistent exception: generative models.

Today's strongest, most scalable generative models (whether autoregressive or diffusion models) are not end-to-end.

During training, they only learn to predict "one small step," but during inference, they have to recurrently unfold this step hundreds or thousands of times.

The sampling methods used for training and inference are not the same. This discrepancy leads to an old problem: errors from one step are fed into the next, the input gradually drifts away from the distribution seen during training, and errors accumulate layer by layer. Academically, this is called "exposure bias."

In other words, the core tenet of deep learning—"end-to-end is better"—has for over a decade failed to fully materialize in generative modeling.

Recently, however, a paper from UIUC and Harvard University attempts to complete this final piece of the puzzle.

The authors named this new paradigm Explorative Modeling, abbreviated as XM. Its idea is so simple it borders on naive, yet points to a bold conclusion: Beyond parameters and data, generative models actually have a third axis that can be scaled.

Project Website: https://explorative-modeling.github.io

Paper: https://arxiv.org/abs/2607.27372

Code Repository: https://github.com/alexiglad/XM

The Root Problem: Models Only Know How to "Take the Average"

To understand what this paper solves, one must first understand why generation is hard.

In ordinary supervised learning (e.g., classification), each input essentially has only one correct answer; the model learns a deterministic mapping.

But generation is different. When you ask a model to "generate a dog," there can be infinitely many correct answers. These valid outputs are the many modes (i.e., distinct peaks) within the data distribution. Generation is difficult precisely because it must capture all these modes simultaneously.

The trouble is, mainstream generative models are trained with reconstruction losses (e.g., squared error). When an input is paired with many different valid targets at random, the optimal solution for a reconstruction loss is the average of those targets. For most real-world data, this average does not lie on the data manifold but falls between modes, resembling none of them.

A figure in the paper illustrates this clearly: when asked to regress directly end-to-end without any tricks, three clusters of points would be predicted as a single point in the middle, a photo of a dog would blur into a smudge, and a sentence would degenerate into endless repetitions of "the." This is "mode blurring," where the optimal solution is precisely the answer least resembling real data.

How do existing models circumvent this? By breaking the "generation" process into tiny pieces. Autoregressive models predict only one element at a time; diffusion models remove a little noise at each step. Each small-step target is sliced until it contains essentially a single mode, preventing the reconstruction loss from averaging.

This strategy of "splitting the generative process" is precisely why diffusion and autoregressive models can produce high-quality samples, but it is also why they cannot be end-to-end.

The authors thus pose a crucial question: A generative model has only two things that can be broken down—how it generates and how it trains.

Since breaking the generation process destroys end-to-end capability, why not break the training instead?

A For Loop: The Entire Essence of Explorative Modeling

Explorative Modeling breaks down the training loop itself.

Its mechanism can be stated in one sentence: In each training step, instead of generating one sample to fit the target, the model generates K candidates and then selects only the one closest to the real data for training and gradient backpropagation. The paper implements this as a 3-5 line for loop, so simple it's almost suspicious (Algorithm 1).

Why does this solve mode blurring?

Think of a real-life analogy: guessing dart landing positions. If you are only allowed one guess, your optimal strategy is to guess the average position of all darts—but that's often a spot on the board where few darts actually land. However, if you are allowed K guesses and are scored only on your closest guess, the optimal strategy immediately changes: you would spread your guesses, letting each cover a different cluster of landing points.

The model behaves similarly. When allowed to explore K candidates, different input noises will "claim" different modes, instead of all crowding towards the middle to take the average. The number of explorations directly determines how many modes the model can stably capture.

The authors name this long-overlooked capability "generative expressivity," noting that it is determined by the training objective itself—no matter how much you scale parameters and data, it won't increase on its own.

This also explains a phenomenon long observed in the field: why today's best models are so reliant on "guidance" techniques.

Classifier-free guidance essentially "pushes" the prediction away from that blurry average. But if the model itself weren't blurry, why push it? The usefulness of guidance stems precisely from the lingering disease of mode blurring.

The paper also introduces Forward and Reverse exploration directions. Forward fixes a real target and searches among its own generations for the closest one, favoring "recall" (covering all modes). Reverse fixes a generation and searches among real data for the closest one, favoring "precision," and incurs almost no additional computational cost, at the risk of collapsing to a few modes. The two are complementary and can be used in combination.

The Third Axis: Greater Gains as You Scale

The most substantial conclusion of this paper is that it validates "exploration" as a genuine scaling axis.

The authors applied exploration to diffusion/flow models, Jumpy models, and even masked diffusion language models, observing consistent, monotonic performance improvements across image, video, and language modalities. More importantly, the trend of gains with scale: the bigger you go, the greater the benefit.

The numbers reported in the paper show: as data scale increases, gains from exploration rise from 7% to 36%; as model size grows, from 13% to 23%; when compute is tripled, efficiency gains more than double.

Specifically in efficiency, exploration improved FLOP efficiency by 4.1x, sample efficiency by 6.2x, and parameter efficiency by 47%. In image generation, it pushed the current strongest RAE formulation to an unguided FID of 1.43 on ImageNet, nearing the state-of-the-art.

A Large model exploring 5 modes can even outperform an XLarge model with 47% more parameters but no exploration.

The implications of this trend are significant. The authors explain: at small scale, models are primarily bottlenecked by parameters and data; generative expressivity isn't the limiting factor yet. But once parameters and data are scaled to the point where they are "no longer the limit," generative expressivity increasingly becomes the real bottleneck. And that is exactly what exploration directly scales.

Considering that true foundation model training today uses roughly four orders of magnitude more compute than the largest experiment in this paper, the authors believe the reported numbers likely represent only a lower bound for the gains at larger scales.

Truly End-to-End Generation

If exploration is pushed to the extreme, what happens? The answer circles back to the opening suspense: Generative models can finally be end-to-end.

The authors used XM as a standalone end-to-end model for robot control tasks. In Behavior Cloning, their Explorative Policy matched or even surpassed Diffusion Policy (which requires 100 forward passes) using only a single network forward pass. In goal-directed world modeling, the Explorative World Model achieved better average performance than Diffuser, using 16 to 256 times less inference compute.

The source of this gap is clear: diffusion models trade hundreds of inference steps for expressivity, while end-to-end XM shifts this cost to exploration during training, allowing inference with just one forward pass. "Handling multi-modality"—the same thing—can be done either by breaking it down slowly during inference or by exploring it thoroughly once during training; this paper chooses the latter.

About the Authors

The first author of this paper, Alexi Gladstone, is not a newcomer. As recently as July 2025, his work on Energy-Based Transformers (EBT) sparked considerable discussion on social platforms.

That work claimed to "beat" the scaling curve of standard feedforward Transformers on multiple dimensions for the first time and attempted to generalize "System 2 thinking" to arbitrary modes. See the Machine Heart report: "New Paradigm Arrived! New Energy Model Breaks Transformer++ Scaling Limits, Training Scaling Rate 35% Faster."

Explorative Modeling aligns with his consistent line of thinking: questioning "whether a model can, through some form of search or exploration, achieve what a single forward pass cannot." Furthermore, this paper builds upon another theory, Mode Forcing, by him and his collaborators (Yilun Du and Heng Ji). The paper candidly states that because of that prior theoretical foundation, most of XM's results were theoretically predicted first and then experimentally verified, which is relatively rare in the deep learning field, where the norm is often "run experiments first, explain later."

The authors are also frank about limitations: the best-of-K idea itself is not new and has been tried many times before; their real contribution is clarifying what this simple loop actually does: it directly amplifies generative expressivity without breaking down the generative process.

Additionally, autoregressive language models remain the toughest challenge, and pure end-to-end Forward XM is still too expensive for highly multi-modal distributions (like image generation), leaving room for future work.

For over a decade, we've grown accustomed to tuning generative models with two knobs: making them bigger and feeding them more data.

The third knob proposed in this paper is remarkably simple: let the model guess multiple times and keep only the best one. But if the trend it reveals holds true, then as scale continues to expand, the first two knobs will eventually be maxed out, while the third has only just begun to turn.

Reference Links

https://x.com/AlexiGlad/status/2083230922196107288

This article is from the WeChat public account "Machine Heart" (ID: almosthuman2014), author: Panda

Trending Cryptos

Related Questions

QWhat is the core idea behind 'Explorative Modeling (XM)' as described in the article?

AThe core idea is to modify the training loop by having the model generate K candidate samples for each training step and then backpropagate gradients only from the candidate closest to the real data. This simple 'best-of-K' for-loop strategy helps the model capture multiple modes of the data distribution, solving the 'mode blurring' problem that plagues direct end-to-end generative modeling.

QAccording to the article, what is the 'third axis' for scaling generative models introduced by Explorative Modeling?

AThe 'third axis' is 'generative expressivity,' which is the inherent ability of a model to capture multiple modes in a data distribution. The paper argues that this expressivity is determined by the training objective itself and does not automatically scale with more parameters or data. Explorative Modeling directly amplifies this axis by increasing the exploration factor K during training.

QWhat problem do current powerful generative models (like autoregressive and diffusion models) have that prevents them from being truly end-to-end?

ACurrent powerful generative models are not truly end-to-end because they train to predict only 'one small step' (e.g., the next token or a denoising step), but during inference, they must unroll this step hundreds or thousands of times iteratively. This creates a mismatch between training and inference distributions, leading to the 'exposure bias' problem where errors accumulate step-by-step.

QWhat key advantage does an end-to-end Explorative Model (XM) demonstrate in the robotics control tasks mentioned in the article?

AIn robotics control tasks like Behavior Cloning, the Explorative Policy, using a single network forward pass, matched or exceeded the performance of Diffusion Policy, which required 100 forward passes. For goal-directed world modeling, the Explorative World Model achieved better average performance using 16 to 256 times less inference compute than a Diffuser model, highlighting its efficiency gains from shifting the computational cost from inference to training-time exploration.

QWho is the first author of the Explorative Modeling paper, and what was their related prior work mentioned in the article?

AThe first author is Alexi Gladstone. His related prior work is 'Energy-Based Transformers (EBT),' which claimed to surpass the scaling curves of standard feedforward Transformers across multiple dimensions. His research consistently explores how models can use search or exploration mechanisms to achieve what a single forward pass cannot.

Related Reads

Rubin Ultra Makes Major Cuts, Even Nvidia Can't Handle Memory Price Hikes?

NVIDIA's Rubin Ultra, the top-tier variant of the newly announced Rubin AI accelerators, has reportedly seen significant specification downgrades, according to an industry report from SemiAnalysis. Initially designed with four compute dies (4-die), the Rubin Ultra is now said to be reduced to a 2-die design. Key changes highlighted in the report include: * **No increase in peak theoretical compute performance**, remaining at 35 PFLOPs like the standard Rubin. * **Severe reduction in memory capacity** to 192GB using 8-Hi HBM stacks, which is less than the standard Rubin's 288GB using 12-Hi stacks. * **Negligible memory bandwidth improvement** of only 1 TB/s. * **Slightly higher chip-level power consumption**. * The **primary upgrade is a massive increase in scale-up interconnect capacity**, supporting connections for up to 576 GPUs via NVLink, compared to 72 for the standard Rubin. The report suggests the redesign is primarily a cost-optimization move driven by the sharp rise in HBM (High-Bandwidth Memory) prices. By reducing the expensive HBM content and shifting investment towards enhanced system-scale networking, NVIDIA aims to maintain the platform's value for large-scale AI training clusters while managing soaring material costs. The news reportedly triggered a sell-off in South Korean memory stocks, with SK Hynix and Samsung shares falling around 8%, as markets grew concerned that NVIDIA—a major HBM buyer—might be reducing its reliance on high-capacity memory, potentially capping future pricing power for memory makers.

Odaily星球日报7m ago

Rubin Ultra Makes Major Cuts, Even Nvidia Can't Handle Memory Price Hikes?

Odaily星球日报7m ago

Annual Salary of Millions Competing for Electricians, Meta Rushes to Open Its Own Technical School

The AI boom is facing an unexpected bottleneck: a severe shortage of skilled construction workers and electricians. As tech giants like Meta, OpenAI, and Alphabet race to build massive data centers—such as OpenAI's $16 billion "Stargate" project—they are hitting a critical labor wall. The U.S. needs an estimated 130,000 more electricians, 240,000 construction workers, and 150,000 supervisors by 2030 for AI infrastructure alone, but tens of thousands of electrician jobs go unfilled each year. While AI companies offer high premiums, with electricians earning up to $280,000 annually, worker scarcity still causes massive losses—delays on a single project can cost $14.2 million per month. The complexity of building AI data centers, which require immense power (equivalent to powering hundreds of thousands of homes), sophisticated electrical systems, and advanced liquid cooling solutions, demands highly skilled technicians who are in short supply. To combat this, companies are investing heavily in training. Meta has committed $115 million to a free training school offering tuition, housing, and stipends, targeting 5,000 new workers. OpenAI is partnering with unions to secure skilled labor. These efforts are paying off, with a significant rise in Gen Z interest in trade schools over college. However, the power demands are staggering. AI data centers are driving a rapid surge in electricity consumption, projected to account for up to 12% of U.S. power use by 2028 and raising costs for consumers. Furthermore, the construction boom is project-based, leading to a potential future glut of trained workers once building peaks, which could depress wages industry-wide. The race for AI supremacy now depends as much on skilled hands as on advanced chips.

marsbit2h ago

Annual Salary of Millions Competing for Electricians, Meta Rushes to Open Its Own Technical School

marsbit2h ago

Trading

Spot

Hot Articles

How to Buy CORE

Welcome to HTX.com! We've made purchasing CORE (CORE) simple and convenient. Follow our step-by-step guide to embark on your crypto journey.Step 1: Create Your HTX AccountUse your email or phone number to sign up for a free account on HTX. Experience a hassle-free registration journey and unlock all features.Get My AccountStep 2: Go to Buy Crypto and Choose Your Payment MethodCredit/Debit Card: Use your Visa or Mastercard to buy CORE (CORE) instantly.Balance: Use funds from your HTX account balance to trade seamlessly.Third Parties: We've added popular payment methods such as Google Pay and Apple Pay to enhance convenience.P2P: Trade directly with other users on HTX.Over-the-Counter (OTC): We offer tailor-made services and competitive exchange rates for traders.Step 3: Store Your CORE (CORE)After purchasing your CORE (CORE), store it in your HTX account. Alternatively, you can send it elsewhere via blockchain transfer or use it to trade other cryptocurrencies.Step 4: Trade CORE (CORE)Easily trade CORE (CORE) on HTX's spot market. Simply access your account, select your trading pair, execute your trades, and monitor in real-time. We offer a user-friendly experience for both beginners and seasoned traders.

6.0k Total ViewsPublished 2024.03.29Updated 2026.06.02

How to Buy CORE

Discussions

Welcome to the HTX Community. Here, you can stay informed about the latest platform developments and gain access to professional market insights. Users' opinions on the price of CORE (CORE) are presented below.

活动图片