activity
20172026
most citedSet Transformer: A Framework for Attention-based Permutation-Invariant Neural Networks

150 citations · 234 across the 29 of their papers we have counts for

collaborators
Showing cs.LGShow all

15 papers · 1 filter

cs.LG2026

Transformers in the Dark: Navigating Unknown Search Spaces via Bandit Feedback

Jungtaek Kim, Thomas Zeng, Ziqian Lin +5

Effective problem solving with Large Language Models (LLMs) can be enhanced when they are paired with external search algorithms. By viewing the space of diverse ideas and their fo…

cs.LG2025

ReJump: A Tree-Jump Representation for Analyzing and Improving LLM Reasoning

Yuchen Zeng, Shuibai Zhang, Wonjun Kang +9

Large Reasoning Models (LRMs) are Large Language Models (LLMs) explicitly trained to generate long-form Chain-of-Thoughts (CoTs), achieving impressive success on challenging tasks…

cs.LG2025

VersaPRM: Multi-Domain Process Reward Model via Synthetic Reasoning Data

Thomas Zeng, Shuibai Zhang, Shutong Wu +13

Process Reward Models (PRMs) have proven effective at enhancing mathematical reasoning for Large Language Models (LLMs) by leveraging increased inference-time computation. However,…

cs.LG2024

Exploiting Preferences in Loss Functions for Sequential Recommendation via Weak Transitivity

Hyunsoo Chung, Jungtaek Kim, Hyungeun Jo +1

A choice of optimization objective is immensely pivotal in the design of a recommender system as it affects the general modeling process of a user's intent from previous interactio…

cs.LG2024

Beyond Regrets: Geometric Metrics for Bayesian Optimization

Jungtaek Kim

Bayesian optimization is a principled optimization strategy for a black-box objective function. It shows its effectiveness in a wide variety of real-world applications such as scie…

cs.LG2023

Generative Neural Fields by Mixtures of Neural Implicit Functions

Tackgeun You, Mijeong Kim, Jungtaek Kim +1

We propose a novel approach to learning the generative neural fields represented by linear combinations of implicit basis networks. Our algorithm learns basis networks in the form…

cs.LG2023★ 1 cited

Datasets and Benchmarks for Nanophotonic Structure and Parametric Design Simulations

Jungtaek Kim, Mingxuan Li, Oliver Hinder +1

Nanophotonic structures have versatile applications including solar cells, anti-reflective coatings, electromagnetic interference shielding, optical filters, and light emitting dio…

cs.LG2023

Generalized Neural Sorting Networks with Error-Free Differentiable Swap Functions

Jungtaek Kim, Jeongbeen Yoon, Minsu Cho

Sorting is a fundamental operation of all computer systems, having been a long-standing significant research topic. Beyond the problem formulation of traditional sorting algorithms…

cs.LG2023

Density Ratio Estimation-based Bayesian Optimization with Semi-Supervised Learning

Jungtaek Kim

Bayesian optimization has attracted huge attention from diverse research areas in science and engineering, since it is capable of efficiently finding a global optimum of an expensi…

cs.LG2022

Budget-Aware Sequential Brick Assembly with Efficient Constraint Satisfaction

Seokjun Ahn, Jungtaek Kim, Minsu Cho +1

We tackle the problem of sequential brick assembly with LEGO bricks to create combinatorial 3D structures. This problem is challenging since this brick assembly task encompasses th…

cs.LG2022★ 9 cited

On Evaluation Metrics for Graph Generative Models

Rylee Thompson, Boris Knyazev, Elahe Ghalebi +2

In image generation, generative models can be evaluated naturally by visually inspecting model outputs. However, this is not always the case for graph generative models (GGMs), mak…

cs.LG2021★ 10 cited

Brick-by-Brick: Combinatorial Construction with Deep Reinforcement Learning

Hyunsoo Chung, Jungtaek Kim, Boris Knyazev +4

Discovering a solution in a combinatorial space is prevalent in many real-world problems but it is also challenging due to diverse complex constraints and the vast number of possib…

cs.LG2020★ 8 cited

Bootstrapping Neural Processes

Juho Lee, Yoonho Lee, Jungtaek Kim +3

Unlike in the traditional statistical modeling for which a user typically hand-specify a prior, Neural Processes (NPs) implicitly define a broad class of stochastic processes with…

cs.LG2019★ 5 cited

MxML: Mixture of Meta-Learners for Few-Shot Classification

Minseop Park, Jungtaek Kim, Saehoon Kim +2

A meta-model is trained on a distribution of similar tasks such that it learns an algorithm that can quickly adapt to a novel task with only a handful of labeled examples. Most of…

cs.LG2018★ 150 cited

Set Transformer: A Framework for Attention-based Permutation-Invariant Neural Networks

Juho Lee, Yoonho Lee, Jungtaek Kim +3

Many machine learning tasks such as multiple instance learning, 3D shape recognition, and few-shot image classification are defined on sets of instances. Since solutions to such pr…