Liouville-Regularized Flow-Matching World Models

  • Proposed Liouville (squared divergence) and Jacobian Frobenius pointwise regularizers for flow-matching latent world models, reducing long-horizon latent drift by 17 orders of magnitude at H=128 on DMControl Walker-Walk versus an unregularized baseline.
  • Formally proved cycle-consistency is structurally vacuous for rectified flows in the autonomous limit.
  • Confirmed empirically that Liouville variants achieve cycle-error floors 3 orders of magnitude lower than naive cycle regularization across all solver tolerances.
Abhijit Chunduru
Abhijit Chunduru
Master’s Student

ML Systems and Trustworthy AI at UMass Amherst