Jun-Kun Chen

Legal name: Junkun Chen · Publications as Jun-Kun Chen / J.-K. Chen

Ph.D. Candidate in Computer Science, University of Illinois Urbana-Champaign

I work on diffusion-based controllable image and video generation at UIUC, advised by Yu-Xiong Wang. I design the supervision, training data, and curricula that teach pretrained generative models new capabilities. So far these include new visual controls, multi-view and temporal consistency, and identity-consistent video at minute scale.

I am now training visual world models through multimodal post-training. They predict the next screen after a UI action, physical events from observed video, and robot–object interactions under a prescribed motion.

Graduating Dec. 2026 Open to Research Scientist / Research Engineer roles starting 2027

Portrait of Jun-Kun Chen

Selected Publications

Google Scholar

Static 3D scenes

Spatial consistency

Videos & 4D scenes

Temporal consistency

Human-centric video

Multimodal control, long horizon

Interactive worlds · current

Predicting visual outcomes

CVPR 2024First author · with Meta Reality Labs Zurich

ConsistDreamer: 3D-Consistent 2D Diffusion for High-Fidelity Scene Editing

Jun-Kun Chen, Samuel Rota Bulò, Norman Müller, Lorenzo Porzi, Peter Kontschieder, Yu-Xiong Wang

Teaches 2D diffusion to edit 3D scenes consistently across views. Uses cross-view self-supervision and staged adaptation, with consistency losses along the whole denoising trajectory. An asynchronous multi-GPU pipeline overlaps diffusion training and image generation with scene fitting.

Under reviewFirst author · with SpreeAI

Dress&Dance: Dress up and Dance as You Like It

Jun-Kun Chen, Aayush Bansal, Minh Phuoc Vo, Yu-Xiong Wang

Teaches pretrained video diffusion new garment, identity, and motion controls through target-steered supervision and staged training. Generates 1152×720, 24-FPS try-on videos. Scores 20–59% lower FVD than a Kling 1.6 try-on-and-animation pipeline on Internet and captured benchmarks.

Show all 14 publications Hide full publication list
  1. 2026

    1. HDEdit: Editing Videos and 3D Scenes with Video Diffusion Through Hierarchical Task Decomposition3DV 2026Yanming Zhang*, Jun-Kun Chen*, Jipeng Lyu, Yu-Xiong Wang ProjectarXiv
  2. 2025

    1. Virtual Fitting Room: Generating Arbitrarily Long Videos of Virtual Try-On from a Single ImageNeurIPS 2025Jun-Kun Chen, Aayush Bansal, Minh Phuoc Vo, Yu-Xiong Wang ProjectarXiv
    2. Dress&Dance: Dress up and Dance as You Like ItUnder reviewJun-Kun Chen, Aayush Bansal, Minh Phuoc Vo, Yu-Xiong Wang ProjectarXiv
  3. 2024

    1. ProEdit: Simple Progression is All You Need for High-Quality 3D Scene EditingNeurIPS 2024Jun-Kun Chen, Yu-Xiong Wang ProjectarXiv
    2. SceneCraft: Layout-Guided 3D Scene GenerationNeurIPS 2024Xiuyu Yang*, Yunze Man*, Jun-Kun Chen, Yu-Xiong Wang ProjectarXiv
    3. ConsistDreamer: 3D-Consistent 2D Diffusion for High-Fidelity Scene EditingCVPR 2024Jun-Kun Chen, Samuel Rota Bulò, Norman Müller, Lorenzo Porzi, Peter Kontschieder, Yu-Xiong Wang ProjectarXiv
    4. Instruct 4D-to-4D: Editing 4D Scenes as Pseudo-3D Scenes Using 2D DiffusionCVPR 2024Linzhan Mou*, Jun-Kun Chen*, Yu-Xiong Wang ProjectarXiv
  4. 2023

    1. Contrastive Learning Relies More on Spatial Inductive Bias Than Supervised Learning: An Empirical StudyICCV 2023Yuanyi Zhong*, Haoran Tang*, Jun-Kun Chen*, Yu-Xiong Wang
    2. NeuralEditor: Editing Neural Radiance Fields via Manipulating Point CloudsCVPR 2023Jun-Kun Chen*, Jipeng Lyu*, Yu-Xiong Wang
  5. 2022

    1. PointTree: Transformation-Robust Point Cloud Encoder with Relaxed K-D TreesECCV 2022Jun-Kun Chen, Yu-Xiong Wang
    2. Is Self-Supervised Contrastive Learning More Robust Than Supervised Learning?ICML 2022 WorkshopYuanyi Zhong*, Haoran Tang*, Jun-Kun Chen, Jian Peng, Yu-Xiong Wang
    3. TorchDrug: A Powerful and Flexible Machine Learning Platform for Drug DiscoveryarXiv 2022Zhaocheng Zhu, Chence Shi, Zuobai Zhang, …, Jun-Kun Chen, …, Jian Tang
  6. 2021

    1. RNNLogic: Learning Logic Rules for Reasoning on Knowledge GraphsICLR 2021Meng Qu*, Jun-Kun Chen*, Louis-Pascal Xhonneux, Yoshua Bengio, Jian Tang
  7. 2020

    1. Pre-training Assisted Scene-aware Dialogue GenerationDSTC8 Workshop at AAAI 2020Jun-Kun Chen, Huang He, Siqi Bao, Fan Wang

* Equal contribution. Publications appear under Jun-Kun Chen; earlier papers may use Junkun Chen or J. Chen.

Current Research: Generative World Models

Research direction: connect language reasoning with visual generation, and use the outcomes the model predicts as training feedback.

Training-data example: screen + tap → predicted next screen (middle) vs. real (right).

Computer-use visual world model

Image diffusion · SenseNova MoT

Built a next-screen world model on SenseNova MoT: a language Reasoner describes what an action changes and what must stay the same, and a diffusion Generator adapted through diffusion post-training renders the next screenshot. Ongoing work trains the Reasoner on a teacher's corrections of the model's own rendering mistakes.

PhysicsIQ example: past video → predicted future (middle) vs. real (right).

Physical-event world model

Video diffusion · Cosmos3

Adapted Cosmos3 video diffusion to predict physical events from past RGB frames, conditioned on structured event descriptions. Implemented teacher distillation: a teacher that sees the whole video supervises a language Reasoner that sees only the past frames. An earlier online variant also updated the Generator from generated-versus-real comparisons.

Held-out DROID scene: predicted (left) vs. real (right) interaction video.

Robot-interaction world model

Motion-conditioned video diffusion · Cosmos3

Trained a motion-conditioned video world model on simulated and real robot data (MimicGen, RoboTwin, DROID, RH20T, AgiBot). It predicts robot–object interaction video from a starting scene, recent frames, and a rendering of the prescribed robot motion. Ongoing work studies adaptation to new robot and camera setups without retraining.

Experience

Meta Reality Labs

Research Scientist Intern Bay Area, CA, U.S. · Zurich, Switzerland

May–Dec. 2025Bay Area · part-time continuation Aug.–Dec.

Designed and implemented a GPU-role-separated pipeline for data processing and model training at 128–256-GPU scale. Built on Meta's distributed infrastructure to support 3D/video-generation research.

May–Aug. 2023Zurich

Took ConsistDreamer (CVPR 2024) from method design through implementation and evaluation. Designed cross-view self-supervision and staged diffusion adaptation. Built an asynchronous pipeline that overlaps diffusion training and image generation with scene fitting.

SpreeAI

AI Research Intern Remote, U.S.

May 2024 – May 2025

Dress&Dance: taught pretrained video diffusion new garment, identity, and motion controls. Designed and implemented target-steered supervision, training triplets with synthetic conditions and real targets, and staged image/video curricula. Evaluated garment and identity fidelity, motion, and video quality.

Virtual Fitting Room (NeurIPS 2025): designed training on arbitrary pairs of short clips. The model learns reference appearance and prefix continuation, with no canonical 360° anchors needed for training. At inference, a generated 360° reference and anchored autoregression keep the video consistent at minute scale. Targeted augmentation trains a refiner to remove the generator's typical artifacts.

Mila – Quebec AI Institute

Research Intern Montreal, Canada

Feb.–Aug. 2020Knowledge-graph reasoning with Prof. Jian Tang. Co-first author of RNNLogic (ICLR 2021).

Baidu, NLP Department

Research / Development Intern Beijing, China

July 2019 – Jan. 2020Scene-aware dialogue generation. Top-3 system at DSTC8, presented at the DSTC8 Workshop at AAAI 2020.

Education

University of Illinois Urbana-Champaign

Ph.D. Candidate in Computer Science · 2021 – Dec. 2026 (expected)
Advisor: Yu-Xiong Wang

Tsinghua University

B.Eng. in Computer Science, Yao Class (IIIS) · 2017 – 2021
Major GPA: 3.94/4.0

Skills

Tools: PyTorch, Python, C++, FSDP2, torch.compile, Slurm, MuJoCo

Models: image/video diffusion, diffusion Transformers (DiT), video world models (NVIDIA Cosmos), unified multimodal models (mixture-of-transformers), vision-language models, LoRA

Practice: training diffusion models with rectified-flow / flow-matching objectives, diffusion post-training (supervised fine-tuning, LoRA, online teacher distillation, reward-based tuning with DiffusionNFT), action and multimodal conditioning, video and robot-data pipelines, multi-GPU training with exact checkpoint resume, held-out benchmark design, human and VLM-judge evaluation, paired-bootstrap statistics

Selected Honors

  • ICPC EC-Final — Gold Medal, 4th place (2018)
  • ICPC Qingdao Regional — Gold Medal, 2nd place (2018)
  • CCPC Harbin Regional — Gold Medal (2017)
  • National Olympiad in Informatics — Gold Medal (2016)