Skip to main content

이기민 교수

Kimin Lee

KAIST 김재철AI대학원 · 컴퓨터과학

연구실 소개

이기민 교수의 연구실은 딥 뉴럴 네트워크의 일반화 능력과 신뢰성 향상을 핵심 목표로 삼고 있습니다. 특히, 오염된 데이터나 이상치 샘플에 대해 강인한 예측을 가능하게 하는 추론 기법과, 강화학습에서의 환경 간 일반화 문제를 해결하기 위한 새로운 아키텍처 및 알고리즘 개발에 주력하고 있습니다. 연구는 주로 생성 모델 기반 추론, 불확실성 기반 정책 선택, 그리고 랜덤화 기반 특징 학습 기법을 통해 이루어지며, 실제 응용에 적합한 안정적이고 신뢰할 수 있는 AI 모델을 구현하는 데 초점을 맞춥니다.

생성 모델불확실성 추정강화학습 일반화오염된 데이터추론 기법

연구 현황

논문 수
125
총 인용 수
3,830
최근 5년 논문
57
주요 분야
컴퓨터과학

연구 성과 추이

표시된 성과는 수집된 데이터 기준으로 산출되며, 일부 차이가 있을 수 있습니다.

5개년 연도별 논문 게재 수
57총합
2022
2023
2024
2025
2026
5개년 연도별 피인용 수
222총합
20222023202420252026

주요 논문

15
1
preprint|인용수 976·2018
A Simple Unified Framework for Detecting Out-of-Distribution Samples and Adversarial Attacks
Kimin Lee, Kibok Lee, Honglak Lee, Jinwoo Shin
arXiv (Cornell University)OA

Detecting test samples drawn sufficiently far away from the training distribution statistically or adversarially is a fundamental requirement for deploying a good classifier in many real-world machine learning applications. However, deep neural networks with the softmax classifier are known to produce highly overconfident posterior distributions even for such abnormal samples. In this paper, we propose a simple yet effective method for detecting any abnormal samples, which is applicable to any p

Artificial IntelligenceComputer Science
2
preprint|인용수 365·2017
Training Confidence-calibrated Classifiers for Detecting Out-of-Distribution Samples
Kimin Lee, Honglak Lee, Kibok Lee, Jinwoo Shin
arXiv (Cornell University)OA

The problem of detecting whether a test sample is from in-distribution (i.e.,\ntraining distribution by a classifier) or out-of-distribution sufficiently\ndifferent from it arises in many real-world machine learning applications.\nHowever, the state-of-art deep neural networks are known to be highly\noverconfident in their predictions, i.e., do not distinguish in- and\nout-of-distributions. Recently, to handle this issue, several threshold-based\ndetectors have been proposed given pre-trained ne

Statistics, Probability and UncertaintyDecision Sciences
3
논문|인용수 61·2019
Robust Inference via Generative Classifiers for Handling Noisy Labels
Kimin Lee, Sukmin Yun, Kibok Lee, Honglak Lee, Bo Li, Jinwoo Shin
arXiv (Cornell University)OA

Large-scale datasets may contain significant proportions of noisy (incorrect) class labels, and it is well-known that modern deep neural networks (DNNs) poorly generalize from such noisy training datasets. To mitigate the issue, we propose a novel inference method, termed Robust Generative classifier (RoG), applicable to any discriminative (e.g., softmax) neural classifier pre-trained on noisy datasets. In particular, we induce a generative classifier on top of hidden feature spaces of the pre-t

Artificial IntelligenceComputer Science
4
논문|인용수 48·2020
Network Randomization: A Simple Technique for Generalization in Deep Reinforcement Learning
Kimin Lee, Kibok Lee, Jinwoo Shin, Honglak Lee
International Conference on Learning Representations

Deep reinforcement learning (RL) agents often fail to generalize to unseen environments (yet semantically similar to trained agents), particularly when they are trained on high-dimensional state spaces, such as images. In this paper, we propose a simple technique to improve a generalization ability of deep RL agents by introducing a randomized (convolutional) neural network that randomly perturbs input observations. It enables trained agents to adapt to new domains by learning robust features in

Artificial IntelligenceComputer Science
5
preprint|인용수 47·2019
Network Randomization: A Simple Technique for Generalization in Deep Reinforcement Learning
Kimin Lee, Kibok Lee, Jinwoo Shin, Honglak Lee
ArXiv.orgOA

Deep reinforcement learning (RL) agents often fail to generalize to unseen environments (yet semantically similar to trained agents), particularly when they are trained on high-dimensional state spaces, such as images. In this paper, we propose a simple technique to improve a generalization ability of deep RL agents by introducing a randomized (convolutional) neural network that randomly perturbs input observations. It enables trained agents to adapt to new domains by learning robust features in

Artificial IntelligenceComputer Science
6
preprint|인용수 47·2020
SUNRISE: A Simple Unified Framework for Ensemble Learning in Deep Reinforcement Learning
Kimin Lee, Michael Laskin, Aravind Srinivas, Pieter Abbeel
arXiv (Cornell University)OA

Off-policy deep reinforcement learning (RL) has been successful in a range of challenging domains. However, standard off-policy RL algorithms can suffer from several issues, such as instability in Q-learning and balancing exploration and exploitation. To mitigate these issues, we present SUNRISE, a simple unified ensemble method, which is compatible with various off-policy RL algorithms. SUNRISE integrates two key ingredients: (a) ensemble-based weighted Bellman backups, which re-weight target Q

Artificial IntelligenceComputer Science
7
preprint|인용수 42·2018
A Simple Unified Framework for Detecting Out-of-Distribution Samples and Adversarial Attacks
Kimin Lee, Kibok Lee, Honglak Lee, Jinwoo Shin
arXiv (Cornell University)OA

Detecting test samples drawn sufficiently far away from the training distribution statistically or adversarially is a fundamental requirement for deploying a good classifier in many real-world machine learning applications. However, deep neural networks with the softmax classifier are known to produce highly overconfident posterior distributions even for such abnormal samples. In this paper, we propose a simple yet effective method for detecting any abnormal samples, which is applicable to any p

Artificial IntelligenceComputer Science
8
preprint|인용수 35·2023
Aligning Text-to-Image Models using Human Feedback
Kimin Lee, Hao Liu, Moonkyung Ryu, Olivia Watkins, Yuqing Du, Craig Boutilier, Pieter Abbeel, Mohammad Ghavamzadeh, Shixiang Gu
arXiv (Cornell University)OA

Deep generative models have shown impressive results in text-to-image synthesis. However, current text-to-image models often generate images that are inadequately aligned with text prompts. We propose a fine-tuning method for aligning such models using human feedback, comprising three stages. First, we collect human feedback assessing model output alignment from a set of diverse text prompts. We then use the human-labeled image-text dataset to train a reward function that predicts human feedback

Computer Vision and Pattern RecognitionComputer Science
9
preprint|인용수 28·2020
Context-aware Dynamics Model for Generalization in Model-Based Reinforcement Learning
Kimin Lee, Younggyo Seo, Seunghyun Lee, Honglak Lee, Jinwoo Shin
arXiv (Cornell University)OA

Model-based reinforcement learning (RL) enjoys several benefits, such as\ndata-efficiency and planning, by learning a model of the environment's\ndynamics. However, learning a global model that can generalize across different\ndynamics is a challenging task. To tackle this problem, we decompose the task\nof learning a global dynamics model into two stages: (a) learning a context\nlatent vector that captures the local dynamics, then (b) predicting the next\nstate conditioned on it. In order to en

Artificial IntelligenceComputer Science
10
논문|인용수 20·2017
Confident multiple choice learning
Kimin Lee, Chang Ho Hwang, Kyong Soo Park, Jinwoo Shin
International Conference on Machine Learning

Ensemble methods are arguably the most trustworthy techniques for boosting the performance of machine learning models. Popular independent ensembles (IE) relying on naive averaging/voting scheme have been of typical choice for most applications involving deep neural networks, but they do not consider advanced collaboration among ensemble models. In this paper, we propose new ensemble methods specialized for deep neural networks, called confident multiple choice learning (CMCL): it is a variant o

Artificial IntelligenceComputer Science
11
preprint|인용수 18·2017
Confident Multiple Choice Learning
Kimin Lee, Chang Ho Hwang, KyoungSoo Park, Jinwoo Shin
arXiv (Cornell University)OA

Ensemble methods are arguably the most trustworthy techniques for boosting the performance of machine learning models. Popular independent ensembles (IE) relying on naive averaging/voting scheme have been of typical choice for most applications involving deep neural networks, but they do not consider advanced collaboration among ensemble models. In this paper, we propose new ensemble methods specialized for deep neural networks, called confident multiple choice learning (CMCL): it is a variant o

PharmacologyPharmacology, Toxicology and Pharmaceutics
12
preprint|인용수 13·2021
PEBBLE: Feedback-Efficient Interactive Reinforcement Learning via Relabeling Experience and Unsupervised Pre-training
Kimin Lee, Laura Smith, Pieter Abbeel
arXiv (Cornell University)OA

Conveying complex objectives to reinforcement learning (RL) agents can often be difficult, involving meticulous design of reward functions that are sufficiently informative yet easy enough to provide. Human-in-the-loop RL methods allow practitioners to instead interactively teach agents through tailored feedback; however, such approaches have been challenging to scale since human feedback is very expensive. In this work, we aim to make this process more sample- and feedback-efficient. We present

Artificial IntelligenceComputer Science
13
논문|인용수 7·2017
Making Stochastic Neural Networks from Deterministic Ones
Kimin Lee, Jaehyung Kim, Song Chong, Jinwoo Shin

It has been believed that stochastic feedforward neural networks (SFNN) have several advantages beyond deterministic deep neural networks (DNN): they have more expressive power allowing multi-modal mappings and regularize better due to their stochastic nature. However, training SFNN is notoriously harder. In this paper, we aim at developing efficient training methods for large-scale SFNN, in particular using known architectures and pre-trained parameters of DNN. To this end, we propose a new int

Computer Vision and Pattern RecognitionComputer Science
14
논문|인용수 6·2016
Just-in-time WLANs: On-demand interference-managed WLAN infrastructures
Kimin Lee, Yeonkeun Kim, Seokhyun Kim, Jinwoo Shin, Seungwon Shin, Song Chong

In the past years, the centralized management of dense wireless local area networks has been emerged as a powerful paradigm for improving energy efficiency as well as avoiding severe interference. In this paper, we study the joint optimization on power-operation modes in access points (APs), channel selections and user-AP associations for improving energy efficiency and avoiding interference without sacrificing users' demands. To this end, we first formulate it as a mixed-integer programming usi

Computer Networks and CommunicationsComputer Science
15
preprint|인용수 4·2017
Simplified Stochastic Feedforward Neural Networks
Kimin Lee, Jaehyung Kim, Song Chong, Jinwoo Shin
arXiv (Cornell University)OA

It has been believed that stochastic feedforward neural networks (SFNNs) have several advantages beyond deterministic deep neural networks (DNNs): they have more expressive power allowing multi-modal mappings and regularize better due to their stochastic nature. However, training large-scale SFNN is notoriously harder. In this paper, we aim at developing efficient training methods for SFNN, in particular using known architectures and pre-trained parameters of DNN. To this end, we propose a new i

Computer Vision and Pattern RecognitionComputer Science

대표 연구 분야

Artificial IntelligenceComputer Vision and Pattern RecognitionElectrical and Electronic EngineeringComputer Networks and CommunicationsControl and Systems EngineeringSignal Processing

이기민 교수의 연구를 Nubint에서 더 깊이 살펴보세요

이 연구실의 논문을 앱에서 열어 AI와 함께 읽고, 핵심을 요약하고, 내 글에 인용하세요.