Bang Xiao 肖 棒

Hi👋 I am Bang Xiao, an undergraduate student major in Computer Science at Shanghai Jiao Tong University. I am selected as a member of Zhiyuan Honors Program.

I am currently a research visitor at Rehg Lab, University of Illinois at Urbana-Champaign, supervised by James M. Rehg. I am actively seeking for PhD opportunities for 2027 Fall!

Previously, I was a research intern at GenAI Group, Microsoft Research Asia, supervised by Shaohan Huang and Tengchao Lv.

My research interests broadly span generative models, world models, multi-modal learning, and reinforcement learning. My long-term goal is to build intelligent systems that can truly understand the physical world, and are capable of reasoning and planning.


Education
  • Shanghai Jiao Tong University
    Shanghai Jiao Tong University
    B.S. in Computer Science, Zhiyuan Honors Program
    Sep. 2023 - Now
  • No.1 Middle School Affiliated to Central China Normal University
    No.1 Middle School Affiliated to Central China Normal University
    High School
    Sep. 2020 - Jun. 2023
Experience
  • Rehg Lab, University of Illinois at Urbana-Champaign <br/> Superviser: James M. Rehg
    Rehg Lab, University of Illinois at Urbana-Champaign
    Superviser: James M. Rehg
    Research Visitor
    July. 2026 - Now
  • GenAI Group, Microsoft Research Asia <br/> Superviser: Shaohan Huang and Tengchao Lv
    GenAI Group, Microsoft Research Asia
    Superviser: Shaohan Huang and Tengchao Lv
    Research Intern
    May. 2025 - Nov. 2025
  • SJTU MVIG Lab <br/> Superviser: Cewu Lu and Yonglu Li
    SJTU MVIG Lab
    Superviser: Cewu Lu and Yonglu Li
    Research Intern
    Dec. 2024 - Now
  • SJTU EPIC Lab <br/> Superviser: Linfeng Zhang
    SJTU EPIC Lab
    Superviser: Linfeng Zhang
    Research Intern
    Aug. 2024 - Dec. 2024
News
2026
I am joining Rehg's Lab at UIUC as a research visitor. See you in UIUC! 🌽🌽🌽
Jul 20
One paper is accepted to ICLR 2026. See you in Brazil!
Jan 26
2025
Join MSRA GenAI Group as a student research intern!
May 19
Publications (view all )
Code Aesthetics with Agentic Reward Feedback
Code Aesthetics with Agentic Reward Feedback

Bang Xiao*, Lingjie Jiang*, Shaohan Huang#, Tengchao Lv, Yupan Huang, Xun Wu, Lei Cui, Furu Wei (* equal contribution, # corresponding author)

International Conference on Learning Representations (ICLR) 2026

We present a unified framework for aesthetic code generation that substantially improves both visual quality and functionality. With a large-scale dataset, agentic reward feedback, and a new benchmark, our 4B model outperforms GPT-4o and GPT-4.1 and matches the performance of open-source models hundreds of times larger. This work establishes a strong and scalable approach to advancing code aesthetics in large language models.

Code Aesthetics with Agentic Reward Feedback

Bang Xiao*, Lingjie Jiang*, Shaohan Huang#, Tengchao Lv, Yupan Huang, Xun Wu, Lei Cui, Furu Wei (* equal contribution, # corresponding author)

International Conference on Learning Representations (ICLR) 2026

We present a unified framework for aesthetic code generation that substantially improves both visual quality and functionality. With a large-scale dataset, agentic reward feedback, and a new benchmark, our 4B model outperforms GPT-4o and GPT-4.1 and matches the performance of open-source models hundreds of times larger. This work establishes a strong and scalable approach to advancing code aesthetics in large language models.

Efficient and Scalable Monocular Human-Object Interaction Motion Reconstruction
Efficient and Scalable Monocular Human-Object Interaction Motion Reconstruction

Boran Wen*, Ye Lu*, Keyan Wan, Sirui Wang, Jiahong Zhou, Junxuan Liang, Xinpeng Liu, Bang Xiao, Dingbang Huang, Ruiyang Liu, Yonglu Li# (* equal contribution, # corresponding author)

Under review. 2026

We introduce 4DHOISolver, a human-in-the-loop framework that enables scalable and physically plausible 4D human–object interaction reconstruction from monocular internet videos. Using this approach, we build Open4DHOI, a large-scale dataset with 144 object types and 103 actions, and show that the recovered interactions can drive RL-based imitation learning. Our results also reveal that accurate contact prediction remains a key open challenge for current 3D foundation models.

Efficient and Scalable Monocular Human-Object Interaction Motion Reconstruction

Boran Wen*, Ye Lu*, Keyan Wan, Sirui Wang, Jiahong Zhou, Junxuan Liang, Xinpeng Liu, Bang Xiao, Dingbang Huang, Ruiyang Liu, Yonglu Li# (* equal contribution, # corresponding author)

Under review. 2026

We introduce 4DHOISolver, a human-in-the-loop framework that enables scalable and physically plausible 4D human–object interaction reconstruction from monocular internet videos. Using this approach, we build Open4DHOI, a large-scale dataset with 144 object types and 103 actions, and show that the recovered interactions can drive RL-based imitation learning. Our results also reveal that accurate contact prediction remains a key open challenge for current 3D foundation models.

Token Pruning for Caching Better: 9$\times$ Acceleration on Stable Diffusion for Free
Token Pruning for Caching Better: 9$\times$ Acceleration on Stable Diffusion for Free

Evelyn Zhang*, Bang Xiao*, Jiayi Tang, Qianli Ma, Chang Zou, Xuefei Ning, Xuming Hu, Linfeng Zhang# (* equal contribution, # corresponding author)

arXiv preprint 2024

Based on token prune and layer cache technology, we present a new Stable Diffusion acceleration method named dynamics-aware token pruning (DaTo). In the COCO-30k, we observed a 7$\times$ acceleration coupled with a notable FID reduction of 2.17.

Token Pruning for Caching Better: 9$\times$ Acceleration on Stable Diffusion for Free

Evelyn Zhang*, Bang Xiao*, Jiayi Tang, Qianli Ma, Chang Zou, Xuefei Ning, Xuming Hu, Linfeng Zhang# (* equal contribution, # corresponding author)

arXiv preprint 2024

Based on token prune and layer cache technology, we present a new Stable Diffusion acceleration method named dynamics-aware token pruning (DaTo). In the COCO-30k, we observed a 7$\times$ acceleration coupled with a notable FID reduction of 2.17.

All publications