← Search

Aaron Klein

15 accepted papers

2026

Improving LLM-based Global Optimization with Search Space Partitioning

ICLR 2026poster

Large Language Models (LLMs) have recently emerged as effective surrogate models and candidate generators within global optimization frameworks for expensive blackbox functions. Despite promising results, LLM-based methods often struggle in high-dimensional search spaces or when lacking domain-speci…

Cited by 0SourcecodeScholar
2025

Hyperband-based Bayesian Optimization for Black-box Prompt Selection

ICML 2025poster

Optimal prompt selection is crucial for maximizing large language model (LLM) performance on downstream tasks, especially in black-box settings where models are only accessible via APIs. Black-box prompt selection is challenging due to potentially large, combinatorial search spaces, absence of gradi…

Cited by 3SourcePDFScholar
2024

HW-GPT-Bench: Hardware-Aware Architecture Benchmark for Language Models

NeurIPS 2024poster

The increasing size of language models necessitates a thorough analysis across multiple dimensions to assess trade-offs among crucial hardware metrics such as latency, energy consumption, GPU memory usage, and performance. Identifying optimal model configurations under specific hardware constraints…

Cited by 1SourcePDFScholar
2023

Optimizing Hyperparameters with Conformal Quantile Regression

ICML 2023poster

Many state-of-the-art hyperparameter optimization (HPO) algorithms rely on model-based optimizers that learn surrogate models of the target function to guide the search. Gaussian processes are the de facto surrogate model due to their ability to capture uncertainty. However, they make strong assumpt…

2021

BORE: Bayesian Optimization by Density-Ratio Estimation

ICML 2021oral

Bayesian optimization (BO) is among the most effective and widely-used blackbox optimization methods. BO proposes solutions according to an explore-exploit trade-off criterion encoded in an acquisition function, many of which are computed from the posterior predictive of a probabilistic surrogate mo…

2021

HPOBench: A Collection of Reproducible Multi-Fidelity Benchmark Problems for HPO

NeurIPS 2021poster

To achieve peak predictive performance, hyperparameter optimization (HPO) is a crucial component of machine learning and its applications. Over the last years, the number of efficient algorithms and tools for HPO grew substantially. At the same time, the community is still lacking realistic, diverse…

Cited by 105SourcecodeScholar
2021

Hyperparameter Transfer Learning with Adaptive Complexity

AISTATS 2021poster

Bayesian optimization (BO) is a data-efficient approach to automatically tune the hyperparameters of machine learning models. In practice, one frequently has to solve similar hyperparameter tuning problems sequentially. For example, one might have to tune a type of neural network learned across a se…

2019

Meta-Surrogate Benchmarking for Hyperparameter Optimization

NeurIPS 2019poster

Despite the recent progress in hyperparameter optimization (HPO), available benchmarks that resemble real-world scenarios consist of a few and very large problem instances that are expensive to solve. This blocks researchers and practitioners no only from systematically running large-scale compariso…

2019

NAS-Bench-101: Towards Reproducible Neural Architecture Search

ICML 2019oral

Recent advances in neural architecture search (NAS) demand tremendous computational resources, which makes it difficult to reproduce experiments and imposes a barrier-to-entry to researchers without access to large-scale computation. We aim to ameliorate these problems by introducing NAS-Bench-101,…

2018

BOHB: Robust and Efficient Hyperparameter Optimization at Scale

ICML 2018oral

Modern deep learning methods are very sensitive to many hyperparameters, and, due to the long training times of state-of-the-art models, vanilla Bayesian hyperparameter optimization is typically computationally infeasible. On the other hand, bandit-based configuration evaluation approaches based on…

2018

Uncertainty Estimates and Multi-Hypotheses Networks for Optical Flow

ECCV 2018poster

Optical flow estimation can be formulated as an end-to-end supervised learning problem, which yields estimates with a superior accuracy-runtime tradeoff compared to alternative methodology. In this paper, we make such networks estimate their local uncertainty about the correctness of their predictio…

Cited by 270SourcePDFScholar
2017

Fast Bayesian Optimization of Machine Learning Hyperparameters on Large Datasets

AISTATS 2017poster

Bayesian optimization has become a successful tool for hyperparameter optimization of machine learning algorithms, such as support vector machines or deep neural networks. Despite its success, for large datasets, training and validating a single configuration often takes hours, days, or even weeks,…

2017

Learning Curve Prediction with Bayesian Neural Networks

ICLR 2017poster

Different neural network architectures, hyperparameters and training protocols lead to different performances as a function of time. Human experts routinely inspect the resulting learning curves to quickly terminate runs with poor hyperparameter settings and thereby considerably speed up manual hype…

Cited by 301SourceScholar
2016

Bayesian Optimization with Robust Bayesian Neural Networks

NeurIPS 2016oral

Bayesian optimization is a prominent method for optimizing expensive to evaluate black-box functions that is prominently applied to tuning the hyperparameters of machine learning algorithms. Despite its successes, the prototypical Bayesian optimization approach - using Gaussian process models - does…

Cited by 593SourcePDFScholar
2015

Efficient and Robust Automated Machine Learning

NeurIPS 2015poster

The success of machine learning in a broad range of applications has led to an ever-growing demand for machine learning systems that can be used off the shelf by non-experts. To be effective in practice, such systems need to automatically choose a good algorithm and feature preprocessing steps for a…