Research

PSF predictive states: a flow prior turns noise into 8 states; state k summarizes only the blocks before k (a lower triangle), and block k, unseen by its state, is written by the block decoder

Rethinking the Roles of Diffusion and Decoding: Predictive-State Flows for Parallel Language Generation arXiv preprint, 2026 Jinjin He*, Xinsong Feng*, Lucheng Fu, Sinan Wang, Bo Zhu, Jiayun Wang (*: co-first author)

Paper Code Models Project Page

Chinese poem transported by the VIOT incompressible flow operator

3D volumetric letters transported through G, R, A, P, and H

A Variational Optimal Transport Operator on Incompressible Flow arXiv preprint, 2026 Jinjin He, Shenyifan Lu, Sinan Wang, Zhiqi Li, Duowen Chen, and Bo Zhu

Paper Code Project Page

Neural Harmonic Measure Operator: harmonic-measure density with Walk-on-Spheres paths and the residual lift

Neural Harmonic Measure Operator NeurIPS 2026 (Conference on Neural Information Processing Systems) Jinjin He, Sinan Wang, Yuchen Sun, Bo Zhu

Paper Code Project Page

Audible World Models: Spatially Aware Sound Generation for 3D Worlds NeurIPS 2026 (Conference on Neural Information Processing Systems) Duowen Chen, Jinjin He, Gouthaman KV, Sandeep Bangalore Venkatesh, Bo Zhu

Paper Project Page

Coupled nonlinear surface waves and three-dimensional fluid flow

Hamiltonian Two-Way Coupling of Nonlinear Waves and 3D Flows SIGGRAPH Asia 2026 (ACM Transactions on Graphics) to appear Sinan Wang*, Ruicheng Wang*, Taiyuan Zhang, Fan Feng, Jinjin He, Yuchen Sun, Zhiqi Li, and Bo Zhu (*: co-first author)

Webpage Paper Code Video

Image generation results using unified long-short flow maps

Unified Long-Short Flow Maps for Fast Generation SIGGRAPH Asia 2026 (Conference Papers) to appear Zhiqi Li, Jinjin He, Duowen Chen, Ruicheng Wang, Shenyifan Lu, Yuxuan Liao, Yuchen Sun, Sinan Wang, and Bo Zhu

Two-phase Impulse Fluid on Particle Flow Map TVCG 2026 (IEEE Transactions on Visualization and Computer Graphics) Hui Wang, Jinjin He, Xubo Yang

Paper Project Page

Hermite-NGP: Gradient-Augmented Hash Encoding for Learning PDEs ICML 2026 (International Conference on Machine Learning) Jinjin He, Zhiqi Li, Sinan Wang, Bo Zhu

Paper Code Project Page

Trajectory Consistency for One-Step Generation on Euler Mean Flows ICML 2026 (International Conference on Machine Learning) Zhiqi Li, Yuchen Sun, Duowen Chen, Jinjin He, Bo Zhu

Paper Project Page

Generative Modeling with Orbit-Space Particle Flow Matching SIGGRAPH 2026 (ACM Transactions on Graphics) Sinan Wang*, Jinjin He*, Shenyifan Lu, Ruicheng Wang, Greg Turk, Bo Zhu(* co-first author)

Paper Code Project Page

A Level Set Method on Particle Flow Maps IJNME 2026 (International Journal for Numerical Methods in Engineering) Jinjin He, Taiyuan Zhang, Zhiqi Li, Junwei Zhou, Duowen Chen, Bo Zhu

Paper arXiv Code

An Adjoint Method for Differentiable Fluid Simulation on Flow Maps SIGGRAPH Asia 2025 (Conference Papers) Zhiqi Li*, Jinjin He*, Barnabás Börcsök, Taiyuan Zhang, Duowen Chen, Tao Du, Ming Lin, Greg Turk, Bo Zhu (* co-first author)

Paper Code Project Page

Fluid Simulation on Compressible Flow Maps SIGGRAPH 2025 (ACM Transactions on Graphics) Duowen Chen*, Zhiqi Li*, Taiyuan Zhang, Jinjin He, Junwei Zhou, Bart G van Bloemen Waanders, Bo Zhu(* co-first author)

Paper Code Project Page

Multi-level Partition of Unity on Differentiable Moving Particles SIGGRAPH Asia 2024 (ACM Transactions on Graphics) Jinjin He, Taiyuan Zhang, Hiroki Kobayashi, Atsushi Kawamoto, Yuqing Zhou, Tsuyoshi Nomura, Bo Zhu

Paper Project Page Code