Drawing the amortization boundary
Every learned system divides problem solving between choices absorbed into reusable computation during training and choices resolved for the problem at hand. I call the placement of this division the amortization boundary. Choices that recur, including choices whose dependence on the input can be learned reliably, can be amortized. Choices that a future problem may need to reconsider must remain explicitly addressable. Here I use inference to mean problem-specific search, selection, or refinement over choices that learning deliberately left open.
The boundary trades computation for flexibility. If a recurring decision remains open even though the current problem provides no new information about it, inference wastes work. If nearly every decision is fixed during training, a new goal or constraint may have nowhere to enter. The aim is therefore not to maximize inference, but to leave open the choices whose values cannot safely be determined in advance.
A useful analogy is program specialization. Assumptions known in advance can be compiled into an efficient executable. When those assumptions change, however, the system must return to a higher-level program in which the relevant functions and variables are still explicit. A direct neural policy resembles an aggressively specialized executable: it is fast to run, but difficult to adapt when a new problem violates the assumptions compiled into it. An energy function or world model can preserve a higher-level description of solutions or futures, leaving sampling, optimization, or planning to specialize that description for the current problem.
The boundary concerns not only whether a choice remains open, but also the level at which it is represented. A planner need not search over every motor command. It can search over task-dependent states, subgoals, or trajectories while learned skills handle familiar local transitions. A useful representation leaves the distinctions that the current problem may change explicit without forcing inference to reconstruct every recurring detail.
This leads to a practical design procedure:
- Identify the choice that depends on the current goal, constraint, observation, or resource.
- Choose a level of representation at which that choice can be inspected and revised.
- Learn reusable criteria, components, or local procedures around it, rather than a single mapping that fixes it.
- Use inference to resolve it for the current problem.
My PhD thesis developed this view for planning and the composition of learned energy functions. The framework below extends it to the computation and organization of reasoning.
Hierarchical inference
The amortization boundary can be drawn at different scopes. At the narrowest, inference searches for a solution under a fixed model. It may also construct the model itself, plan actions that change the world, choose the procedure used to reason, or decide which reasoners participate. These scopes form a hierarchy because each leaves a broader part of problem solving open for the current problem. Across the hierarchy, the principle is the same: learn reusable structure and use inference to resolve what cannot be fixed in advance. I group the five scopes into two families: the first three construct the problem and its solution, while the final two construct the computation used to solve it. A system can use any subset; they are not mandatory stages.
Constructing the problem and its solution
These scopes leave open what must be constructed for the current problem: a candidate, the model that judges it, or the actions that change the world.Constructing the computation
These scopes leave open how the answer is produced: the next reasoning operation or the participating reasoners and the connections among them.Together, these scopes move from choosing a solution to organizing the system that finds it. Solution inference searches for a candidate under a fixed model. Model composition constructs the model that judges candidates. Action planning uses that model to choose actions and replans as the world changes. Procedure control chooses how these computations are carried out, while reasoner organization chooses who participates and how information moves among them. A system can use any subset of the five scopes.
The scopes classify the choice left open, not the kind of model being used. A world model can therefore appear at several scopes: searching among possible futures under a fixed model is solution inference; combining the model with a new goal or constraint is model composition; and comparing action-conditioned futures to choose what to do is action planning. The same distinction applies to computation. Procedure control changes what the currently assembled system does next: which solver or check to run, what intermediate state to preserve, and when to stop. This remains procedure control even when the system contains specialized modules, as long as the participants and their connections stay fixed. Reasoner organization changes the assembled system itself: which differentiated reasoners participate, what roles they take, or how information moves among them. The scope is determined by what inference can reconfigure, not by whether a component is called a model, solver, or agent.
The hierarchy becomes especially concrete in robotics, where an agent must construct both a desired outcome and a way to reach it while its own actions change the problem. Even a task that is easy to state can leave choices open at several different scopes. A single episode can therefore show how the scopes depend on one another.
Consider a household robot asked to bring a mug of coffee from a crowded kitchen counter to someone at the table. It must lift the mug without disturbing the surrounding dishes, carry it without spilling, and set it within easy reach. The robot may have encountered every object and practiced every skill involved without having faced this particular combination. Solution inference searches for a stable placement. Model composition combines support, uprightness, clearance, and reachability into the model that judges it. Action planning searches for actions whose consequences reach that outcome. Procedure control decides when to inspect, verify, repair, or act, while reasoner organization brings in a perception or grasping specialist when the current system lacks the required knowledge.
Search a learned solution space
Solution inference leaves the learned model fixed while keeping the candidate open. Learning specifies what makes a candidate good without fixing the candidate itself. A direct predictor has to learn both what makes an answer appropriate and a fixed computation for producing it. An energy function or learned evaluator separates these roles: the model scores proposed answers, while a sampler or optimizer constructs one that scores well. The important property is not stochasticity or iteration by itself. The candidate must remain an explicit object of inference—sampled, compared, or revised under the fixed model—rather than being fixed by a single learned input–output pass. An externally supplied proposal can initialize that search; adding a separately addressable constraint or evaluator changes the model and crosses into model composition.
Holding the task model fixed, the robot can search over candidate placements and revise one that is unstable or difficult for the person to reach rather than accept its first proposal. For some problems, learning to judge such a candidate may be easier than learning to produce a correct one in a single pass. A graded criterion can also indicate how the candidate should change. Learning builds a landscape that distinguishes better from worse solutions; inference searches it.
We first explored this separation in Implicit Generation and Generalization in Energy-Based Models. The learned object was a landscape rather than a fixed generator: Langevin dynamics, an iterative stochastic sampler, could search it without imposing a fixed generation order. The sampler could also start from an external proposal or run for different numbers of steps. Learning Iterative Reasoning through Energy Minimization then treated reasoning itself as minimization over a conditional landscape. Its verifier-generator analysis makes the tradeoff explicit: a criterion for recognizing a valid solution can be smaller than a procedure that constructs one, although the resulting search may still be hard.
Subsequent work asks what geometry makes this separation useful in practice. Learning Iterative Reasoning through Energy Diffusion moves through a sequence of landscapes, from broad, smooth basins toward sharper constraints. Reasoning with Sampling applies the same separation to a frozen language model, using its likelihood as the target of an iterative sampler that can revisit earlier choices. These settings differ, but they pose the same question: which learned representation makes revisable inference reliable?
Selected work
- Implicit Generation and Generalization in Energy-Based ModelsOne learned landscape supports generation, refinement, and variable inference effort.
- Learning Iterative Reasoning through Energy MinimizationAnswers are represented as minima, with computation adjustable to problem difficulty.
- Learning Iterative Reasoning through Energy DiffusionA sequence of learned landscapes supports continuous, discrete, and planning problems.
- Energy-Based Transformers are Scalable Learners and ThinkersInput-candidate compatibility is learned across modalities and solved by energy minimization.
- Equilibrium MatchingA time-independent implicit landscape supports optimization-based generation.
- Reasoning with SamplingA frozen language model's likelihood becomes the target of an iterative sampler.
- Generative Modeling via DriftingIterative distribution refinement is moved into training when one-step generation is sufficient.
Compose the model required by the problem
Solution inference searches for a candidate under a fixed model. Model composition changes the model that defines this search. The current task determines which pieces of learned knowledge—such as a relation, goal, physical constraint, or reward—should constrain the answer. Inference selects, composes, or infers these factors. What composition produces is not yet an answer, but a problem-specific criterion or distribution that solution inference can search. A new task need not be reduced to an input for a monolithic predictor; it can become a new inference problem assembled from reusable knowledge.
The task model used to judge those placements need not be learned as a whole for this exact kitchen scene. The robot can construct it by combining learned factors for keeping the mug upright, avoiding nearby dishes, resting it securely on the table, and leaving it within the person's reach. The complete arrangement is globally unfamiliar, but each factor expresses locally familiar structure. Generalization comes from constructing a new global model from these reusable local judgments, not from asking one model to extrapolate across the entire combination at once.
Symbolic systems obtained much of their flexibility by composing reusable predicates and operators. The aim is not to recover a particular symbolic vocabulary, but the operational separation these systems exploited: reusable pieces remain addressable after the task is known, while their continuous semantics can be learned from data. The components need not share an architecture or modality, but they must agree on the object being constructed, how each component can evaluate or guide it, and an inference procedure capable of reconciling their constraints.
Our early visual work began by composing learned concept distributions through conjunction, disjunction, and negation. It then moved from assembling named spatial relations to combining independently trained score models during denoising. Iterative consensus broadened the same idea beyond a common architecture: a generator and independently trained vision, language, or task-specific scorers communicate through a shared candidate.
A compositional system must also learn the pieces it will later reuse. Unsupervised Learning of Compositional Energy Concepts discovers reusable factors without concept labels, while Few-Shot Task Learning through Inverse Generative Modeling infers a new task factor from demonstrations without changing the pretrained generator. Learning therefore need not determine the final task model in advance; it can instead produce a vocabulary of factors whose selection and combination remain open. Once learned, a new factor should compose with earlier factors without retraining them, enlarging the family of tasks the system can construct.
Composition must be grounded and matched to an inference procedure. Set It Up! uses a symbolic relation graph to select learned continuous factors and ground them into object poses. Reduce, Reuse, Recycle shows why the solver cannot be treated as an afterthought: standard reverse diffusion need not realize the product distribution implied by composed scores. Generalizable Reasoning through Compositional Energy Minimization carries the same principle into reasoning, where local learned criteria become reusable verifiers inside a larger problem. The factors, their interface, and the procedure that reconciles them must therefore be designed together.
Decision making is a particularly important instance of model composition. A model of the desired future represents what outcomes matter; action-conditioned dynamics represent what proposed interventions could cause. Both sides can themselves be composed: dynamics from reusable models of objects, agents, and interactions, and objectives from goals, rewards, and constraints. Model composition brings these pieces together to define the decision problem. Action planning then searches for interventions whose predicted consequences reach the desired outcome.
Selected work
- Compositional Visual Generation and Inference with Energy-Based ModelsConcept distributions are composed through logical operations.
- Learning to Compose Visual RelationsNamed relations are represented as reusable energy functions.
- Compositional Visual Generation with Composable Diffusion ModelsSeparate diffusion models are combined during denoising.
- Unsupervised Learning of Compositional Energy ConceptsReusable energy factors are discovered without concept labels.
- Few-Shot Task Learning through Inverse Generative ModelingA new task representation is inferred from demonstrations without changing pretrained model weights.
- Composing Ensembles of Pre-trained Models via Iterative ConsensusPretrained generators and heterogeneous scorers are coupled through iterative feedback.
- Compositional Generative Inverse DesignGlobally unfamiliar designs are constructed from locally familiar constraints.
- Product of Experts for Visual GenerationGenerators, learned rewards, and simulators are reconciled in one target.
- Set It Up!Symbolic relation graphs select learned continuous factors that are grounded into object poses.
- PoCoHeterogeneous robot policies and analytic costs compose at inference time.
- Reduce, Reuse, RecycleSampler design recovers behavior already present in composed distributions.
- Compositional Generative Modeling: A Single Model Is Not All You NeedNew task distributions are programmed from reusable learned models.
- Generalizable Reasoning through Compositional Energy MinimizationSmall learned energies compose into larger reasoning problems and new constraints.
Plan through possible futures
Composing a desired outcome with a model of what can happen defines a decision problem, but it does not yet determine the actions an agent should take to reach the outcome. In the coffee task, the robot must still choose a grasp, lift clear of nearby dishes, carry the mug without spilling, and place it at the selected location. Through planning, we infer the action sequence to execute. The planner compares the predicted consequences of possible actions until it finds an action sequence expected to reach the desired state.
A direct policy instead compiles the use of dynamics and objectives into a mapping from observations and goals to actions. This makes action selection efficient, but commits in advance to how the agent will use its knowledge. Such a policy can generalize when a new situation is covered by regularities captured in the learned mapping. When the situation demands a combination the mapping does not support, however, the policy itself exposes neither queryable dynamics nor a search process with which to construct a new behavior. The distinction is therefore not policy versus planning as permanent alternatives, but which decisions should be compiled and which must remain open.
Planning does not always operate directly on individual motor commands. It can search over subgoals, intermediate states, or complete trajectories, depending on what must be decided for the current task. Model-Based Planning with Energy-Based Models inferred intermediate states under learned dynamics, allowing a planner to route around obstacles not encountered during training. Planning with Diffusion for Flexible Behavior Synthesis instead represented complete state-action trajectories as a generative solution space, so new rewards, endpoints, or constraints could guide the same learned model. In both cases, learning captures what recurs across tasks while search resolves the route required by the task at hand.
Predictive control repeats this search as the world changes. Flexible Locomotion Learning with Diffusion Model Predictive Control combines a learned trajectory prior with new rewards and constraints, while Model-Based Diffusion Sampling for Predictive Control in Offline Decision Making composes a task planner with action-conditioned dynamics so that task-aligned trajectories remain feasible.
For longer-horizon tasks, video provides an intermediate representation between a high-level goal and low-level control. It can represent what should happen in the scene while leaving an embodiment-specific model to determine how a particular robot makes it happen. Learning Universal Policies via Text-Guided Video Generation learns a task-level visual trajectory that can serve as a proposal from which actions are extracted. Video Language Planning shows how search can compare and connect such proposals into longer-horizon plans. Video is useful because it exposes an intermediate future that subsequent search or control can compare, extend, reject, or re-ground before execution.
Different models can also divide the problem across levels of abstraction. Compositional Foundation Models for Hierarchical Planning uses language to propose high-level steps, video to ground them in a scene, and an action model to test their physical executability. Planning becomes a search for a course of action supported at every level. This is how planning grows out of model composition: the components provide complementary views of the same evolving future.
World Action Planner makes the separation between action proposal, consequence prediction, and search explicit. A vision-language model proposes actions, an action-conditioned world model predicts their consequences, and global correction and local search revise the proposal for the current task. The same verification–generation separation that motivates solution inference therefore reappears over time: judging whether an imagined consequence makes progress can be easier than generating the entire successful behavior at once. Perceptual realism alone is not enough; a planning world model must preserve the distinctions that change a decision, including controllability, contact, object persistence, and progress toward the goal.
Planning and policy can therefore occupy different points on a moving amortization boundary. Unfamiliar behavior can initially be constructed through search; behavior that recurs can later be compiled into a reusable policy or skill. Once the robot begins executing a plan, however, the world may still depart from its predictions: the mug may shift in its grasp or a dish may block the route. The planner can use the resulting state to search again rather than continue an obsolete action sequence. Replanning changes the action sequence, but it may still use the same search procedure each time.
Selected work
- Model-Based Planning with Energy-Based ModelsIntermediate state trajectories are inferred through learned dynamics.
- Planning with Diffusion for Flexible Behavior SynthesisWhole trajectories are generated and guided as a planning representation.
- Compositional Foundation Models for Hierarchical PlanningLanguage, video, and action models are composed across abstraction levels.
- Inference-Time Enhancement of Generative Robot Policies via Predictive World ModelingA frozen policy proposes actions that a predictive model ranks or refines.
- DriftWorld: Fast World Modeling through DriftingOne-step action-conditioned rollouts let a planner compare many policy proposals.
- Test-Time Graph Search for Goal-Conditioned Reinforcement LearningGraph search turns a frozen local policy into a long-horizon planning tool.
- Flexible Locomotion Learning with Diffusion Model Predictive ControlA learned trajectory prior is optimized under new rewards and projected constraints.
- Model-Based Diffusion Sampling for Predictive Control in Offline Decision MakingA diffusion planner and dynamics model are composed during predictive control.
- Learning Universal Policies via Text-Guided Video GenerationLanguage goals become visual plans from which actions are extracted.
- Video Language PlanningTree search composes short predicted videos into long-horizon visual plans.
- Large Video PlannerLarge-scale video pretraining supports zero-shot plans retargeted to robots.
- World Action Models are Zero-shot Policies (DreamZero)Joint video–action prediction provides a closed-loop proposal policy at the compiled end of the planning boundary.
- World Action Planner: Generalizable Decision-Making with Action-Conditioned World ModelsAction proposals are revised through search over imagined consequences.
- World-in-WorldClosed-loop evaluation separates visual realism from controllability and decision utility.
- Structured 4D Latent Predictive Model for Robot PlanningStructured three-dimensional latents preserve geometry for prediction and control.
- Self-Improving Loops for Visual Robotic PlanningTrajectories produced by a visual planner are consolidated into improved behavior.
- COMBOPer-agent action conditions compose into a video world model whose predicted consequences guide cooperative planning under partial observation.
Control the reasoning process
The first three scopes leave an object of inference open—a candidate, a task model, or a course of action—but may still run a fixed outer procedure. The same construction principle can be applied to that procedure. The need to replan, for example, raises a further question: what part of the current computation should be preserved, and what should run next? Procedure control leaves this next reasoning operation open. It does not require inventing a new algorithm at test time; the outer controller may remain fixed. What remains unresolved is which member of an explicit repertoire—sample, expand, verify, repair, switch methods, or stop—should be applied next to the current reasoning state, and how much computation it deserves. If every problem receives the same solver for a preset number of steps, only the object is being inferred; the procedure remains fixed.
Temporal Backtracking Search for Test-time Generative Video Reasoning makes this distinction concrete for generative video. The video model is trained to resume from an arbitrary valid prefix. After each rollout, a process verifier localizes the first failure, identifies a safe restart point, and ranks the surviving prefixes. The outer generate–verify–restart loop remains fixed, but evidence from the current rollout determines where generation resumes, which prefix receives the next computation, and whether search stops. This kind of procedure control is possible because the model exposes the intermediate state that the controller needs to preserve and revise.
Procedure control must also decide how much of an existing solution to preserve and how much computation the situation permits. Adaptive Online Replanning with Diffusion Models uses evidence from execution to choose whether a plan should be retained, repaired, or regenerated. Win Fast or Lose Slow dynamically selects model size and quantization according to real-time demands, showing that the best procedure depends on the cost of delay as well as answer quality. A controller therefore needs signals about both what can be recovered and how much time the current problem allows.
For the robot, this means that a poor view of the mug or a slight shift in its grasp need not trigger a complete restart. The controller may move the camera, resample the grasp, check that the mug is still upright, repair only the affected part of the plan, switch methods, or stop deliberating and continue.
Procedure control assumes that the sources of knowledge available to the system are already fixed. When the next computation also depends on which differentiated generator, verifier, or specialist participates and how their outputs are connected, the organization itself remains open.
Selected work
- Adaptive Online Replanning with Diffusion ModelsPlan likelihood determines whether a plan is reused, repaired, or regenerated.
- Temporal Backtracking Search for Test-time Generative Video ReasoningVerifier evidence determines which valid prefix is preserved, expanded, or abandoned.
- Inference-Time Scaling of Diffusion Models through Classical SearchAdaptive backtracking allocates computation across diffusion paths.
- Win Fast or Lose SlowDynamic model and precision choices balance answer quality against latency.
Organize multiple reasoners
Learning can create a population of reasoners with different knowledge, tools, local state, and procedures. Reasoner organization leaves open how those members are assembled for the current problem: who participates, what roles they play, and how their outputs are connected.
Procedure control asks what operation the available system should run next. Reasoner organization asks which differentiated sources should contribute and how their outputs should be connected. Running the same verifier again changes the procedure; adding or replacing a verifier changes the organization because it changes the evidence brought to bear. Running several copies of the same model and voting does not cross this boundary, nor does packaging a scorer or solver as an “agent.”
Changing the procedure is not always enough: if the robot encounters an unfamiliar, fragile mug and lacks a reliable model of safe contact, another search step cannot supply that knowledge. It can consult a grasp specialist, pass the proposed contact geometry to the motion planner, and invoke a perception specialist if additional visual evidence is needed.
Earlier work established the value of multiple reasoners while keeping their organization fixed. Multiagent Debate made identical model instances expose and challenge one another's reasoning through a fixed interaction pattern. Multi-Agent Verification combined a fixed, domain-specific set of aspect verifiers whose different prompts, strategies, or base models supplied complementary judgments. These systems expose organizational variables—who participates, what each reasoner checks, and how their outputs are combined—even though those choices remain fixed within a deployment.
SLM-MUX makes group composition an explicit design variable. It searches before deployment for a complementary subset of small language models. The selected group remains fixed across test problems within a benchmark, but the result shows why choosing complementary participants can matter more than simply adding more interaction.
Multiagent Finetuning asks how learning can create useful differences among reasoners. Starting from the same base model, separate generation and critic agents are fine-tuned on distinct data produced through multiagent interaction. This preserves diverse reasoning chains across repeated rounds of self-improvement. Its participant set and debate protocol remain fixed at test time; its contribution here is to learn the differentiated population that organization can later draw upon.
Economy of Minds makes the organization itself adaptive. At each step, eligible agents compete for the right to act, producing a problem-dependent sequence of specialists without a central scheduler. Over longer timescales, wealth, bankruptcy, and mutation reshape the population according to the downstream value its members create.
Selected work
- Improving Factuality and Reasoning in Language Models through Multiagent DebateIdentical model instances reason through a fixed debate protocol, establishing interaction as a reasoning substrate.
- Multi-Agent VerificationA fixed, domain-specific set of aspect verifiers combines complementary judgments at test time.
- Multiagent Finetuning: Self Improvement with Diverse Reasoning ChainsSeparate generation and critic agents are trained on different interaction data to preserve diverse reasoning chains.
- SLM-MUX: Orchestrating Small Language Models for ReasoningPre-deployment search identifies a complementary model subset, showing why participant choice matters.
- Economy of MindsAgents compete to act at each step while economic selection reshapes the population.
Why embodiment is central
The ultimate goal of AI is to build systems that can act reliably in the physical world. Robotics is therefore the natural integration test for generalization by construction: the entire hierarchy of inference must work together while every action changes the next problem. In the coffee task, an unreachable placement leaves planning with no feasible route, a failed grasp changes the next observation, and persistent uncertainty may require a different procedure or specialist. A robot cannot infer everything from scratch before every action, but it also cannot compile away choices that an unfamiliar goal, constraint, or failure may require it to reconsider. The amortization boundary must be placed across the full system so that action remains both timely and adaptable.
Generalization as construction
Logic- and planning-based symbolic systems preserved a separation between knowledge and inference, but relied on representations built by hand. Neural models made representations learnable from data, but often compiled away flexibility in how those representations could be used. The synthesis proposed here is to learn representations that preserve the choices inference may need to make as the current problem unfolds.
A problem can be globally unfamiliar even when the structure needed to solve it is locally familiar. If learning preserves that structure in reusable form, inference can assemble it for a problem whose exact solution was never encountered during training. Across the five scopes, what gets constructed expands: a candidate, the model defining the problem, a course of action, a problem-specific computation, or an organization of reasoners.
The boundary between training and inference is not fixed. Computation that repeatedly proves useful at test time can later be absorbed into training. Generative Modeling via Drifting illustrates this idea by learning to perform in one step what would otherwise require iterative sampling. Inference can therefore handle unfamiliar problems first, while learning later makes recurring computation fast.
In conclusion, generalization by construction is not an argument for leaving every decision open. It is an argument for learning representations that keep the right decisions revisable until the goal, world state, resources, and available expertise are known.