← Search

Hae Beom Lee

15 accepted papers

2025

Bayesian Neural Scaling Law Extrapolation with Prior-Data Fitted Networks

ICML 2025poster

Scaling has been a major driver of recent advancements in deep learning. Numerous empirical studies have found that scaling laws often follow the power-law and proposed several variants of power-law functions to predict the scaling behavior at larger scales. However, existing methods mostly rely on…

Cited by 0SourcePDFScholar
2025

Cost-Sensitive Freeze-thaw Bayesian Optimization for Efficient Hyperparameter Tuning

NeurIPS 2025poster

In this paper, we address the problem of cost-sensitive hyperparameter optimization (HPO) built upon freeze-thaw Bayesian optimization (BO). Specifically, we assume a scenario where users want to early-stop the HPO process when the expected performance improvement is not satisfactory with respect to…

Cited by 0SourceScholar
2024

Delta-AI: Local objectives for amortized inference in sparse graphical models

ICLR 2024poster

We present a new algorithm for amortized inference in sparse probabilistic graphical models (PGMs), which we call $\Delta$-amortized inference ($\Delta$-AI). Our approach is based on the observation that when the sampling of variables in a PGM is seen as a sequence of actions taken by an agent, spar…

2022

Meta Learning Low Rank Covariance Factors for Energy Based Deterministic Uncertainty

ICLR 2022poster

Numerous recent works utilize bi-Lipschitz regularization of neural network layers to preserve relative distances between data instances in the feature spaces of each layer. This distance sensitivity with respect to the data aids in tasks such as uncertainty calibration and out-of-distribution (OOD)…

Cited by 2SourcePDFScholar
2022

Online Hyperparameter Meta-Learning with Hypergradient Distillation

ICLR 2022spotlight

Many gradient-based meta-learning methods assume a set of parameters that do not participate in inner-optimization, which can be considered as hyperparameters. Although such hyperparameters can be optimized using the existing gradient-based hyperparameter optimization (HO) methods, they suffer from…

Cited by 9SourcePDFScholar
2022

Sequential Reptile: Inter-Task Gradient Alignment for Multilingual Learning

ICLR 2022poster

Multilingual models jointly pretrained on multiple languages have achieved remarkable performance on various multilingual downstream tasks. Moreover, models finetuned on a single monolingual downstream task have shown to generalize to unseen languages. In this paper, we first show that it is crucial…

Cited by 16SourcePDFScholar
2021

Large-Scale Meta-Learning with Continual Trajectory Shifting

ICML 2021spotlight

Meta-learning of shared initialization parameters has shown to be highly effective in solving few-shot learning tasks. However, extending the framework to many-shot scenarios, which may further enhance its practicality, has been relatively overlooked due to the technical difficulties of meta-learnin…

Cited by 19SourcePDFScholar
2020

Learning to Balance: Bayesian Meta-Learning for Imbalanced and Out-of-distribution Tasks

ICLR 2020talk

While tasks could come with varying the number of instances and classes in realistic settings, the existing meta-learning approaches for few-shot classification assume that number of instances per task and class is fixed. Due to such restriction, they learn to equally utilize the meta-knowledge acro…

Cited by 144SourcecodeScholar
2020

Meta Dropout: Learning to Perturb Latent Features for Generalization

ICLR 2020poster

A machine learning model that generalizes well should obtain low errors on unseen test examples. Thus, if we know how to optimally perturb training examples to account for test examples, we may achieve better generalization performance. However, obtaining such perturbation is not possible in standar…

Cited by 60SourcecodeScholar
2020

Meta Variance Transfer: Learning to Augment from the Others

ICML 2020poster

Humans have the ability to robustly recognize objects with various factors of variations such as nonrigid transformations, background noises, and changes in lighting conditions. However, training deep learning models generally require huge amount of data instances under diverse variations, to ensure…

Cited by 60SourcePDFScholar
2020

MetaPerturb: Transferable Regularizer for Heterogeneous Tasks and Architectures

NeurIPS 2020spotlight

Regularization and transfer learning are two popular techniques to enhance model generalization on unseen data, which is a fundamental problem of machine learning. Regularization techniques are versatile, as they are task- and architecture-agnostic, but they do not exploit a large amount of data ava…

2018

Uncertainty-Aware Attention for Reliable Interpretation and Prediction

NeurIPS 2018poster

Attention mechanism is effective in both focusing the deep learning models on relevant features and interpreting them. However, attentions may be unreliable since the networks that generate them are often trained in a weakly-supervised manner. To overcome this limitation, we introduce the notion of…