Skip to main content

Hwanjun Song

Korea Advanced Institute of Science and Technology · Computer Science

About the Lab

Professor Hwanjun Song's research lab specializes in scalable and robust machine learning, with a focus on addressing real-world challenges in data quality, efficiency, and system design. The lab explores robust training techniques for deep learning under noisy or imbalanced labels, develops efficient parallel algorithms for clustering and density-based clustering (e.g., DBSCAN and k-medoids), and advances transformer-based architectures for computer vision tasks. Their work bridges theoretical insights with practical systems, emphasizing high performance, accuracy, and scalability in big data environments.

robust learningparallel clusteringtransformer-based visionnoisy label learningdata augmentation

Research Overview

Papers
99
Total Citations
2,087
Papers (5y)
72
Primary Field
Computer Science

Research Output Trend

Figures are computed from collected data and may differ slightly.

Publications per year (5y)
72total
2022
2023
2024
2025
2026
Citations per year (5y)
1,473total
20222023202420252026

Selected Papers

15
1
Article|1,079 citations·2022
Learning From Noisy Labels With Deep Neural Networks: A Survey
Hwanjun Song, Minseok Kim, Dongmin Park, Yooju Shin, Jae-Gil Lee
SJR Q1IEEE Transactions on Neural Networks and Learning Systems

Deep learning has achieved remarkable success in numerous domains with help from large amounts of big data. However, the quality of data labels is a concern because of the lack of high-quality labels in many real-world scenarios. As noisy labels severely degrade the generalization performance of deep neural networks, learning from noisy labels (robust training) is becoming an important task in modern deep learning applications. In this survey, we first describe the problem of learning with label

Artificial IntelligenceComputer Science
2
Article|196 citations·2019
SELFIE: Refurbishing Unclean Samples for Robust Deep Learning
Hwanjun Song, Minseok Kim, Jae-Gil Lee
International Conference on Machine Learning
Artificial IntelligenceComputer Science
3
Preprint|91 citations·2020
Learning from Noisy Labels with Deep Neural Networks: A Survey
Hwanjun Song, Minseok Kim, Dongmin Park, Yooju Shin, Jae-Gil Lee
arXiv (Cornell University)OA

Deep learning has achieved remarkable success in numerous domains with help from large amounts of big data. However, the quality of data labels is a concern because of the lack of high-quality labels in many real-world scenarios. As noisy labels severely degrade the generalization performance of deep neural networks, learning from noisy labels (robust training) is becoming an important task in modern deep learning applications. In this survey, we first describe the problem of learning with label

Artificial IntelligenceComputer Science
4
Article|59 citations·2018
RP-DBSCAN
Hwanjun Song, Jae-Gil Lee

In most parallel DBSCAN algorithms, neighboring points are assigned to the same data partition for parallel processing to facilitate calculation of the density of the neighbors. This data partitioning scheme causes a few critical problems including load imbalance between data partitions, especially in a skewed data set. To remedy these problems, we propose a cell-based data partitioning scheme, pseudo random partitioning , that randomly distributes small cells rather than the points themselves.

Artificial IntelligenceComputer Science
5
Preprint|46 citations·2021
ViDT: An Efficient and Effective Fully Transformer-based Object Detector
Hwanjun Song, Sun Deqing, Sanghyuk Chun, Varun Jampani, Dongyoon Han, Byeongho Heo, Wonjae Kim, Ming–Hsuan Yang
arXiv (Cornell University)OA

Transformers are transforming the landscape of computer vision, especially for recognition tasks. Detection transformers are the first fully end-to-end learning systems for object detection, while vision transformers are the first fully transformer-based architecture for image classification. In this paper, we integrate Vision and Detection Transformers (ViDT) to build an effective and efficient object detector. ViDT introduces a reconfigured attention module to extend the recent Swin Transforme

Computer Vision and Pattern RecognitionComputer Science
6
Article|41 citations·2017
PAMAE
Hwanjun Song, Jae-Gil Lee, Wook-Shin Han

The k-medoids algorithm is one of the best-known clustering algorithms. Despite this, however, it is not as widely used for big data analytics as the k-means algorithm, mainly because of its high computational complexity. Many studies have attempted to solve the efficiency problem of the k-medoids algorithm, but all such studies have improved efficiency at the expense of accuracy. In this paper, we propose a novel parallel k-medoids algorithm, which we call PAMAE, that achieves both high accurac

Signal ProcessingComputer Science
7
Article|25 citations·2019
Prestopping: How Does Early Stopping Help Generalization Against Label Noise?
Hwanjun Song, Minseok Kim, Dongmin Park, Jae-Gil Lee
arXiv (Cornell University)OA
Signal ProcessingComputer Science
8
Article|22 citations·2020
Ada-boundary: accelerating DNN training via adaptive boundary batch selection
Hwanjun Song, Sundong Kim, Minseok Kim, Jae-Gil Lee
SJR Q1Machine LearningOA
Computer Vision and Pattern RecognitionComputer Science
9
Article|20 citations·2024
Toward Robustness in Multi-Label Classification: A Data Augmentation Strategy against Imbalance and Noise
Hwanjun Song, Minseok Kim, Jae-Gil Lee
Proceedings of the AAAI Conference on Artificial IntelligenceOA

Multi-label classification poses challenges due to imbalanced and noisy labels in training data. In this paper, we propose a unified data augmentation method, named BalanceMix, to address these challenges. Our approach includes two samplers for imbalanced labels, generating minority-augmented instances with high diversity. It also refines multi-labels at the label-wise granularity, categorizing noisy labels as clean, re-labeled, or ambiguous for robust optimization. Extensive experiments on thre

Artificial IntelligenceComputer Science
10
Article|20 citations·2024
FineSurE: Fine-grained Summarization Evaluation using LLMs
Hwanjun Song, Hang Su, Igor Shalyminov, Jason Cai, Saab Mansour
OA

Automated evaluation is crucial for streamlining text summarization benchmarking and model development, given the costly and timeconsuming nature of human evaluation.Traditional methods like ROUGE do not correlate well with human judgment, while recently proposed LLM-based metrics provide only summary-level assessment using Likertscale scores.This limits deeper model analysis, e.g., we can only assign one hallucination score at the summary level, while at the sentence level, we can count sentenc

Artificial IntelligenceComputer Science
11
Preprint|19 citations·2019
How does Early Stopping Help Generalization against Label Noise?
Hwanjun Song, Minseok Kim, Dongmin Park, Jae-Gil Lee
arXiv (Cornell University)OA

Noisy labels are very common in real-world training data, which lead to poor generalization on test data because of overfitting to the noisy labels. In this paper, we claim that such overfitting can be avoided by "early stopping" training a deep neural network before the noisy labels are severely memorized. Then, we resume training the early stopped network using a "maximal safe set," which maintains a collection of almost certainly true-labeled samples at each epoch since the early stop point.

Artificial IntelligenceComputer Science
12
Article|18 citations·2024
Prompt-guided DETR with RoI-pruned masked attention for open-vocabulary object detection
Hwanjun Song, Jihwan Bang
SJR Q1Pattern Recognition
Computer Vision and Pattern RecognitionComputer Science
13
Preprint|11 citations·2022
An Extendable, Efficient and Effective Transformer-based Object Detector
Hwanjun Song, Deqing Sun, Sanghyuk Chun, Varun Jampani, Dongyoon Han, Byeongho Heo, Wonjae Kim, Ming–Hsuan Yang
arXiv (Cornell University)OA

Transformers have been widely used in numerous vision problems especially for visual recognition and detection. Detection transformers are the first fully end-to-end learning systems for object detection, while vision transformers are the first fully transformer-based architecture for image classification. In this paper, we integrate Vision and Detection Transformers (ViDT) to construct an effective and efficient object detector. ViDT introduces a reconfigured attention module to extend the rece

Computer Vision and Pattern RecognitionComputer Science
14
Article|8 citations·2020
Carpe Diem, Seize the Samples Uncertain "at the Moment" for Adaptive Batch Selection
Hwanjun Song, Minseok Kim, Sundong Kim, Jae-Gil Lee

The accuracy of deep neural networks is significantly affected by how well mini-batches are constructed during the training step. In this paper, we propose a novel adaptive batch selection algorithm called Recency Bias that exploits the uncertain samples predicted inconsistently in recent iterations. The historical label predictions of each training sample are used to evaluate its predictive uncertainty within a sliding window. Then, the sampling probability for the next mini-batch is assigned t

Artificial IntelligenceComputer Science
15
Preprint|3 citations·2023
Prompt-Guided Transformers for End-to-End Open-Vocabulary Object Detection
Hwanjun Song, Jihwan Bang
arXiv (Cornell University)OA

Prompt-OVD is an efficient and effective framework for open-vocabulary object detection that utilizes class embeddings from CLIP as prompts, guiding the Transformer decoder to detect objects in both base and novel classes. Additionally, our novel RoI-based masked attention and RoI pruning techniques help leverage the zero-shot classification ability of the Vision Transformer-based CLIP, resulting in improved detection performance at minimal computational cost. Our experiments on the OV-COCO and

Computer Vision and Pattern RecognitionComputer Science

Research Areas

Artificial IntelligenceComputer Vision and Pattern RecognitionInformation SystemsSignal ProcessingManagement Science and Operations ResearchEconomics and Econometrics

Dive deeper into Hwanjun Song's research on Nubint

Open this lab's papers in the app to read with AI, summarize, and cite in your writing.