What Should World Models Forget?

NeurIPS 2026 · Workshop on Continual World Models

What Should World Models Forget?

Stratified Retention for Continual Adaptation

Nishit AnandRamani Duraiswami†Dinesh Manocha†
University of Maryland, College Park, USA†Equal advising
University of Maryland
TL;DRContinual world models must revise what the world has invalidated. We stratify their knowledge by invariance timescale and propose differential retention, which reports invariant retention and revision latency separately, not as one forgetting score.

When the world changes, revising is not forgetting

Continual learning treats any drop in performance on previously seen data as forgetting, a convention inherited from settings where the prediction target is stationary. A world model's target is the environment, which changes, so knowledge that was accurate when learned can later become false, and revising it is correct behavior rather than a defect. World models also encode structural knowledge, such as physical laws and object permanence, that must never be degraded.

We argue

Obsolescence

Retaining knowledge that has become false is a failure mode of continual world models, distinct from catastrophic forgetting, and current benchmarks exclude it by construction.

We propose

Strata by invariance timescale

Organize world-model knowledge by how long it stays true, τ, and protect knowledge more strongly the longer it stays true.

We propose

Differential retention

An evaluation protocol that measures invariant retention and revision latency separately, instead of one forgetting score.

Three phenomena, one score

A household robot has learned where the furniture is. Then someone moves the chair. Current metrics cannot tell these three apart.

The change chair by the windowchair at the table the table does not move
A defect

Catastrophic forgetting

The loss of knowledge that is still true.

In the houseForgetting where the unmoved table stands.
Well studied

Deliberate unlearning

The removal of knowledge that is still true but unwanted, for instance for privacy.

In the houseNot part of the robot example.
Revision required

Obsolescence

The retention of knowledge that has become false.

In the houseStill predicting the chair by the window.

Changing ground truth is well studied in concept drift and in the temporal factuality of language models, but it has not been formulated for world models. World models are different: alongside facts that change, they encode physical, geometric and causal structure, such as that unsupported objects fall and hidden objects continue to exist. That structure should survive every update, and no single label can check it.

Protection should grow with τ

The invariance timescale τ is the expected time until the environment makes a piece of knowledge false. It is a property of the environment, not of the model.

StratumExamplesRequired behaviorMeasured by
Invariantsτ = ∞
Examplesform of physical laws, object permanence, geometric consistency, action–effect structure
Required behaviorrefined, never degraded
Measured byprobes every round
Environment regularitieslong
Examplesroom layout, road network, friction, mass, the agent's body
Required behaviorrevised on accumulated evidence
Measured bylatency; not revised by one-off events
Instance configurationshort
Examplesobject positions, door states, lighting
Required behaviorrevised promptly
Measured bylatency; never-revised count
Transient stateimmediate
Exampleswho is moving where, momentary occlusion
Required behaviorkept in current state, not in weights
Measured bynot measured
Protection CurrentProposed

Refined, never degraded. A model's belief about an invariant may be wrong and should be refined as evidence accumulates, but it must never get worse. A Kalman filter makes this trade-off: its update rate falls as τ grows and, for τ = ∞, shrinks toward zero without reaching it.

Current practice fails in both directions

Under-protected

Invariants

Standard pipelines do not check whether invariants survived an update. EWC protects parameters by their importance for earlier tasks, but importance is not the same as persistence: a parameter can encode a fact the environment has since invalidated, and EWC then protects it most strongly.

Over-protected

Lower strata

A replay buffer that stores the old layout of a room keeps reasserting it after the room has changed. Methods that replace outdated exemplars or evict stale transitions address this; protecting invariants has no counterpart in either setting.

Forgetting metrics reward the frozen model

Three robots are tested on their original test data after the chair has moved. Switch between how standard metrics score them and what differential retention reports.

CorrectRevises the chair, keeps the table.
Chairrevised
Tablekept
Loses 1 itemThe chair's old position now counts as forgetting.
BrokenKeeps the stale chair, forgets the table.
Chairstale
Tableforgotten
Loses 1 itemThe same score as the correct model, although it failed twice.
FrozenNever updated.
Chairstale
Tablekept
Loses nothingReceives the best score.

Every correct revision counts as forgetting, so the metric favors models that revise as little as possible, and the frozen model, the very failure continual adaptation is meant to remedy, scores best.

Differential retention

Two quantities, reported separately, because a single score would hide which failure occurred.

Measurement (a)

Invariant retention

A fixed set of probes, small tests that each check one invariant, drawn from the τ = ∞ stratum and held out from the adaptation data. They are re-run after every adaptation round, because a single evaluation at the end cannot show which update damaged an invariant. Unchanged lower-stratum facts, such as the table, are re-tested alongside.

Hide a ball behind the couch. Does the model still predict the ball is there?

Measurement (b)

Revision latency

For each fact the environment changes, how many observations the model needs before its predictions favor the new outcome over the old one on held-out queries. Facts that are never revised are counted separately. This generalizes the LoCA regret from a single local change to every changed fact.

How many times must the robot see the chair at the table?

Neither quantity is informative on its own. A frozen model loses no invariants but never revises. A model that adapts by overwriting revises quickly but damages its invariants.

Together, the two measurements place a model in one of four quadrants. Forgetting metrics alone reward high retention and slow revision, and so favor the frozen model.

To our knowledge, such probes have not been applied as a regression suite across an adaptation stream.

Probes beyond physics

Each kind of invariant needs its own probe. For action-conditioned or latent world models, a probe can read out the model's own prediction error.

Physical laws (form)

ProbeA possible and an impossible continuation of the same scene: a dropped cup that falls, or one that floats.
PassHigher surprise on the impossible continuation.

Object permanence

ProbeHide an object behind an occluder, then remove the occluder: a ball behind the couch.
PassObject predicted to still be there; surprise if it has vanished.

Impenetrability

ProbeMove into a wall or a solid object.
PassPredicted state does not pass through it.

Geometric consistency

ProbeLoop closure, such as walking around the table; several views of one scene.
PassPredicted end state matches the start; views agree with each other.

Causal action–effect

ProbeFrom one state, compare two different actions, such as pushing or pulling the door; also the null action.
PassPredicted difference matches the simulator; the null action leaves a static scene unchanged.

Open questions

1

Estimating τ

The strata are a starting proposal with domain-dependent boundaries. In benchmarks τ is set by design; estimating it reliably in deployment, especially for slowly changing regularities, remains open.

2

Probes for geometry and causality

Mature probes exist mainly for physical invariants in generative video models; probes for geometric and causal structure still need to be developed and validated.

3

Recurring knowledge

A door that is open on some days and closed on others should be suspended and later reinstated rather than overwritten.

4

Entangled invariants

If invariants are entangled with other knowledge inside a learned world model, protecting them separately may require architectural support.

BibTeX

@inproceedings{anand2026whatshould,
  title         = {What Should World Models Forget? Stratified Retention for Continual Adaptation},
  author        = {Anand, Nishit and Duraiswami, Ramani and Manocha, Dinesh},
  booktitle     = {NeurIPS 2026 Workshop on Continual World Models},
  year          = {2026},
  eprint        = {2610.03713},
  archivePrefix = {arXiv},
  url           = {https://arxiv.org/abs/2610.03713}
}