← Search

Tomer Koren

55 accepted papers

2026

Flat Minima and Generalization: Insights from Stochastic Convex Optimization

ICML 2026poster

Understanding the generalization behavior of learning algorithms is a central goal of learning theory. A recently emerging explanation is that learning algorithms are successful in practice because they converge to flat minima, which have been consistently associated with improved generalization per…

Cited by 0SourceScholar
2026

Optimal Learning from Label Proportions with General Loss Functions

ICML 2026poster

Motivated by problems in online advertising, we address the task of Learning from Label Proportions (LLP). We introduce a novel and versatile low-variance debiasing methodology to learn from aggregate label information, significantly advancing the state of the art in LLP. Our debiasing approach exhi…

Cited by 0SourceScholar
2026

Towards Fully Parameter-Free Stochastic Optimization: Grid Search with Self-Bounding Analysis

ICML 2026poster

Parameter-free stochastic optimization aims to design algorithms that are agnostic to the underlying problem parameters while still achieving convergence rates competitive with optimally tuned methods. While some parameter-free methods do not require the specific values of the problem parameters, th…

Cited by 0SourceScholar
2025

Convergence of Policy Mirror Descent Beyond Compatible Function Approximation

ICML 2025poster

Modern policy optimization methods roughly follow the policy mirror descent (PMD) algorithmic template, for which there are by now numerous theoretical convergence results. However, most of these either target tabular environments, or can be applied effectively only when the class of policies be…

Cited by 0SourcePDFScholar
2025

Fast Last-Iterate Convergence of SGD in the Smooth Interpolation Regime

NeurIPS 2025poster

We study population convergence guarantees of stochastic gradient descent (SGD) for smooth convex objectives in the interpolation regime, where the noise at optimum is zero or near zero. The behavior of the last iterate of SGD in this setting---particularly with large (constant) stepsizes---has rece…

Cited by 0SourceScholar
2025

Faster Stochastic Optimization with Arbitrary Delays via Adaptive Asynchronous Mini-Batching

ICML 2025poster

We consider the problem of asynchronous stochastic optimization, where an optimization algorithm makes updates based on stale stochastic gradients of the objective that are subject to an arbitrary (possibly adversarial) sequence of delays. We present a procedure which, for any given $q \in (0,1]$, t…

Cited by 0SourcePDFScholar
2025

From Contextual Combinatorial Semi-Bandits to Bandit List Classification: Improved Sample Complexity with Sparse Rewards

NeurIPS 2025poster

We study the problem of contextual combinatorial semi-bandits, where input contexts are mapped into subsets of size $m$ of a collection of $K$ possible actions. In each round of the interaction, the learner observes feedback consisting of the realized reward of the predicted actions. Motivated by pr…

Cited by 0SourceScholar
2025

Locally Optimal Descent for Dynamic Stepsize Scheduling

AISTATS 2025poster

We introduce a novel dynamic learning-rate scheduling scheme grounded in theory with the goal of simplifying the manual and time-consuming tuning of schedules in practice. Our approach is based on estimating the locally-optimal stepsize, guaranteeing maximal descent in the direction of the stochast…

Cited by 0SourceScholar
2025

Multiclass Loss Geometry Matters for Generalization of Gradient Descent in Separable Classification

NeurIPS 2025poster

We study the generalization performance of unregularized gradient methods for separable linear classification. While previous work mostly deal with the binary case, we focus on the multiclass setting with $k$ classes and establish novel population risk bounds for Gradient Descent for loss functions…

Cited by 0SourceScholar
2025

Nearly Optimal Sample Complexity for Learning with Label Proportions

ICML 2025poster

We investigate Learning from Label Proportions (LLP), a partial information setting where examples in a training set are grouped into bags, and only aggregate label values in each bag are available. Despite the partial observability, the goal is still to achieve small regret at the level of individu…

Cited by 0SourcePDFScholar
2025

Optimal Rates in Continual Linear Regression via Increasing Regularization

NeurIPS 2025poster

We study realizable continual linear regression under random task orderings, a common setting for developing continual learning theory. In this setup, the worst-case expected loss after $k$ learning iterations admits a lower bound of $\Omega(1/k)$. However, prior work using an unregularized scheme…

Cited by 0SourceScholar
2025

Rapid Overfitting of Multi-Pass SGD in Stochastic Convex Optimization

ICML 2025spotlight

We study the out-of-sample performance of multi-pass stochastic gradient descent (SGD) in the fundamental stochastic convex optimization (SCO) model. While one-pass SGD is known to achieve an optimal $\Theta(1/\sqrt{n})$ excess population loss given a sample of size $n$, much less is understood abou…

Cited by 0SourcePDFScholar
2024

Fast Rates for Bandit PAC Multiclass Classification

NeurIPS 2024poster

We study multiclass PAC learning with bandit feedback, where inputs are classified into one of $K$ possible labels and feedback is limited to whether or not the predicted labels are correct. Our main contribution is in designing a novel learning algorithm for the agnostic $(\varepsilon,\delta)$-PAC…

Cited by 1SourcePDFScholar
2024

Rate-Optimal Policy Optimization for Linear Markov Decision Processes

ICML 2024oral

We study regret minimization in online episodic linear Markov Decision Processes, and propose a policy optimization algorithm that is computationally efficient, and obtains rate optimal $\widetilde O (\sqrt K)$ regret where $K$ denotes the number of episodes. Our work is the first to establish the o…

Cited by 12SourcePDFScholar
2023

Improved Regret for Efficient Online Reinforcement Learning with Linear Function Approximation

ICML 2023poster

We study reinforcement learning with linear function approximation and adversarially changing cost functions, a setup that has mostly been considered under simplifying assumptions such as full information feedback or exploratory conditions. We present a computationally efficient policy optimization…

Cited by 22SourcePDFScholar
2023

Near-Optimal Algorithms for Private Online Optimization in the Realizable Regime

ICML 2023poster

We consider online learning problems in the realizable setting, where there is a zero-loss solution, and propose new Differentially Private (DP) algorithms that obtain near-optimal regret bounds. For the problem of online prediction from experts, we design new algorithms that obtain near-optimal reg…

Cited by 12SourcePDFScholar
2023

Regret Minimization and Convergence to Equilibria in General-sum Markov Games

ICML 2023poster

An abundance of recent impossibility results establish that regret minimization in Markov games with adversarial opponents is both statistically and computationally intractable. Nevertheless, none of these results preclude the possibility of regret minimization under the assumption that all parties…

Cited by 30SourcePDFScholar
2023

SGD with AdaGrad Stepsizes: Full Adaptivity with High Probability to Unknown Parameters, Unbounded Gradients and Affine Variance

ICML 2023poster

We study Stochastic Gradient Descent with AdaGrad stepsizes: a popular adaptive (self-tuning) method for first-order stochastic optimization. Despite being well studied, existing analyses of this method suffer from various shortcomings: they either assume some knowledge of the problem parameters, im…

Cited by 28SourcePDFScholar
2022

Better Best of Both Worlds Bounds for Bandits with Switching Costs

NeurIPS 2022accept

We study best-of-both-worlds algorithms for bandits with switching cost, recently addressed by Rouyer et al., 2021. We introduce a surprisingly simple and effective algorithm that simultaneously achieves minimax optimal regret bound (up to logarithmic factors) of $\mathcal{O}(T^{2/3})$ in the oblivi…

Cited by 21SourcePDFScholar
2021

Asynchronous Stochastic Optimization Robust to Arbitrary Delays

NeurIPS 2021poster

We consider the problem of stochastic optimization with delayed gradients in which, at each time step $t$, the algorithm makes an update using a stale stochastic gradient from step $t - d_t$ for some arbitrary delay $d_t$. This setting abstracts asynchronous distributed optimization where a centra…

Cited by 37SourcePDFScholar
2021

Online Policy Gradient for Model Free Learning of Linear Quadratic Regulators with $\sqrt$T Regret

ICML 2021spotlight

We consider the task of learning to control a linear dynamical system under fixed quadratic costs, known as the Linear Quadratic Regulator (LQR) problem. While model-free approaches are often favorable in practice, thus far only model-based methods, which rely on costly system identification, have b…

Cited by 18SourcePDFScholar
2021

Private Stochastic Convex Optimization: Optimal Rates in L1 Geometry

ICML 2021oral

Stochastic convex optimization over an $\ell_1$-bounded domain is ubiquitous in machine learning applications such as LASSO but remains poorly understood when learning with differential privacy. We show that, up to logarithmic factors the optimal excess population loss of any $(\epsilon,\delta)$-dif…

Cited by 114SourcePDFScholar
2021

Stochastic Multi-Armed Bandits with Unrestricted Delay Distributions

ICML 2021spotlight

We study the stochastic Multi-Armed Bandit (MAB) problem with random delays in the feedback received by the algorithm. We consider two settings: the {\it reward dependent} delay setting, where realized delays may depend on the stochastic rewards, and the {\it reward-independent} delay setting. Our m…

Cited by 61SourcePDFScholar
2020

Bandit Linear Control

NeurIPS 2020spotlight

We consider the problem of controlling a known linear dynamical system under stochastic noise, adversarially chosen costs, and bandit feedback. Unlike the full feedback setting where the entire cost function is revealed after each decision, here only the cost incurred by the learner is observed. We…

Cited by 25SourcePDFScholar
2020

Can Implicit Bias Explain Generalization? Stochastic Convex Optimization as a Case Study

NeurIPS 2020poster

The notion of implicit bias, or implicit regularization, has been suggested as a means to explain the surprising generalization ability of modern-days overparameterized learning algorithms. This notion refers to the tendency of the optimization algorithm towards a certain structured solution that of…

Cited by 24SourcePDFScholar
2020

Stochastic Optimization with Laggard Data Pipelines

NeurIPS 2020poster

State-of-the-art optimization is steadily shifting towards massively parallel pipelines with extremely large batch sizes. As a consequence, CPU-bound preprocessing and disk/memory/network operations have emerged as new performance bottlenecks, as opposed to hardware-accelerated gradient computations…

Cited by 17SourcePDFScholar
2019

Robust Bi-Tempered Logistic Loss Based on Bregman Divergences

NeurIPS 2019poster

We introduce a temperature into the exponential function and replace the softmax output layer of the neural networks by a high-temperature generalization. Similarly, the logarithm in the loss we use for training is replaced by a low-temperature logarithm. By tuning the two temperatures, we create lo…

2019

Semi-Cyclic Stochastic Gradient Descent

ICML 2019oral

We consider convex SGD updates with a block-cyclic structure, i.e., where each cycle consists of a small number of blocks, each with many samples from a possibly different, block-specific, distribution. This situation arises, e.g., in Federated Learning where the mobile devices available for updates…

Cited by 134SourcePDFScholar
2018

Online Linear Quadratic Control

ICML 2018oral

We study the problem of controlling linear time-invariant systems with known noisy dynamics and adversarially chosen quadratic losses. We present the first efficient online learning algorithms in this setting that guarantee $O(\sqrt{T})$ regret under mild assumptions, where $T$ is the time horizon.…

Cited by 169SourcePDFScholar
2016

Online Pricing with Strategic and Patient Buyers

NeurIPS 2016poster

We consider a seller with an unlimited supply of a single good, who is faced with a stream of $T$ buyers. Each buyer has a window of time in which she would like to purchase, and would buy at the lowest price in that window, provided that this price is lower than her private value (and otherwise, wo…

Cited by 31SourcePDFScholar
2015

Bandit Smooth Convex Optimization: Improving the Bias-Variance Tradeoff

NeurIPS 2015spotlight

Bandit convex optimization is one of the fundamental problems in the field of online learning. The best algorithm for the general bandit convex optimization problem guarantees a regret of $\widetilde{O}(T^{5/6})$, while the best known lower bound is $\Omega(T^{1/2})$. Many attemptshave been made to…

Cited by 40SourcePDFScholar