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 )
LeRF: Learning Reference Coordinate Frames for Perspective Taking Reasoning
LeRF: Learning Reference Coordinate Frames for Perspective Taking Reasoning

Bang Xiao, Wenqi Jia, Ozgur Kara, Tiancheng Shen, Yibo Yang, Bolin Lai#, Junho Kim#, James Matthew Rehg# (# corresponding author)

arXiv preprint 2026

Perspective taking requires spatial reasoning from viewpoints beyond the camera, yet current VLMs often struggle to make this shift. We introduce LeRF, which trains VLMs to selectively construct and render explicit entity-centered reference frames and use these visual cues for viewpoint-dependent reasoning, through supervised fine-tuning followed by reinforcement learning. LeRF consistently improves over its backbone across diverse perspective-taking benchmarks and also yields better reference-frame grounding and orientation estimation.

LeRF: Learning Reference Coordinate Frames for Perspective Taking Reasoning

Bang Xiao, Wenqi Jia, Ozgur Kara, Tiancheng Shen, Yibo Yang, Bolin Lai#, Junho Kim#, James Matthew Rehg# (# corresponding author)

arXiv preprint 2026

Perspective taking requires spatial reasoning from viewpoints beyond the camera, yet current VLMs often struggle to make this shift. We introduce LeRF, which trains VLMs to selectively construct and render explicit entity-centered reference frames and use these visual cues for viewpoint-dependent reasoning, through supervised fine-tuning followed by reinforcement learning. LeRF consistently improves over its backbone across diverse perspective-taking benchmarks and also yields better reference-frame grounding and orientation estimation.

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)

arXiv preprint 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)

arXiv preprint 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