← Search

Soochan Lee

7 accepted papers

2025

Distilling Reinforcement Learning Algorithms for In-Context Model-Based Planning

ICLR 2025poster

Recent studies have shown that Transformers can perform in-context reinforcement learning (RL) by imitating existing RL algorithms, enabling sample-efficient adaptation to unseen tasks without parameter updates. However, these models also inherit the suboptimal behaviors of the RL algorithms they im…

2024

Learning to Continually Learn with the Bayesian Principle

ICML 2024poster

In the present era of deep learning, continual learning research is mainly focused on mitigating forgetting when training a neural network with stochastic gradient descent on a non-stationary stream of data. On the other hand, in the more classical literature of statistical machine learning, many mo…

2023

Recursion of Thought: A Divide-and-Conquer Approach to Multi-Context Reasoning with Language Models

ACL 2023findings

Generating intermediate steps, or Chain of Thought (CoT), is an effective way to significantly improve language models’ (LM) multi-step reasoning capability. However, the CoT lengths can grow rapidly with the problem complexity, easily exceeding the maximum context size. Instead of increasing the co…

2020

A Neural Dirichlet Process Mixture Model for Task-Free Continual Learning

ICLR 2020poster

Despite the growing interest in continual learning, most of its contemporary works have been studied in a rather restricted setting where tasks are clearly distinguishable, and task boundaries are known during training. However, if our goal is to develop an algorithm that learns as humans do, this s…

Cited by 280SourcecodeScholar
2019

Harmonizing Maximum Likelihood with GANs for Multimodal Conditional Generation

ICLR 2019poster

Recent advances in conditional image generation tasks, such as image-to-image translation and image inpainting, are largely accounted to the success of conditional GAN models, which are often optimized by the joint use of the GAN loss with the reconstruction loss. However, we reveal that this traini…

Cited by 32SourcePDFScholar
2018

Improving Occlusion and Hard Negative Handling for Single-Stage Pedestrian Detectors

CVPR 2018poster

We propose methods of addressing two critical issues of pedestrian detection: (i) occlusion of target objects as false negative failure, and (ii) confusion with hard negative examples like vertical structures as false positive failure. Our solutions to these two problems are general and flexible eno…

Cited by 110SourcePDFScholar