Publications - Jinwoo Kim

Look Back: Reinforcement Learning with Counterfactual World Models

Jinwoo Kim*, Toghrul Abbasov*, Lucius E.J. Bynum, Aahlad Manas Puli, Stephanie Milani, Hyunwoo Ryu, Seunghoon Hong, Kyunghyun Cho^†, Rajesh Ranganath^†

Under review

Why Does Rewriting Pretraining Data Improve Generalization?

Jiyun Park*, Jinwoo Kim*, Kyungbin Min, Hoyeon Chang, Sohee Yang, Kyunghyun Cho^†, Seunghoon Hong^†

Under review

Generalization in Continuous Flows for Categorical Data

Manan Agarwal, Jinwoo Kim

Under review

Asynchronous Flow Map Language Models

Chanhyuk Lee, Wonjung Kim, Patrick Pynadath, Jaehoon Yoo, Satoshi Hayakawa, Ruqi Zhang, Jinwoo Kim^†, Seunghoon Hong^†

arXiv 2026

paper code

Large Flow Language Models

Arnav Chavan*, Chanhyuk Lee*, Udbhav Bamba*, Jaehoon Yoo, Seunghoon Hong, Aditi Raghunathan^†, Jinwoo Kim^†

arXiv 2026

paper code

Thinking with Looped Flows

Ayhan Suleymanzade, Chanhyuk Lee, Floor Eijkelboom, Nicholas M. Boffi, İsmail İlkan Ceylan^†, Jinwoo Kim^†

arXiv 2026

paper code

MUX: Continuous Reasoning via Multiplexed Tokens

Ayhan Suleymanzade, Halil Alperen Gozeten, Michael Bronstein, İsmail İlkan Ceylan^†, Jinwoo Kim^†

NeurIPS 2026 Spotlight (292/30709=0.95%)

paper code

Flow Map Language Models: One-step Language Modeling via Continuous Denoising

Chanhyuk Lee, Jaehoon Yoo, Manan Agarwal, Sheel Shah, Jerry Huang, Aditi Raghunathan, Seunghoon Hong, Nicholas M. Boffi^†, Jinwoo Kim^†

NeurIPS 2026 / COLM NonAR-LM Workshop 2026 Spotlight

paper code blog

Multi-View Relational Distillation for Spatial Reasoning with Vision-Language Models

Kiet T. Nguyen, Hanbo Shim, Jinwoo Kim^†, Seunghoon Hong^†

NeurIPS 2026

paper

RelAgent: LLM Agents as Data Scientists for Relational Learning

Xingyue Huang, Louis Tichelman, Jinwoo Kim, Krzysztof Olejniczak, İsmail İlkan Ceylan

NeurIPS 2026

paper

To Learn is to Wander: Learning Across Graphs and Tasks with Random Walks

Louis Tichelman, Xingyue Huang, Jinwoo Kim, İsmail İlkan Ceylan

NeurIPS 2026 GDDL Workshop

paper

One-step Generation via Riemannian Wasserstein Gradient Flows

David Li, Chanhyuk Lee, Jaehoon Yoo, Nikita Gushchin, Floor Eijkelboom, Eric Moulines, Maxim Panov, Alexander Korotin, Seunghoon Hong^†, Jinwoo Kim^†

NeurIPS 2026 DiffuLM Workshop

paper

Self-conditioned Flow Map Language Models via Fixed-point Flows

Jaehoon Yoo*, Wonjung Kim*, Floor Eijkelboom, Chanhyuk Lee, Nicholas M. Boffi, Seunghoon Hong^†, Jinwoo Kim^†

COLM NonAR-LM Workshop 2026 Oral

paper code

Posterior Refinement: Fast Language Generation via Any-Order Flow Maps

Manan Agarwal*, Sheel Shah*, Chanhyuk Lee, Jaehoon Yoo, Jerry Huang, Seunghoon Hong, Aditi Raghunathan, Jinwoo Kim^†, Nicholas M. Boffi^†

COLM NonAR-LM Workshop 2026 Spotlight

paper code

What are the Right Symmetries for Formal Theorem Proving?

Krzysztof Olejniczak, Radoslav Dimitrov, Xingyue Huang, Bernardo Cuenca Grau, Jinwoo Kim^†, İsmail İlkan Ceylan^†

ICML AI4Math Workshop 2026

paper code

Inverting Data Transformations via Diffusion Sampling

Jinwoo Kim*, Sékou-Oumar Kaba*, Jiyun Park, Seunghoon Hong^†, Siamak Ravanbakhsh^†

ICML 2026

paper code

Flock: A Knowledge Graph Foundation Model via Learning on Random Walks

Jinwoo Kim*, Xingyue Huang*, Krzysztof Olejniczak, Kyungbin Min, Michael Bronstein, Seunghoon Hong, İsmail İlkan Ceylan

ICLR 2026

paper code

Sequence Modeling with Spectral Mean Flows

Jinwoo Kim, Max Beier, Petar Bevanda, Nayun Kim, Seunghoon Hong

NeurIPS 2025

paper code

Revisiting Random Walks for Learning on Graphs

Jinwoo Kim, Olga Zaghen*, Ayhan Suleymanzade*, Youngmin Ryou, Seunghoon Hong

ICLR 2025 Spotlight (380/11672=3.26%) / ICML GRaM Workshop 2024 ELLIS Mobility Grant

paper code poster

3D Denoisers are Good 2D Teachers: Molecular Pretraining via Denoising and Cross-Modal Distillation

Sungjun Cho, Dae-Woong Jeong, Sung Moon Ko, Jinwoo Kim, Sehui Han, Seunghoon Hong, Honglak Lee, Moontae Lee

AAAI 2025 Oral

paper

Simulation-Free Training of Neural ODEs on Paired Data

Semin Kim*, Jaehoon Yoo*, Jinwoo Kim, Yeonwoo Cha, Saehoon Kim, Seunghoon Hong

NeurIPS 2024

paper code

Learning Symmetrization for Equivariance with Orbit Distance Minimization

Tien Dat Nguyen*, Jinwoo Kim*, Hongseok Yang, Seunghoon Hong

NeurIPS NeurReps Workshop 2023

paper code poster

Learning Probabilistic Symmetrization for Architecture Agnostic Equivariance

Jinwoo Kim, Tien Dat Nguyen, Ayhan Suleymanzade, Hyeokjun An, Seunghoon Hong

NeurIPS 2023 Spotlight (378/12345=3.06%)

paper code poster slides extended slides

Universal Few-shot Learning of Dense Prediction Tasks with Visual Token Matching

Donggyun Kim, Jinwoo Kim, Seongwoong Cho, Chong Luo, Seunghoon Hong

ICLR 2023 Outstanding Paper Award (4/4955=0.08%) / Silver Prize, Samsung Humantech Paper Award 2023

paper code

Pure Transformers are Powerful Graph Learners

Jinwoo Kim, Tien Dat Nguyen, Seonwoo Min, Sungjun Cho, Moontae Lee, Honglak Lee^†, Seunghoon Hong^†

NeurIPS 2022

paper code talk poster slides extended slides

Transformers Meet Stochastic Block Models: Attention with Data-Adaptive Sparsity and Cost

Sungjun Cho, Seonwoo Min, Jinwoo Kim, Moontae Lee, Honglak Lee, Seunghoon Hong

NeurIPS 2022

paper code poster

Equivariant Hypergraph Neural Networks

Jinwoo Kim, Saeyoon Oh, Sungjun Cho, Seunghoon Hong

ECCV 2022

paper code poster slides

Transformers Generalize DeepSets and Can be Extended to Graphs and Hypergraphs

Jinwoo Kim, Saeyoon Oh, Seunghoon Hong

NeurIPS 2021 / Qualcomm Innovation Fellowship Korea 2021

paper code poster slides

SetVAE: Learning Hierarchical Composition for Generative Modeling of Set-Structured Data

Jinwoo Kim*, Jaehoon Yoo*, Juho Lee, Seunghoon Hong

CVPR 2021

paper code project page poster slides

Spontaneous Retinal Waves Can Generate Long-Range Horizontal Connectivity in Visual Cortex

Jinwoo Kim*, Min Song*, Jaeson Jang, Se-Bum Paik

The Journal of Neuroscience 40(34) 2020

paper