← Search

Byung-Jun Lee

21 accepted papers

2026

Beyond RAG vs. Long-Context: Learning Distraction-Aware Retrieval for Efficient Knowledge Grounding

ICLR 2026poster

Retrieval-Augmented Generation (RAG) is a framework for grounding Large Language Models (LLMs) in external, up-to-date information. However, recent advancements in context window size allow LLMs to process inputs of up to 128K tokens or more, offering an alternative strategy: supplying the full docu…

Cited by 0SourceScholar
2026

FALCON: False-Negative Aware Learning of Contrastive Negatives in Vision-Language Alignment

CVPR 2026

False negatives pose a critical challenge in vision-language pretraining (VLP) due to the many-to-many correspondence between images and texts in large-scale datasets. These false negatives introduce conflicting supervision signals that degrade the learned embedding space and diminish the effectiven

Cited by 0SourcecodeScholar
2026

SCAD: Super-Class-Aware Debiasing for Long-Tailed Semi-Supervised Learning

ICLR 2026poster

In long-tailed semi-supervised learning (LTSSL), pseudolabeling often creates a vicious cycle of bias amplification, a problem that recent state-of-the-art methods attempt to mitigate using logit adjustment (LA). However, their adjustment schemes, inherited from LA, remain inherently hierarchyagnost…

Cited by 0SourcecodeScholar
2026

TABX: A High-Throughput Sandbox Battle Simulator for Multi-Agent Reinforcement Learning

ICML 2026poster

The design of environments plays a critical role in shaping the development and evaluation of cooperative multi-agent reinforcement learning (MARL) algorithms. While existing benchmarks highlight critical challenges, they often lack the modularity required to design custom evaluation scenarios. We i…

Cited by 0SourceScholar
2026

uCLIP: Parameter-Efficient Multilingual Extension of Vision-Language Models with Unpaired Data

AAAI 2026technical

Contrastive Language–Image Pre-training (CLIP) has demonstrated strong generalization across a wide range of visual tasks by leveraging large-scale English–image pairs. However, its extension to low-resource languages remains limited due to the scarcity of high-quality multilingual image–text data.

Cited by 0SourcePDFScholar
2025

Adaptive Non-Uniform Timestep Sampling for Accelerating Diffusion Model Training

CVPR 2025poster

As a highly expressive generative model, diffusion models have demonstrated exceptional success across various domains, including image generation, natural language processing, and combinatorial optimization. However, as data distributions grow more complex, training these models to convergence beco…

Cited by 0SourcePDFScholar
2025

FairDICE: Fairness-Driven Offline Multi-Objective Reinforcement Learning

NeurIPS 2025poster

Multi-objective reinforcement learning (MORL) aims to optimize policies in the presence of conflicting objectives, where linear scalarization is commonly used to reduce vector-valued returns into scalar signals. While effective for certain preferences, this approach cannot capture fairness-oriented…

Cited by 0SourceScholar
2025

Iterative Prompt Refinement for Safer Text-to-Image Generation

EMNLP 2025

Text-to-Image (T2I) models have made remarkable progress in generating images from text prompts, but their output quality and safety still depend heavily on how prompts are phrased. Existing safety methods typically refine prompts using large language models (LLMs), but they overlook the images prod

2025

K/DA: Automated Data Generation Pipeline for Detoxifying Implicitly Offensive Language in Korean

ACL 2025long

Language detoxification involves removing toxicity from offensive language. While a neutral-toxic paired dataset provides a straightforward approach for training detoxification models, creating such datasets presents several challenges: i) the need for human annotation to build paired data, and ii)…

2025

NBDI: A Simple and Effective Termination Condition for Skill Extraction from Task-Agnostic Demonstrations

ICML 2025poster

Intelligent agents are able to make decisions based on different levels of granularity and duration. Recent advances in skill learning enabled the agent to solve complex, long-horizon tasks by effectively guiding the agent in choosing appropriate skills. However, the practice of using fixed-length s…

2025

Prior-Guided Diffusion Planning for Offline Reinforcement Learning

NeurIPS 2025poster

Diffusion models have recently gained prominence in offline reinforcement learning due to their ability to effectively learn high-performing, generalizable policies from static datasets. Diffusion-based planners facilitate long-horizon decision-making by generating high-quality trajectories through…

Cited by 0SourcecodeScholar
2025

Rethinking DPO: The Role of Rejected Responses in Preference Misalignment

EMNLP 2025

Direct Preference Optimization (DPO) is a simple and efficient framework that has attracted substantial attention. However, it often struggles to meet its primary objectives—increasing the generation probability of chosen responses while reducing that of rejected responses—due to the dominant influe

2025

Test-Time Adaptation for Online Vision-Language Navigation with Feedback-based Reinforcement Learning

ICML 2025poster

Navigating in an unfamiliar environment during deployment poses a critical challenge for a vision-language navigation (VLN) agent. Yet, test-time adaptation (TTA) remains relatively underexplored in robotic navigation, leading us to the fundamental question: what are the key properties of TTA for on…

Cited by 0SourcePDFScholar
2024

Mitigating Covariate Shift in Behavioral Cloning via Robust Stationary Distribution Correction

NeurIPS 2024poster

We consider offline imitation learning (IL), which aims to train an agent to imitate from the dataset of expert demonstrations without online interaction with the environment. Behavioral Cloning (BC) has been a simple yet effective approach to offline IL, but it is also well-known to be vulnerable t…

Cited by 0SourcePDFScholar
2024

ROIDICE: Offline Return on Investment Maximization for Efficient Decision Making

NeurIPS 2024poster

In this paper, we propose a novel policy optimization framework that maximizes Return on Investment (ROI) of a policy using a fixed dataset within a Markov Decision Process (MDP) equipped with a cost function. ROI, defined as the ratio between the return and the accumulated cost of a policy, serves…

Cited by 0SourcePDFScholar
2024

Relaxed Stationary Distribution Correction Estimation for Improved Offline Policy Optimization

AAAI 2024technical

One of the major challenges of offline reinforcement learning (RL) is dealing with distribution shifts that stem from the mismatch between the trained policy and the data collection policy. Stationary distribution correction estimation algorithms (DICE) have addressed this issue by regularizing the…

Cited by 1SourcePDFScholar
2022

Local Metric Learning for Off-Policy Evaluation in Contextual Bandits with Continuous Actions

NeurIPS 2022accept

We consider local kernel metric learning for off-policy evaluation (OPE) of deterministic policies in contextual bandits with continuous action spaces. Our work is motivated by practical scenarios where the target policy needs to be deterministic due to domain requirements, such as prescription of t…

2021

Winning the L2RPN Challenge: Power Grid Management via Semi-Markov Afterstate Actor-Critic

ICLR 2021spotlight

Safe and reliable electricity transmission in power grids is crucial for modern society. It is thus quite natural that there has been a growing interest in the automatic management of power grids, exemplified by the Learning to Run a Power Network Challenge (L2RPN), modeling the problem as a reinforc…

Cited by 70SourcePDFScholar
2020

Reinforcement Learning for Control with Multiple Frequencies

NeurIPS 2020poster

Many real-world sequential decision problems involve multiple action variables whose control frequencies are different, such that actions take their effects at different periods. While these problems can be formulated with the notion of multiple action persistences in factored-action MDP (FA-MDP), i…