Cohen, Lior

2 publications

ICLR 2026 Horizon Imagination: Efficient On-Policy Rollout in Diffusion World Models Lior Cohen, Ofir Nabati, Kaixin Wang, Navdeep Kumar, Shie Mannor

ICML 2024 Improving Token-Based World Models with Parallel Observation Prediction Lior Cohen, Kaixin Wang, Bingyi Kang, Shie Mannor