Jinheon Baek

CV (Curriculum Vitae)

In this page, I try to keep all my records during undergraduate and graduate schools,
thus it might not appropriate to the person who want to see the brief introduction here.

About Me

Hi, I’m a master’s student in the Graduate school of AI at KAIST (MLAI Lab), where I am fortunate to be advised by Prof. Sung Ju Hwang. I am interested in Machine Learning for Graphs. Prior to studying at KAIST, I obtained my B.S. (Computer Science and Engineering) and B.E. (Software Technology and Enterprise Program) degrees at Korea University, where I partially studied Machine Learning under the guidance of Prof. Jaewoo Kang.

Also, I was a technical director of Jobshopper ((주)잡쇼퍼), an AI-based education startup company in Seoul, Korea. During one year of full-time work experience at Jobshopper as a lead engineer and researcher, I led the development and research groups of more than 10 people. Now, I am technically advising Jobshopper, as well as MajorMap (메이저맵(주)) spin-offed from Jobshopper.

Research Interests

My research interest is mainly on developing novel machine learning models and algorithms for graphs.
I recently focus on following three topics:
    1) representing or generating graph-structured data such as a molecule, society, and neural network,
    2) modeling interactions between knowledge from relational-structured or even raw data with graphs,
    3) and tackling challenges on real-world graphs having noise or temporality.


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)

Here is the list of courses that I took at KAIST:
2021: Machine Learning Thoery (S), M.S. Seminar <Colloquium> (S), M.S. Thesis Research (S)
2020: Advanced Deep Learning (A+), Deep Learning (A+), Graph Mining and Social Network Analysis (A+), Machine Learning for AI (A0), M.S. Seminar <Colloquium> (S), Programming for AI (S), Optimization for AI (S), Special Lecture on Leadership (S), Scientific Writing (S)
○ A+: 4.3/4.3, A0: 4.0/4.3, …, S: Satisfactory, U: Unsatisfactory, …

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)

Here is the list of Computer Science and Engineering courses that I took at Korea University:
2019: Introduction to Convex Optimization (A+), Software Project Management (A+), Operating Systems (A+), Project for Graduation (Capstone Design) I (P)
2018: Artificial Intelligence (A+), Data Science (A+), Databases (A+), Programming Language (A+), Software Engineering (A+), Machine Learning (A+), Entrepreneurship & Leadership (A+), Creative Software Design (A+), Computer Network (A+), Computer System Design (A), Internship Program I (P), Internship Program II (P)
2017: Python Programming for Everybody (A+), Discrete Mathematics (A+), Theory of Computation (A+), Digital Logic Design (A+), Data Structure (A+), Algorithms (A+), Computer Architecture (A+)
2016: Computer Programming I (A+), Mathematics for Computer Science I (A+), Computer Programming II (A+), Mathematics for Computer Science II (A+)
○ A+: 4.5/4.5, A: 4.0/4.5, …, P: Pass, F: Fail


* denotes the equal contribution.


  • Jaehyeong Jo*, Jinheon Baek*, Seul Lee*, Dongki Kim, Minki Kang, and Sung Ju Hwang. Edge Representation Learning with Hypergraphs. Conference on Neural Information Processing Systems (NeurIPS), 2021. [paper]
  • Wonyong Jeong*, Hayeon Lee*, Gun Park*, Eunyoung Hyung, Jinheon Baek, and Sung Ju Hwang. Task-Adaptive Neural Network Retrieval with Meta-Contrastive Learning. Conference on Neural Information Processing Systems (NeurIPS), 2021. (Spotlight Presentation) [paper]
  • Jinheon Baek*, MinKi Kang*, and Sung Ju Hwang. Accurate Learning of Graph Representations with Graph Multiset Pooling. International Conference on Learning Representations (ICLR), 2021. [paper] [code]
  • 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 Systems (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]


  • 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. (Oral Presentation) [paper]

Domestic (Korean) Journals & Conferences

  • Jinheon Baek, Hayeon Kim, and Kiwon Kwon. Artificial Intelligence-Based High School Course and University Major Recommendation System for Course-Related Career Exploration. KIPS Transactions on Software and Data Engineering (KTSDE), 2021.
  • Jinheon Baek, Minki Kang, and Sung Ju Hwang. Graph Representation Learning with Attention-based Set Pooling. Conference of Korean Artificial Intelligence Association (CKAIA), 2020. (Best Paper)
  • Jinheon Baek, Gwanghoon Jang, Soyeong Jeong, Donghyeon Park, Kiwon Kwon, and Jaewoo Kang. Embedding Academic Majors and Lectures for Analyzing Departments in University. Korea Computer Congress (KCC), 2019. (Oral Presentation)
  • Donghyeon Park, Yonggyu Park, Buru Chang, Jinheon Baek, and Jaewoo Kang. Embedding Food Ingredients Based on Chemical Combination in Dense Vector Space. Korea Computer Congress (KCC), 2018. (Best Paper)
  • Jinheon Baek, Dakyeong Lee, Chaeyeon Hong, and Byeongtae Ahn. Multimodal Approach for Blocking Obscene and Violent Contents. Journal of Convergence for Information Technology (JCIT), 7(6), 2017.


  • Kiwon Kwon, and Jinheon Baek. Method for Consulting on Exploration Activities based on Target Departments and Curriculum Achievement Standards. Korean patent number: 10-2021-0017764 (filed on Feb. 08, 2021).
  • Kiwon Kwon, and Jinheon Baek. Method for Building an Artificial Intelligence based Research Topics Database. Korean patent number: 10-2021-0017765 (filed on Feb. 08, 2021).

Academic Services

Conference Reviewers

  • International Conference on Learning Representations (ICLR), 2022
  • Neural Information Processing Systems (NeurIPS), 2021 [link]
  • International Conference on Machine Learning (ICML), 2021 - Selected as one of best reviewers (Top 10%) [link]

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