← Search

Hayeong Lee

4 accepted papers

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
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

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…

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