works on

From the 1 of 15 linked papers with an AI index.

collaborators

15 papers

cs.LG2026

Beyond Myopic World Models: Long-Horizon End-to-End Training for Direct Future Prediction

Xinyi Li, Zaishuo Xia, Chenjie Hao +1

World models are expected to support imagination over extended temporal horizons, yet most are still trained through local few-step prediction objectives and deployed by recursivel…

q-bio.NC2026

Toward a mechanistic understanding of inference in visual cortex and diffusion models

Zeyu Yun, Alexander Belsten, Dasheng Bi +3

We describe a model of perceptual inference in primary visual cortex (V1) equivalent to a minimal diffusion model whose function can be readily understood from its parameters. The…

cs.AI2026

Concept-Guided Spatial Regularization for World Models in Atari Pong

Yukuan Lu, Zaishuo Xia, Weyl Lu +1

The paper evaluates several visual world‑model agents on Atari Pong, identifies systematic rollout failures, and introduces Concept‑Guided Spatial Regularization (CGSReg) to improv…

cs.AI2026

When Models Know When They Do Not Know: Calibration, Cascading, and Cleaning

Chenjie Hao, Weyl Lu, Yuko Ishiwaka +3

When a model knows when it does not know, many possibilities emerge. The first question is how to enable a model to recognize that it does not know. A promising approach is to use…

cs.CV2026

MotionPyramid: Hierarchical Motion Representation and Residual Interfaces

Gao Zhu, Zaishuo Xia, Yubei Chen

We ask whether the representational hierarchy seen in perception, from local primitives such as edges to higher level structures such as parts and objects, can be established for m…

cs.LG2026

Cloning Deterministic Worlds: The Critical Role of Latent Geometry in Long-Horizon World Models

Zaishuo Xia, Yukuan Lu, Xinyi Li +2

A world model is an internal model that simulates how the world evolves. Given past observations and actions, it predicts the future physical state of both the embodied agent and i…