Jinheon Baek

About Me

Hi, I’m a master student in the Graduate school of AI at KAIST (MLAI Lab), advised by Prof. Sung Ju Hwang. I do research on Machine Learning for Graphs. Also, I am a non-executive technical director of Jobshopper ((주)잡쇼퍼), an AI-based education start-up company in Seoul, Korea.


KAIST (Korea Advanced Institute of Science and Technology)

  • M.S. student in Artificial Intelligence, Mar 2020 - Present.
  • GPA: 4.22 / 4.30 (99.1 / 100)

Korea University

  • Bachelor of Science in Computer Science and Engineering, Mar 2016 - Feb 2020.
  • Bachelor of Engineering in Software Technology and Enterprise Program (Interdisciplinary Program), Aug 2017 - Feb 2020.
  • GPA: 4.40 / 4.50 (98.9 / 100), GPA of Computer Science and Engineering: 4.48 / 4.50 (99.5 / 100)

Selected Publication


  • Wonyong Jeong*, Hayeon Lee*, Gun Park*, Eunyoung Hyung, Jinheon Baek, and Sung Ju Hwang. Task-Adaptive Neural Network Retrieval with Meta-Contrastive Learning. arXiv:2103.01495. [paper] (*: equal contribution)


  • Jinheon Baek*, MinKi Kang*, and Sung Ju Hwang. Accurate Learning of Graph Representations with Graph Multiset Pooling. International Conference on Learning Representation (ICLR), 2021. [paper] [code] (*: equal contribution)
  • Jinheon Baek, Dong Bok Lee, and Sung Ju Hwang. Learning to Extrapolate Knowledge: Transductive Few-shot Out-of-Graph Link Prediction. Conference on Neural Information Processing System (NeurIPS), 2020. [paper] [code]


  • Hyunjae Kim*, Yookyung Koh*, Jinheon Baek, and Jaewoo Kang. Exploring The Spatial Reasoning Ability of Neural Models in Human IQ Tests. Neural Networks, 2021. [paper] (*: equal contribution)


  • Soyeong Jeong, Jinheon Baek, ChaeHun Park, and Jong C. Park. Unsupervised Document Expansion for Information Retrieval with Stochastic Text Generation. Scholarly Document Processing at Conference of the North American Chapter of the Association for Computational Linguistics (SDP@NAACL), 2021 (to appear). (Oral Presentation)

You can see more details about me in CV (Curriculum Vitae), or about page.