Hierarchical, Hybrid, and Omnimodal Models › Multiple Temporal and Spatial Scales

Michael BrenndoerferJune 30, 202613 min read
World Models HandbookAll chapters
Part 5 of 12World-Model Architectures
Chapter 30Hierarchical, Hybrid, and Omnimodal Models › Multiple Temporal and Spatial Scales
Reading section 2 of 9 · Multiple Temporal and Spatial Scales · 13 minSection 2 of 9 · 13 min

Book contents

Section 2 of 9 · 13 min read

Hierarchical, Hybrid, and Omnimodal Models › Multiple Temporal and Spatial ScalesOne section · Sections in this chapter2 / 9 ↓
Reading tools Definitions & inline links

Choose your expertise level to adjust how many terms are explained. Beginners see more tooltips, experts see fewer to maintain reading flow. Hover over underlined terms for instant definitions.

Article links

Make inline references clickable

A useful household-robot architecture must span multiple functional timescales and often benefits from predictions and decisions at several levels of abstraction, whether those levels are explicit, implicit, or distributed across cooperating models and controllers. "What will the room look like in thirty seconds if I keep pouring?" is a slow, coarse prediction. "What torque should the wrist apply this millisecond?" is a fast control decision. The relevant state changes at different rates. A fixed-rate model can waste computation if it repeatedly recomputes expensive slow context at the fast control rate, and recursive prediction exposes it to compounding rollout error.

Recall from Part I, Ch 6 that when we roll a learned model forward, each predicted state becomes the input to the next prediction. Consider true and predicted rollouts that start from the same state and use the same action sequence. Suppose the transition is uniformly LL-Lipschitz in state over the rollout region and the one-step model discrepancy is uniformly bounded by ε\varepsilon in the same norm. Their error then obeys eh+1≤Leh+εe_{h+1} \leq L e_h + \varepsilon with e0=0e_0 = 0. Under these conditions, the worst-case envelope can remain bounded, grow linearly, or grow geometrically across HH steps. If the transition amplifies differences, the bound compounds geometrically; at unit gain it grows linearly; if the map is contractive, it saturates. A fast fixed-step model chains more learned transitions over a fixed physical horizon, though cadence is not itself the cause of the error. With initial error e0>0e_0 > 0, the same bound also contains the propagated term LHe0L^H e_0.

The compounding effect is easiest to see directly. The next cell solves that recurrence for a few values of LL, holding the uniform per-step discrepancy fixed and using e0=0e_0 = 0.

In[3]:
Code
import numpy as np

compound_horizon = np.arange(0, 31)
compound_epsilon = 0.01
compound_lipschitz = np.array([0.85, 1.0, 1.10])
compound_envelopes = []
for L in compound_lipschitz:
    if abs(L - 1.0) < 1e-12:
        vals = compound_epsilon * compound_horizon
    else:
        vals = compound_epsilon * (L**compound_horizon - 1.0) / (L - 1.0)
    compound_envelopes.append(vals)
compound_envelopes = np.stack(compound_envelopes, axis=0)
Out[4]:
Visualization
Three bounds for rollouts that start from the same state and use the same actions under uniform one-step discrepancy: the L=0.85 curve saturates, L=1 grows linearly, and L=1.10 bends upward geometrically.
Worst-case rollout error bounds under the stated same-state, same-action, uniform-discrepancy, and uniform-Lipschitz assumptions. At unit gain, errors accumulate linearly; above unit gain, they are amplified geometrically; below unit gain, the bound saturates.

Temporal hierarchy reduces the frequency of high-level goal or option decisions while retaining fast low-level control. If a subgoal latent updates every KK fast steps, a five-minute plan at one hundred control updates per second can contain three hundred one-second macro-decisions instead of thirty thousand primitive decisions. This shortens the high-level search chain. It can reduce compounding prediction error when coarse-transition error remains small enough and fine-level error is controlled, for example by local rollouts or regular correction from observations; hierarchy does not guarantee an empirical advantage.

The Semi-Markov View

A canonical reinforcement-learning formalization of temporal abstraction is the options framework of Sutton, Precup, and Singh (1999). The idea is to let the agent choose not just individual actions but whole temporally extended behaviors, and to reason about those behaviors as if they were single actions. Let S\mathcal{S} and A\mathcal{A} denote the underlying MDP's state and action spaces. An option is a triple (I,π,β)(\mathcal{I}, \pi, \beta) where:

  • I⊆S\mathcal{I} \subseteq \mathcal{S} is the set of states in which the option may be started.
  • π(a∣s)\pi(a \mid s) is the option's internal action distribution: π(a∣s)∈[0,1]\pi(a \mid s) \in [0,1] and ∑a∈Aπ(a∣s)=1\sum_{a \in \mathcal{A}} \pi(a \mid s)=1 for each ss.
  • β:S→[0,1]\beta: \mathcal{S} \to [0,1] is the termination probability.

Read the triple as a small policy with a guard rail. The initiation set I\mathcal{I} says where the option makes sense, the internal policy π\pi says what to do while it runs, and the termination probability β\beta says when to stop. When the option is chosen at time tt, it executes its internal policy until β\beta fires. Assuming the option terminates almost surely, its duration is a positive integer-valued random variable τ∈{1,2,… }\tau \in \{1, 2, \dots\}; depending on the option and environment, τ\tau may be stochastic or degenerate at a fixed duration. This turns the Markov decision process (MDP) into a semi-Markov decision process (SMDP), where the value of taking option oo in state ss is

Q(s,o)=E ⁣[∑i=t+1t+τγi−t−1ri  +  γτV(st+τ) | st=s,ot=o]Q(s, o) = \mathbb{E}\!\left[\sum_{i=t+1}^{t+\tau} \gamma^{i-t-1} r_i \;+\; \gamma^{\tau} V(s_{t+\tau}) \,\middle|\, s_t = s, o_t = o \right]

where:

  • Q(s,o)Q(s,o): the expected return from taking option oo in state ss.
  • ss: the current state.
  • sts_t: the state at time step tt.
  • oo: the option.
  • oto_t: the option chosen at time step tt.
  • τ\tau: the random duration of the option.
  • tt: the current time step.
  • γ\gamma: discount factor.
  • rir_i: reward at time step ii.
  • ii: time index running from t+1t+1 to t+τt+\tau.
  • VV: continuation value under the policy followed after the option terminates.
  • st+τs_{t+\tau}: the state after option oo terminates.

Here the expectation is over the option's internal action choices, the environment's transitions and rewards, and the random termination time.

Compare this to an ordinary one-step action value, which contains one immediate reward plus one continuation value discounted by γ\gamma. The option-value function instead sums the rewards earned during the whole option and then discounts the continuation value by γτ\gamma^{\tau}, the discount over the option's realized duration. Different options therefore carry different effective continuation discounts, and the high-level policy chooses temporally extended actions rather than atomic ones. For 0<γ<10<\gamma<1, continuation value after a longer realized duration receives a smaller weight.

The effective discount for an option is γτ\gamma^{\tau}, so duration directly changes the weights the high-level policy compares.

In[5]:
Code
import numpy as np

option_durations = np.arange(1, 21)
discount_rates = np.array([0.90, 0.95, 0.99])
option_discounts = np.stack(
    [gamma**option_durations for gamma in discount_rates], axis=0
)
Out[6]:
Visualization
Three monotonically decreasing curves of gamma raised to duration; smaller gamma values decline much faster.
Effective discount applied to an option's continuation value as a function of its realized duration. Longer options are discounted more heavily, so the high-level policy compares options at different effective horizons.

Getting the discounting right is essential because an incorrect duration discount can distort option values whenever it changes the weight on a nonzero continuation or reward term. For example, replacing γτ\gamma^\tau with γ\gamma overweights positive continuation value after long options; the direction of the resulting preference also depends on within-option rewards and the sign of the continuation value.

Temporal Abstraction

Temporal abstraction treats sequences of primitive actions as single high-level actions. It is a common way to extend the effective high-level planning horizon because one abstract decision can cover many primitive steps.

Two-Level and Multi-Level Latent Architectures

Several hierarchical world-model architectures implement temporal abstraction with multiscale latent states, latent skills, or latent goals rather than literal call-and-return option triples. One two-level generative design stacks latent variables that update at different rates:

  • A slow latent zksz^{s}_k that changes once every KK fast steps, indexed by macro-time kk.
  • A fast latent ztfz^{f}_t that updates every step and receives z⌊t/K⌋sz^{s}_{\lfloor t/K \rfloor} as conditioning context.

When training makes it encode persistent behavior-level context, the slow latent can play a role analogous to option identity or high-level intent. The fast latent updates each step and can represent variation that the persistent context does not explain. Conditioning the fast transition on the current slow latent lets its predictions specialize to that context.

The joint generative factorization is

p(z1:Tf,z1:Ls∣z0f,z0s,a0:T−1)=∏k=0L−1[p(zk+1s∣zks)∏t=kK(k+1)K−1p(zt+1f∣ztf,at,z⌊t/K⌋s)]p(z^{f}_{1:T}, z^{s}_{1:L} \mid z^{f}_0, z^{s}_0, a_{0:T-1}) = \prod_{k=0}^{L-1} \left[ p(z^{s}_{k+1} \mid z^{s}_k) \prod_{t = kK}^{(k+1)K - 1} p(z^{f}_{t+1} \mid z^{f}_t, a_t, z^{s}_{\lfloor t/K \rfloor}) \right]

where:

  • z1:Tfz^{f}_{1:T}: sequence of fast latent states over fast time steps t=1,…,Tt = 1, \dots, T.
  • z1:Lsz^{s}_{1:L}: sequence of slow latent states over slow time steps k=1,…,Lk = 1, \dots, L.
  • z0fz^{f}_0: initial fast latent state before the sequence.
  • z0sz^{s}_0: initial slow latent state.
  • LL: number of slow steps (macro-steps).
  • KK: number of fast steps per slow step, so T=L×KT = L \times K.
  • ata_t: action at fast time step tt; the joint distribution is conditioned on the action sequence a0:T−1a_{0:T-1}.
  • pp: probability distribution.
  • zksz^{s}_k: the slow latent state at macro-time step kk.
  • ztfz^{f}_t: the fast latent state at fast time step tt.
  • ⌊⋅⌋\lfloor \cdot \rfloor: the floor function, giving the macro-time index for fast time tt.

The outer product over kk runs over macro-steps, and the inner product over tt runs over the KK fast steps inside each macro-step. Reading the factorization from left to right, the slow equation defines the temporal coarsening; the fast equation defines the fine-grained dynamics conditioned on the current slow context. Notice how the slow transition p(zk+1s∣zks)p(z^{s}_{k+1} \mid z^{s}_k) does not depend on the actions at all in this minimal form: it is a coarse drift model of how the sub-goal evolves, while the fast level absorbs the effect of actions. In richer variants the slow transition is also conditioned on an abstract action or goal, but the essential split is that the slow level summarizes and the fast level reacts.

Encoders can mirror this structure. A training-time smoothing encoder q(zks∣okK:(k+1)K−1)q(z^{s}_k \mid o_{kK : (k+1)K - 1}) aggregates an entire observation window, while a fast encoder q(ztf∣ot,z⌊t/K⌋s)q(z^{f}_t \mid o_t, z^{s}_{\lfloor t/K \rfloor}) captures per-step detail. Here oto_t is the observation at fast time step tt, and okK:(k+1)K−1o_{kK:(k+1)K-1} is the corresponding window for slow step kk. Because this smoothing form uses the completed window, it is not a causal online encoder for early steps in that same block. An online system can instead infer zksz^s_k from the preceding completed window o(k−1)K:kK−1o_{(k-1)K:kK-1} before block kk, or delay use of the summary until its window is complete. In either case, the asymmetry is the point: the slow encoder compresses a batch of observations, while the fast encoder sees a single observation together with the summary currently in force. A useful analogy is human scene perception: an initial glance can provide scene gist that helps guide subsequent fixations toward task-relevant detail.

FeUdal Networks (Vezhnevets et al., 2017) use a lower-temporal-resolution Manager whose dilated recurrent state and multi-step objective produce persistent goal conditioning, while a Worker chooses primitive actions each step. The Manager and Worker are learned concurrently with separate objectives, without propagating gradients between the two modules. Director adds hierarchical policies over the latent state space of a learned pixel-based world model: a Manager selects latent goals every eight steps and a Worker chooses primitive actions to reach them. These are policy hierarchies rather than evidence that Director contains slow and fast world-model state transitions. More generally, persistent context can spare a fast predictor from rediscovering behavior-level information at every step, but smaller residuals are an empirical outcome rather than a guarantee.

Spatial Scales

Temporal scales have a spatial counterpart. A building-scale robot may use room-level topology for navigation together with millimeter-scale contact geometry and high-rate force or torque signals for manipulation. With a fixed-budget dense representation, one resolution is inefficient across these scales: a fine global grid allocates detail where it is irrelevant, while a coarse grid misses local contact structure. A dense volumetric representation of an entire building at one-millimeter resolution is computationally impractical, so sparse or hierarchical representations allocate fine resolution selectively. Two common strategies are:

  • Coarse-to-fine latent pyramids. The state has a low-resolution "global" component, for example a volumetric or object-level map, plus fine local patches that are only queried where the agent is currently interacting. The global component ignores details that do not matter; the local component ignores the rest of the world. The two are linked by the agent's current focus, so the fine patches are always embedded in a coarse frame of reference.
  • Multi-resolution observation models. The decoder reconstructs a coarse observation everywhere and a fine observation only inside an attention window around the agent. Biological foveation offers an analogy: cone density and visual acuity peak in a small central region, while gaze shifts sample the rest of the scene. Most retinal photoreceptors, predominantly rods, are outside the fovea.

Structure, not resolution alone, should drive the transition model. Regular-grid maps often suit convolution because of spatial locality and translation equivariance; topological maps often suit graph networks because they encode adjacency and relations. Continuous-time models can help with irregularly timed observations or queries, while contact dynamics may additionally require event-aware modeling and accurate numerical integration. Graph networks can themselves be effective for contact dynamics. The practical rule is to match an inductive bias to the representation's actual topology, smoothness, events, and sampling process.

Failure Modes

Hierarchical models add complexity, and that complexity has costs. A slow policy can propose subgoals that a fast policy cannot realize. Termination can be poorly calibrated, causing premature or excessively long execution. When observations support several contexts, slow-state inference can become uncertain or lock onto the wrong one unless history, actions, or later observations correct it. A window encoder can also exploit spurious phase or boundary cues. Window length itself becomes a shortcut when variable lengths or padding patterns correlate with the target. Such correlations fail when task periods or window construction change.

Several of these failure modes can manifest as disagreement between the levels, with no training signal forcing them to reconcile. Silent inconsistency is especially difficult because it may not raise an obvious error. The slow level believes the robot is near the mug, the fast level believes it is near the sink, and the model produces plausible but wrong predictions for both. This drift can remain hidden from aggregate one-step validation loss, so cross-level agreement should be evaluated explicitly or included as an equivalent targeted validation term.

Cross-level consistency losses address the disagreement subset, not every failure above. If the slow level predicts a projected coarse summary of the next window and the fast level predicts its fine state sequence, applying the same projection to the fast prediction gives a shared quantity on which the two levels can be compared. Training on that agreement supplies a direct gradient to the slow level. Subgoal feasibility instead calls for reachability or capability checks; termination needs dedicated closed-loop stopping metrics; and window shortcuts need distribution-shift tests and ablations. Joint end-to-end training and a bounded window matched to the task's timescale may support these safeguards, but do not replace them.