← Search

Jaeho Lee

30 accepted papers

2026

Over-Alignment vs Over-Fitting: The Role of Feature Learning Strength in Generalization

ICML 2026poster

Feature learning strength (FLS), i.e., the inverse of the effective output scaling of a model, plays a critical role in shaping the optimization dynamics of neural nets. While its impact has been extensively studied under the asymptotic regimes---both in training time and FLS---existing theory offer…

Cited by 0SourceScholar
2025

Imagine to Hear: Auditory Knowledge Generation can be an Effective Assistant for Language Models

ACL 2025finding

Language models pretrained on text-only corpora often struggle with tasks that require auditory commonsense knowledge.Previous work addresses this problem by augmenting the language model to retrieve knowledge from external audio databases.This approach has several limitations, such as the potential…

Cited by 0SourcePDFScholar
2025

Riemannian Optimization for LoRA on the Stiefel Manifold

EMNLP 2025

While powerful, large language models (LLMs) present significant fine-tuning challenges due to their size. Parameter-efficient fine-tuning (PEFT) methods like LoRA provide solutions, yet suffer from critical optimizer inefficiencies; notably basis redundancy in LoRA’s B matrix when using AdamW, whic

Cited by 0SourcePDFScholar
2025

ZIP: An Efficient Zeroth-order Prompt Tuning for Black-box Vision-Language Models

ICLR 2025poster

Recent studies have introduced various approaches for prompt-tuning black-box vision-language models, referred to as black-box prompt-tuning (BBPT). While BBPT has demonstrated considerable potential, it is often found that many existing methods require an excessive number of queries (i.e., function…

Cited by 0SourcePDFScholar
2024

Discovering and Mitigating Visual Biases through Keyword Explanation

CVPR 2024highlight

Addressing biases in computer vision models is crucial for real-world AI deployments. However mitigating visual biases is challenging due to their unexplainable nature often identified indirectly through visualization or sample statistics which necessitates additional human supervision for interpret…

2024

In Search of a Data Transformation That Accelerates Neural Field Training

CVPR 2024poster

Neural field is an emerging paradigm in data representation that trains a neural network to approximate the given signal. A key obstacle that prevents its widespread adoption is the encoding speed---generating neural fields requires an overfitting of a neural network which can take a significant num…

2024

Neural Image Compression with Text-guided Encoding for both Pixel-level and Perceptual Fidelity

ICML 2024poster

Recent advances in text-guided image compression have shown great potential to enhance the perceptual quality of reconstructed images. These methods, however, tend to have significantly degraded pixel-wise fidelity, limiting their practicality. To fill this gap, we develop a new text-guided image co…

2024

Prefixing Attention Sinks can Mitigate Activation Outliers for Large Language Model Quantization

EMNLP 2024main

Despite recent advances in LLM quantization, activation quantization remains to be challenging due to the activation outliers. Conventional remedies, e.g., mixing precisions for different channels, introduce extra overhead and reduce the speedup. In this work, we develop a simple yet effective strat…

Cited by 7SourcePDFScholar
2024

Rethinking Pruning Large Language Models: Benefits and Pitfalls of Reconstruction Error Minimization

EMNLP 2024main

This work suggests fundamentally rethinking the current practice of pruning large language models (LLMs). The way it is done is by divide and conquer: split the model into submodels, sequentially prune them, and reconstruct predictions of the dense counterparts on small calibration data one at a tim…

2024

SCANNER: Knowledge-Enhanced Approach for Robust Multi-modal Named Entity Recognition of Unseen Entities

NAACL 2024long

Recent advances in named entity recognition (NER) have pushed the boundary of the task to incorporate visual signals, leading to many variants, including multi-modal NER (MNER) or grounded MNER (GMNER). A key challenge to these tasks is that the model should be able to generalize to the entities uns…

Cited by 3SourcePDFScholar
2023

Learning Large-scale Neural Fields via Context Pruned Meta-Learning

NeurIPS 2023poster

We introduce an efficient optimization-based meta-learning technique for large-scale neural field training by realizing significant memory savings through automated online context point selection. This is achieved by focusing each learning step on the subset of data with the highest expected immedia…

2023

Modality-Agnostic Variational Compression of Implicit Neural Representations

ICML 2023poster

We introduce a modality-agnostic neural compression algorithm based on a functional view of data and parameterised as an Implicit Neural Representation (INR). Bridging the gap between latent coding and sparsity, we obtain compact latent representations non-linearly mapped to a soft gating mechanism.…

Cited by 27SourcePDFScholar
2022

Meta-Learning with Self-Improving Momentum Target

NeurIPS 2022accept

The idea of using a separately trained target model (or teacher) to improve the performance of the student model has been increasingly popular in various machine learning domains, and meta-learning is no exception; a recent discovery shows that utilizing task-wise target models can significantly boo…

2022

Scalable Neural Video Representations with Learnable Positional Features

NeurIPS 2022accept

Succinct representation of complex signals using coordinate-based neural representations (CNRs) has seen great progress, and several recent efforts focus on extending them for handling videos. Here, the main challenge is how to (a) alleviate a compute-inefficiency in training CNRs to (b) achieve hig…

2022

Spread Spurious Attribute: Improving Worst-group Accuracy with Spurious Attribute Estimation

ICLR 2022poster

The paradigm of worst-group loss minimization has shown its promise in avoiding to learn spurious correlations, but requires costly additional supervision on spurious attributes. To resolve this, recent works focus on developing weaker forms of supervision---e.g., hyperparameters discovered with a s…

Cited by 105SourcePDFScholar
2021

An integrated approach for determining objects to be relocated and their goal positions inside clutter for object retrieval

ICRA 2021poster

We consider the problem of rearranging objects in a cluttered and confined space using a robotic manipulator. The goal is to retrieve a target object from the clutter where the target is occluded by other objects. In situations where overhand grasps are not allowed, the robot needs to remove some ob…

Cited by 16SourceScholar
2021

Layer-adaptive Sparsity for the Magnitude-based Pruning

ICLR 2021poster

Recent discoveries on neural network pruning reveal that, with a carefully chosen layerwise sparsity, a simple magnitude-based pruning achieves state-of-the-art tradeoff between sparsity and performance. However, without a clear consensus on ``how to choose,'' the layerwise sparsities are mostly sel…

2021

MASKER: Masked Keyword Regularization for Reliable Text Classification

AAAI 2021technical

Pre-trained language models have achieved state-of-the-art accuracies on various text classification tasks, e.g., sentiment analysis, natural language inference, and semantic textual similarity. However, the reliability of the fine-tuned text classifiers is an often underlooked performance criterion…

2020

Learning from Failure: De-biasing Classifier from Biased Classifier

NeurIPS 2020poster

Neural networks often learn to make predictions that overly rely on spurious corre- lation existing in the dataset, which causes the model to be biased. While previous work tackles this issue by using explicit labeling on the spuriously correlated attributes or presuming a particular bias type, we i…