← Search

Hyungi Lee

19 accepted papers

2026

Confidence is Not Universal: Task-Dependent Calibration and Emergent Behavior in LLMs

ICML 2026poster

Large language models (LLMs) increasingly support human decision-making, rendering human-interpretable confidence essential. However, it remains unclear whether verbalized confidence calibration generalizes across heterogeneous tasks without degrading accuracy. We show that universal confidence cali…

Cited by 0SourceScholar
2026

OCNR: Stabilizing Self-Play by Mitigating Iteration-Collapse With One-Class Novelty Rewards

ICML 2026poster

Training large language models via self-play often suffers from a persistent iteration-collapse, where performance initially improves but subsequently regresses as training iterations increase. We analyze this phenomenon as arising from cross-iteration degeneration, where the task-generation distrib…

Cited by 0SourceScholar
2025

Amortized Control of Continuous State Space Feynman-Kac Model for Irregular Time Series

ICLR 2025oral

Many real-world datasets, such as healthcare, climate, and economics, are often collected as irregular time series, which poses challenges for accurate modeling. In this paper, we propose the Amortized Control of continuous State Space Model (ACSSM) for continuous dynamical modeling of time series f…

2025

Compact Memory for Continual Logistic Regression

NeurIPS 2025poster

Despite recent progress, continual learning still does not match the performance of batch training. To avoid catastrophic forgetting, we need to build compact memory of essential past knowledge, but no clear solution has yet emerged, even for shallow neural networks with just one or two layers. In t…

Cited by 0SourceScholar
2025

PANGEA: Projection-Based Augmentation with Non-Relevant General Data for Enhanced Domain Adaptation in LLMs

NeurIPS 2025poster

Modern large language models (LLMs) achieve competitive performance across a wide range of natural language processing tasks through zero-shot or few-shot prompting. However, domain-specific tasks often still require fine-tuning, which is frequently hindered by data scarcity, i.e., collecting suffic…

Cited by 0SourceScholar
2025

Reliable Decision‑Making via Calibration‑Oriented Retrieval‑Augmented Generation

NeurIPS 2025poster

Recently, Large Language Models (LLMs) have been increasingly used to support various decision-making tasks, assisting humans in making informed decisions. However, when LLMs confidently provide incorrect information, it can lead humans to make suboptimal decisions. To prevent LLMs from generating i…

Cited by 0SourceScholar
2025

Test Time Scaling for Neural Processes

NeurIPS 2025poster

Uncertainty-aware meta-learning aims not only for rapid adaptation to new tasks but also for reliable uncertainty estimation under limited supervision. Neural Processes (NPs) offer a flexible solution by learning implicit stochastic processes directly from data, often using a global latent variable…

Cited by 0SourceScholar
2024

Enhancing Transfer Learning with Flexible Nonparametric Posterior Sampling

ICLR 2024poster

Transfer learning has recently shown significant performance across various tasks involving deep neural networks. In these transfer learning scenarios, the prior distribution for downstream data becomes crucial in Bayesian model averaging (BMA). While previous works proposed the prior over the neura…

Cited by 3SourcePDFScholar
2024

Model Fusion through Bayesian Optimization in Language Model Fine-Tuning

NeurIPS 2024spotlight

Fine-tuning pre-trained models for downstream tasks is a widely adopted technique known for its adaptability and reliability across various domains. Despite its conceptual simplicity, fine-tuning entails several troublesome engineering choices, such as selecting hyperparameters and determining check…

2024

Sparse Weight Averaging with Multiple Particles for Iterative Magnitude Pruning

ICLR 2024poster

Given the ever-increasing size of modern neural networks, the significance of sparse architectures has surged due to their accelerated inference speeds and minimal memory demands. When it comes to global pruning techniques, Iterative Magnitude Pruning (IMP) still stands as a state-of-the-art algorit…

Cited by 1SourcePDFScholar
2023

Traversing Between Modes in Function Space for Fast Ensembling

ICML 2023poster

Deep ensemble is a simple yet powerful way to improve the performance of deep neural networks. Under this motivation, recent works on mode connectivity have shown that parameters of ensembles are connected by low-loss subspaces, and one can efficiently collect ensemble parameters in those subspaces.…

2022

Improving Ensemble Distillation With Weight Averaging and Diversifying Perturbation

ICML 2022spotlight

Ensembles of deep neural networks have demonstrated superior performance, but their heavy computational cost hinders applying them for resource-limited environments. It motivates distilling knowledge from the ensemble teacher into a smaller student network, and there are two important design choices…