Lee, Byung-Jun

17 publications

ICLR 2026 Beyond RAG vs. Long-Context: Learning Distraction-Aware Retrieval for Efficient Knowledge Grounding Seong-Woong Shim, Myunsoo Kim, Jae Hyeon Cho, Byung-Jun Lee

ICLR 2026 SCAD: Super-Class-Aware Debiasing for Long-Tailed Semi-Supervised Learning Sunguk Jang, Jinwoo Jeon, Byung-Jun Lee

CVPR 2025 Adaptive Non-Uniform Timestep Sampling for Accelerating Diffusion Model Training Myunsoo Kim, Donghyeon Ki, Seong-Woong Shim, Byung-Jun Lee

NeurIPS 2025 FairDICE: Fairness-Driven Offline Multi-Objective Reinforcement Learning Woosung Kim, Jinho Lee, Jongmin Lee, Byung-Jun Lee

ICML 2025 NBDI: A Simple and Effective Termination Condition for Skill Extraction from Task-Agnostic Demonstrations Myunsoo Kim, Hayeong Lee, Seong-Woong Shim, Junho Seo, Byung-Jun Lee

NeurIPS 2025 Prior-Guided Diffusion Planning for Offline Reinforcement Learning Donghyeon Ki, JunHyeok Oh, Seong-Woong Shim, Byung-Jun Lee

ICML 2025 Test-Time Adaptation for Online Vision-Language Navigation with Feedback-Based Reinforcement Learning Sungjune Kim, Gyeongrok Oh, Heeju Ko, Daehyun Ji, Dongwook Lee, Byung-Jun Lee, Sujin Jang, Sangpil Kim

CoLLAs 2024 ARCLE: The Abstraction and Reasoning Corpus Learning Environment for Reinforcement Learning Hosung Lee, Sejin Kim, Seungpil Lee, Sanha Hwang, Jihwan Lee, Byung-Jun Lee, Sundong Kim

NeurIPS 2024 Mitigating Covariate Shift in Behavioral Cloning via Robust Stationary Distribution Correction Seokin Seo, Byung-Jun Lee, Jongmin Lee, HyeongJoo Hwang, Hongseok Yang, Kee-Eung Kim

NeurIPS 2024 ROIDICE: Offline Return on Investment Maximization for Efficient Decision Making Woosung Kim, Hayeong Lee, Jongmin Lee, Byung-Jun Lee

AAAI 2024 Relaxed Stationary Distribution Correction Estimation for Improved Offline Policy Optimization Woosung Kim, Donghyeon Ki, Byung-Jun Lee

NeurIPS 2022 Local Metric Learning for Off-Policy Evaluation in Contextual Bandits with Continuous Actions Haanvid Lee, Jongmin Lee, Yunseon Choi, Wonseok Jeon, Byung-Jun Lee, Yung-Kyun Noh, Kee-Eung Kim

ICLR 2021 Representation Balancing Offline Model-Based Reinforcement Learning Byung-Jun Lee, Jongmin Lee, Kee-Eung Kim

ICLR 2021 Winning the L2RPN Challenge: Power Grid Management via Semi-Markov Afterstate Actor-Critic Deunsol Yoon, Sunghoon Hong, Byung-Jun Lee, Kee-Eung Kim

NeurIPS 2020 Reinforcement Learning for Control with Multiple Frequencies Jongmin Lee, Byung-Jun Lee, Kee-Eung Kim

AAAI 2020 Residual Neural Processes Byung-Jun Lee, Seunghoon Hong, Kee-Eung Kim

AISTATS 2017 Hierarchically-Partitioned Gaussian Process Approximation Byung-Jun Lee, Jongmin Lee, Kee-Eung Kim