Hello! I am a first-year CS Ph.D. student at Princeton University, advised by Prof. Olga Russakovsky. During my undergraduate studies at UC San Diego, I was fortunate to be advised by Prof. Rose Yu and Prof. Yian Ma on spatio-temporal causal discovery, and Prof. Manmohan Chandraker on 3D scene generation with Large Language Models.
I'm generally interested in embodied agents and accelerating vision research, currently working on robotics agents and world models.
Warning
Problem: The current name of your GitHub Pages repository ("Solution: Please consider renaming the repository to "
http://".
However, if the current repository name is intended, you can ignore this message by removing "{% include widgets/debug_repo_name.html %}" in index.html.
Action required
Problem: The current root path of this site is "baseurl ("_config.yml.
Solution: Please set the
baseurl in _config.yml to "Education
-
Princeton UniversityDepartment of Computer Science
Ph.D. StudentSep. 2025 - Present -
UC San DiegoB.S. in Mathematics & Computer ScienceSep. 2021 - Jun. 2025
Publications
In-Context Learning Can Help Vision-Language Models Overcome Training Priors
Kun Wang, Xindi Wu, Sanghyuk Chun, Olga Russakovsky, Esin Tureci
Conference on Neural Information Processing Systems (NeurIPS) 2026
Vision-language models often fail on images that violate their training priors—a deficit usually read as missing visual capability—yet we show that controlled visual in-context learning lets them overcome these priors, recovering large gains (up to +30.3%) on prior-conflicting examples while leaving real accuracy unchanged and revealing grounding abilities that standard evaluations overlook.
In-Context Learning Can Help Vision-Language Models Overcome Training Priors
Kun Wang, Xindi Wu, Sanghyuk Chun, Olga Russakovsky, Esin Tureci
Conference on Neural Information Processing Systems (NeurIPS) 2026
Vision-language models often fail on images that violate their training priors—a deficit usually read as missing visual capability—yet we show that controlled visual in-context learning lets them overcome these priors, recovering large gains (up to +30.3%) on prior-conflicting examples while leaving real accuracy unchanged and revealing grounding abilities that standard evaluations overlook.
MLS-Bench: A Holistic and Rigorous Assessment of AI Systems on Building Better AI
Bohan Lyu, Yucheng Yang, Siqiao Huang, Jiaru Zhang, Qixin Xu, Xinghan Li, Xinyang Han, Yicheng Zhang, Huaqing Zhang, Runhan Huang, Kaicheng Yang, Zitao Chen, Wentao Guo, Junlin Yang, Xinyue Ai, Wenhao Chai, Yadi Cao, Ziran Yang, Kun Wang, Dapeng Jiang, Huan-ang Gao, Shange Tang, Chengshuai Shi, Simon S. Du, Max Simchowitz, Jiantao Jiao, Dawn Song, Chi Jin
Conference on Neural Information Processing Systems (NeurIPS) 2026 | Datasets and Benchmarks Track, Oral Presentation
We introduce MLS-Bench, a benchmark of 140 tasks across 12 domains testing whether AI systems can invent generalizable, scalable ML methods rather than only apply existing ones—and find that current agents remain far from surpassing human-designed methods, bottlenecked by the scientific insight needed to plan and validate claims rather than by more search or compute.
MLS-Bench: A Holistic and Rigorous Assessment of AI Systems on Building Better AI
Bohan Lyu, Yucheng Yang, Siqiao Huang, Jiaru Zhang, Qixin Xu, Xinghan Li, Xinyang Han, Yicheng Zhang, Huaqing Zhang, Runhan Huang, Kaicheng Yang, Zitao Chen, Wentao Guo, Junlin Yang, Xinyue Ai, Wenhao Chai, Yadi Cao, Ziran Yang, Kun Wang, Dapeng Jiang, Huan-ang Gao, Shange Tang, Chengshuai Shi, Simon S. Du, Max Simchowitz, Jiantao Jiao, Dawn Song, Chi Jin
Conference on Neural Information Processing Systems (NeurIPS) 2026 | Datasets and Benchmarks Track, Oral Presentation
We introduce MLS-Bench, a benchmark of 140 tasks across 12 domains testing whether AI systems can invent generalizable, scalable ML methods rather than only apply existing ones—and find that current agents remain far from surpassing human-designed methods, bottlenecked by the scientific insight needed to plan and validate claims rather than by more search or compute.
InteriorAgent: LLM Agent for Interior Design-Aware 3D Layout Generation
Kunal Gupta, Ishit Mehta, Kun Wang, Nicholas Chua, Yan Deng, Abhimanyu Krishna, Ravi Ramamoorthi, Manmohan Chandraker
International Conference on 3D Vision (3DV) 2026
We propose InteriorAgent, an LLM-agent-driven framework for text-to-3D indoor scene generation that encodes interior design principles through a novel scene description language and synthesis and optimization tools, producing scenes that users strongly favor over prior state-of-the-art methods.
InteriorAgent: LLM Agent for Interior Design-Aware 3D Layout Generation
Kunal Gupta, Ishit Mehta, Kun Wang, Nicholas Chua, Yan Deng, Abhimanyu Krishna, Ravi Ramamoorthi, Manmohan Chandraker
International Conference on 3D Vision (3DV) 2026
We propose InteriorAgent, an LLM-agent-driven framework for text-to-3D indoor scene generation that encodes interior design principles through a novel scene description language and synthesis and optimization tools, producing scenes that users strongly favor over prior state-of-the-art methods.
Discovering Latent Causal Graphs from Spatio-Temporal Data
Kun Wang*, Sumanth Varambally*, Duncan Watson-Parris, Yian Ma, Rose Yu (* equal contribution)
International Conference on Machine Learning (ICML) 2025 | Oral Presentation at NeurIPS 2024 Causal Representation Learning Workshop
This paper presents a novel approach to discovering latent causal structures from spatio-temporal data, addressing the challenge of identifying causal relationships in complex dynamical systems.
Discovering Latent Causal Graphs from Spatio-Temporal Data
Kun Wang*, Sumanth Varambally*, Duncan Watson-Parris, Yian Ma, Rose Yu (* equal contribution)
International Conference on Machine Learning (ICML) 2025 | Oral Presentation at NeurIPS 2024 Causal Representation Learning Workshop
This paper presents a novel approach to discovering latent causal structures from spatio-temporal data, addressing the challenge of identifying causal relationships in complex dynamical systems.
