← Search

Kwang-Sung Jun

31 accepted papers

2026

Coverage Improvement and Fast Convergence of On-policy Preference Learning

ICML 2026poster

On-policy preference learning algorithms for language model alignment such as online direct policy optimization (DPO) can significantly outperform their offline counterparts. We provide a theoretical explanation for this phenomenon by analyzing how the sampling policy's coverage evolves throughout o…

Cited by 0SourceScholar
2026

Fixed Budget is No Harder Than Fixed Confidence in Best-Arm Identification up to Logarithmic Factors

ICML 2026poster

The best-arm identification (BAI) problem is one of the most fundamental problems in interactive machine learning, which has two flavors: the fixed-budget setting (FB) and the fixed-confidence setting (FC). For $K$-armed bandits with the unique best arm, the optimal sample complexities for both sett…

Cited by 0SourceScholar
2025

Fixing the Loose Brake: Exponential-Tailed Stopping Time in Best Arm Identification

ICML 2025poster

The best arm identification problem requires identifying the best alternative (i.e., arm) in active experimentation using the smallest number of experiments (i.e., arm pulls), which is crucial for cost-efficient and timely decision-making processes. In the fixed confidence setting, an algorithm must…

Cited by 0SourcePDFScholar
2025

GL-LowPopArt: A Nearly Instance-Wise Minimax-Optimal Estimator for Generalized Low-Rank Trace Regression

ICML 2025spotlight

We present `GL-LowPopArt`, a novel Catoni-style estimator for generalized low-rank trace regression. Building on `LowPopArt` (Jang et al., 2024), it employs a two-stage approach: nuclear norm regularization followed by matrix Catoni estimation. We establish state-of-the-art estimation error bounds,…

Cited by 0SourcePDFScholar
2025

HAVER: Instance-Dependent Error Bounds for Maximum Mean Estimation and Applications to Q-Learning and Monte Carlo Tree Search

AISTATS 2025poster

We study the problem of estimating the \emph{value} of the largest mean among $K$ distributions via samples from them (rather than estimating \emph{which} distribution has the largest mean), which arises from various machine learning tasks including Q-learning and Monte Carlo Tree Search (MCTS). Whi…

Cited by 0SourceScholar
2024

A Unified Confidence Sequence for Generalized Linear Models, with Applications to Bandits

NeurIPS 2024poster

We present a unified likelihood ratio-based confidence sequence (CS) for *any* (self-concordant) generalized linear model (GLM) that is guaranteed to be convex and numerically tight. We show that this is on par or improves upon known CSs for various GLMs, including Gaussian, Bernoulli, and Poisson.…

2024

Efficient Low-Rank Matrix Estimation, Experimental Design, and Arm-Set-Dependent Low-Rank Bandits

ICML 2024poster

We study low-rank matrix trace regression and the related problem of low-rank matrix bandits. Assuming access to the distribution of the covariates, we propose a novel low-rank matrix estimation method called *LowPopArt* and provide its recovery guarantee that depends on a novel quantity denoted by…

2024

Improved Regret Bounds of (Multinomial) Logistic Bandits via Regret-to-Confidence-Set Conversion

AISTATS 2024poster

Logistic bandit is a ubiquitous framework of modeling users’ choices, e.g., click vs. no click for advertisement recommender system. We observe that the prior works overlook or neglect dependencies in $S \geq \Vert \theta_\star \Vert_2$, where $\theta_\star \in \mathbb{R}^d$ is the unknown parameter…

2024

Noise-Adaptive Confidence Sets for Linear Bandits and Application to Bayesian Optimization

ICML 2024poster

Adapting to a priori unknown noise level is a very important but challenging problem in sequential decision-making as efficient exploration typically requires knowledge of the noise level, which is often loosely specified. We report significant progress in addressing this issue in linear bandits in…

2023

Kullback-Leibler Maillard Sampling for Multi-armed Bandits with Bounded Rewards

NeurIPS 2023poster

We study $K$-armed bandit problems where the reward distributions of the arms are all supported on the $[0,1]$ interval. Maillard sampling\cite{maillard13apprentissage}, an attractive alternative to Thompson sampling, has recently been shown to achieve competitive regret guarantees in the sub-Gaussi…

2023

Revisiting Simple Regret: Fast Rates for Returning a Good Arm

ICML 2023poster

Simple regret is a natural and parameter-free performance criterion for pure exploration in multi-armed bandits yet is less popular than the probability of missing the best arm or an $\epsilon$-good arm, perhaps due to lack of easy ways to characterize it. In this paper, we make a significant progre…

Cited by 19SourcePDFScholar
2022

An Experimental Design Approach for Regret Minimization in Logistic Bandits

AAAI 2022technical

In this work we consider the problem of regret minimization for logistic bandits. The main challenge of logistic bandits is reducing the dependence on a potentially large problem dependent constant that can at worst scale exponentially with the norm of the unknown parameter vector. Previous works ha…

Cited by 14SourcePDFScholar
2022

Improved Regret Analysis for Variance-Adaptive Linear Bandits and Horizon-Free Linear Mixture MDPs

NeurIPS 2022accept

In online learning problems, exploiting low variance plays an important role in obtaining tight performance guarantees yet is challenging because variances are often not known a priori. Recently, considerable progress has been made by Zhang et al. (2021) where they obtain a variance-adaptive regre…

Cited by 22SourcePDFScholar
2022

Jointly Efficient and Optimal Algorithms for Logistic Bandits

AISTATS 2022poster

Logistic Bandits have recently undergone careful scrutiny by virtue of their combined theoretical and practical relevance. This research effort delivered statistically efficient algorithms, improving the regret of previous strategies by exponentially large factors. Such algorithms are however striki…

Cited by 30SourcePDFScholar
2022

PopArt: Efficient Sparse Regression and Experimental Design for Optimal Sparse Linear Bandits

NeurIPS 2022accept

In sparse linear bandits, a learning agent sequentially selects an action from a fixed action set and receives reward feedback, and the reward function depends linearly on a few coordinates of the covariates of the actions. This has applications in many real-world sequential decision making problems…

2021

Improved Confidence Bounds for the Linear Logistic Model and Applications to Bandits

ICML 2021spotlight

We propose improved fixed-design confidence bounds for the linear logistic model. Our bounds significantly improve upon the state-of-the-art bound by Li et al. (2017) via recent developments of the self-concordant analysis of the logistic loss (Faury et al., 2020). Specifically, our confidence bound…

Cited by 29SourcePDFScholar
2021

Improved Regret Bounds of Bilinear Bandits using Action Space Analysis

ICML 2021spotlight

We consider the bilinear bandit problem where the learner chooses a pair of arms, each from two different action spaces of dimension $d_1$ and $d_2$, respectively. The learner then receives a reward whose expectation is a bilinear function of the two chosen arms with an unknown matrix parameter $\Th…

Cited by 10SourcePDFScholar
2019

Kernel Truncated Randomized Ridge Regression: Optimal Rates and Low Noise Acceleration

NeurIPS 2019poster

In this paper we consider the nonparametric least square regression in a Reproducing Kernel Hilbert Space (RKHS). We propose a new randomized algorithm that has optimal generalization error bounds with respect to the square loss, closing a long-standing gap between upper and lower bounds. Moreover,…

Cited by 19SourcePDFScholar
2017

Improved Strongly Adaptive Online Learning using Coin Betting

AISTATS 2017poster

This paper describes a new parameter-free online learning algorithm for changing environments. In comparing against algorithms with the same time complexity as ours, we obtain a strongly adaptive regret bound that is a factor of at least $\sqrt\log(T)$ better, where $T$ is the time horizon. Empiri…

Cited by 85SourcePDFScholar
2017

Scalable Generalized Linear Bandits: Online Computation and Hashing

NeurIPS 2017poster

Generalized Linear Bandits (GLBs), a natural extension of the stochastic linear bandits, has been popular and successful in recent years. However, existing GLBs scale poorly with the number of rounds and the number of arms, limiting their utility in practice. This paper proposes new, scalable solu…

Cited by 148SourcePDFScholar
2016

Top Arm Identification in Multi-Armed Bandits with Batch Arm Pulls

AISTATS 2016poster

We introduce a new multi-armed bandit (MAB) problem in which arms must be sampled in batches, rather than one at a time. This is motivated by applications in social media monitoring and biological experimentation where such batch constraints naturally arise. This paper develops and analyzes algorith…

Cited by 99SourcePDFScholar
2015

Human Memory Search as Initial-Visit Emitting Random Walk

NeurIPS 2015poster

Imagine a random walk that outputs a state only when visiting it for the first time. The observed output is therefore a repeat-censored version of the underlying walk, and consists of a permutation of the states or a prefix of it. We call this model initial-visit emitting random walk (INVITE). Prior…

Cited by 17SourcePDFScholar