Undergraduate Researcher · Zhejiang University
Jinyuan Mao 茅晋源
I am a senior undergraduate at Zhejiang University, pursuing a dual degree in Statistics and Information Management & Information Systems. My research is on generative AI and efficient AI. I am interested in making video diffusion models and interactive world models faster, more consistent over long horizons, and easier to steer.
Most recently, I was a research intern at PixAI, working with Chenxi Liu (University of Victoria) on interactive image generation. Before that, I spent eight months at the ZIP Lab advised by Prof. Bohan Zhuang, working on sparse attention and KV-cache design for long video generation.
I also enjoy high-performance computing: our team placed 1st globally at the SC25 IndySCC, and I look after the GPU infrastructure at ZJU X-Lab.
News
Awarded the National Scholarship (top 0.2%) for the third consecutive year.
HimmBO, my first-author work on interactive LoRA merging, is submitted to Eurographics 2027.
WorldAttention is on arXiv. It received Accept / Accept / Accept (5/5/5) reviews at NeurIPS 2026.
BIFE is accepted to SIGGRAPH Asia 2026 Technical Communications.
Joined PixAI as a research intern, working with Chenxi Liu on interactive image generation.
Our team won 1st place globally at the SC25 IndySCC Student Cluster Competition.
Awarded the National Scholarship (top 0.2%) for the second consecutive year.
Joined the ZIP Lab at Zhejiang University, advised by Prof. Bohan Zhuang.
National 2nd Prize at the 12th National Parallel Application Challenge (PAC).
Designed and built the official website of ZJU X-Lab from scratch.
My study notes are online: My Notebook.
1st place in the National Finals of the ICT Competition.
Publications
* equal contribution
WorldAttention: An Efficient Attention Architecture for Interactive Video World Models
NeurIPS 2026 Reviews: Accept / Accept / Accept
Attention heads in video diffusion transformers are sparse in very different ways. Hybrid Sparse Attention pairs a linear global branch with head-adaptive block-sparse attention for a 2.21× end-to-end speedup at matched quality.
HimmBO: High-Dimensional Interactive Model-Merging Bayesian Optimization for Image Generation
Under review at Eurographics 2027
Two-stage preferential Bayesian optimization scales interactive LoRA merging from a few dozen to 1,080 adapters. A style-aware kernel shares preferences across similar adapters, so it needs under 1/10 of the image generations used by prior work.
Research Experience
-
PixAI Research Intern
Jun 2026 – Oct 2026Interactive image generation · Advised by Chenxi Liu (University of Victoria)
- Led HimmBO as first author: method design, implementation, experiments and writing.
- Built a two-stage preferential Bayesian optimization framework that scales interactive LoRA merging to 1,080 adapters. Its level-set screening stage generates no images during interaction.
- Designed a style-aware kernel that transfers preference information across visually similar adapters, reaching a target similarity with under 1/10 of the image generations prior work needs.
-
ZIP Lab, Zhejiang University Research Intern
Oct 2025 – Jun 2026Efficient video generation & world models · Advised by Prof. Bohan Zhuang
- Worked on sparse attention and KV-cache design for video generation, as co-first author of WorldAttention and BIFE.
-
Provincial SRTP Project Project Leader
Apr 2024 – Jun 2025Database agent framework · Zhejiang University
- Proposed a two-stage trial-and-error learning framework that improves how database agents interact, with clear gains on research benchmarks. Filed one national invention patent.
Education
Zhejiang University
Sep 2023 – Jun 2027B.S. in Statistics & B.Mgt. in Information Management and Information Systems (dual degree) · Hangzhou, China
GPA 92.5/100Rank 1/20
Honors & Awards
- 1st Place Globally, SC25 IndySCC Student Cluster CompetitionNov 2025
- National 2nd Prize, 12th National Parallel Application Challenge (PAC)Aug 2025
- 1st Place, National Finals, ICT CompetitionMar 2025
- National Scholarship (top 0.2%), awarded three times2024, 2025, 2026
Other Experience
-
ZJU X-Lab Operations & Infrastructure Lead
Jun 2025 – PresentStudent-run innovation organization
- Run the GPU clusters with Docker-isolated environments for 10+ projects. Built the private Overleaf, NAS storage, VPN and the organization's website.
-
Nodesk AI AI Engineering Intern
Jun 2025 – Aug 2025Industry internship · Hangzhou
- Built a smart-reply cross-validation module and fine-tuned it for a 15% accuracy gain. Co-authored a patent on a guardrail that combines reranking models with scenario recognition.
Skills
- Languages
- PythonC/C++SQLLaTeX
- ML & Systems
- PyTorchDiffusersMPIOpenMPDockerLinux