Portrait
Kun Wang
CS PhD Student @ Princeton University
About Me

Hello! I am a first-year CS Ph.D. student at Princeton University, working with 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.

Currently, I'm interested in Cognitive Representations and Memory in Vision Language Models.

Curriculum Vitae
Education
  • Princeton University
    Princeton University
    Department of Computer Science
    Ph.D. Student
    Sep. 2025 - Present
  • UC San Diego
    UC San Diego
    B.S. in Mathematics & Computer Science
    Sep. 2021 - Jun. 2025
Selected Publications (view all )
Discovering Latent Causal Graphs from Spatio-Temporal Data
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.

SceneProg: Program Synthesis for 3D Scene Generation using LLMs
SceneProg: Program Synthesis for 3D Scene Generation using LLMs

Kunal Gupta, Ishit Mehta, Kun Wang, Nicholas Chua, Yan Deng, Abhimanyu Krishna, Ravi Ramamoorthi, Manmohan Chandraker

Under Review 2025

We present SceneProg, a novel approach that leverages Large Language Models for program synthesis to generate complex 3D scenes, enabling intuitive scene creation through natural language instructions.

SceneProg: Program Synthesis for 3D Scene Generation using LLMs

Kunal Gupta, Ishit Mehta, Kun Wang, Nicholas Chua, Yan Deng, Abhimanyu Krishna, Ravi Ramamoorthi, Manmohan Chandraker

Under Review 2025

We present SceneProg, a novel approach that leverages Large Language Models for program synthesis to generate complex 3D scenes, enabling intuitive scene creation through natural language instructions.

All publications