Skip to main content

Young-Joon Lee

Korea Advanced Institute of Science and Technology · Computer Science

About the Lab

Professor Young-Joon Lee's research lab specializes in federated learning and hyperscale artificial intelligence, with a strong focus on privacy-preserving machine learning in critical domains such as healthcare and national defense. The lab explores advanced deep learning architectures—like Kolmogorov-Arnold Networks (KAN)—and efficient training strategies to enhance model performance under non-IID data and resource-constrained conditions. Key research directions include data-free hyperparameter tuning, early stopping mechanisms, and robustness against client-side configuration variations in decentralized learning systems.

federated learningKolmogorov-Arnold Networksprivacy-preserving AIhyperscale AIdata-free training

Research Overview

Papers
11
Total Citations
26
Papers (5y)
11
Primary Field
Computer Science

Research Output Trend

Figures are computed from collected data and may differ slightly.

Publications per year (5y)
11total
2022
2025
2026
Citations per year (5y)
26total
202220252026

Selected Papers

11
1
Article|17 citations·2022
Accelerated Federated Learning via Greedy Aggregation
Youngjoon Lee, Sangwoo Park, Jin-Hyun Ahn, Joonhyuk Kang
SJR Q1IEEE Communications Letters

Federated learning is a distributed computing framework aiming at finding a shared model parameter while protecting the privacy of local agents by sharing only locally updated model parameters without sharing local data with a central server. Through an iterative procedure between agent-side local updates and central server-side aggregation, federated learning reaches its maximum performance after sufficient iterations which is the the possible best performance via central learning. In practice,

Artificial IntelligenceComputer Science
2
Book Chapter|6 citations·2025
Revisit the Stability of Vanilla Federated Learning Under Diverse Conditions
Youngjoon Lee, Jian Gong, Sun Choi, Joonhyuk Kang
SJR Q2Lecture notes in computer science
Artificial IntelligenceComputer Science
3
Article|1 citations·2025
ROK Defense M&S in the Age of Hyperscale AI: Concepts, Challenges, and Future Directions
Youngjoon Lee, Taehyun Park, Yeongjoon Kang, Jonghoe Kim, Joonhyuk Kang
SJR Q1IEEE Internet of Things Magazine

Integrating hyperscale AI into national defense M&S (Modeling and Simulation), under the expanding IoMDT (Internet of Military Defense Things) framework, is crucial for boosting strategic and operational readiness. We examine how IoMDT-driven hyperscale AI can provide high accuracy, speed, and the ability to simulate complex, interconnected battlefield scenarios in defense M&S. Countries like the United States and China are leading the adoption of these technologies, with varying levels of succe

InstrumentationPhysics and Astronomy
4
Book Chapter|1 citations·2026
Debunking Optimization Myths in Federated Learning for Medical Image Classification
Youngjoon Lee, Hyukjoon Lee, Jinu Gong, Yang Cao, Joonhyuk Kang
SJR Q2Lecture notes in computer science
Artificial IntelligenceComputer Science
5
Preprint|1 citations·2025
A Unified Benchmark of Federated Learning with Kolmogorov-Arnold Networks for Medical Imaging
Youngjoon Lee, Jian Gong, Joonhyuk Kang
ArXiv.orgOA

Federated Learning (FL) enables model training across decentralized devices without sharing raw data, thereby preserving privacy in sensitive domains like healthcare. In this paper, we evaluate Kolmogorov-Arnold Networks (KAN) architectures against traditional MLP across six state-of-the-art FL algorithms on a blood cell classification dataset. Notably, our experiments demonstrate that KAN can effectively replace MLP in federated environments, achieving superior performance with simpler architec

Artificial IntelligenceComputer Science
6
Preprint|0 citations·2025
Debunking Optimization Myths in Federated Learning for Medical Image Classification
Youngjoon Lee, Hyukjoon Lee, Jian Gong, Yang Cao, Joonhyuk Kang
ArXiv.orgOA

Federated Learning (FL) is a collaborative learning method that enables decentralized model training while preserving data privacy. Despite its promise in medical imaging, recent FL methods are often sensitive to local factors such as optimizers and learning rates, limiting their robustness in practical deployments. In this work, we revisit vanilla FL to clarify the impact of edge device configurations, benchmarking recent FL methods on colorectal pathology and blood cell classification task. We

Artificial IntelligenceComputer Science
7
Preprint|0 citations·2025
Revisit the Stability of Vanilla Federated Learning Under Diverse Conditions
Youngjoon Lee, Jian Gong, Sun Choi, Joonhyuk Kang
ArXiv.orgOA

Federated Learning (FL) is a distributed machine learning paradigm enabling collaborative model training across decentralized clients while preserving data privacy. In this paper, we revisit the stability of the vanilla FedAvg algorithm under diverse conditions. Despite its conceptual simplicity, FedAvg exhibits remarkably stable performance compared to more advanced FL techniques. Our experiments assess the performance of various FL methods on blood cell and skin lesion classification tasks usi

Artificial IntelligenceComputer Science
8
Preprint|0 citations·2026
Beyond Fixed Rounds: Data-Free Early Stopping for Practical Federated Learning
Youngjoon Lee, Hyukjoon Lee, Seungrok Jung, Andy Luo, Jinu Gong, Yang Cao, Joonhyuk Kang
SJR Q1Open MINDOA

Federated Learning (FL) facilitates decentralized collaborative learning without transmitting raw data. However, reliance on fixed global rounds or validation data for hyperparameter tuning hinders practical deployment by incurring high computational costs and privacy risks. To address this, we propose a data-free early stopping framework that determines the optimal stopping point by monitoring the task vector's growth rate using only server-side parameters. The numerical results on skin lesion/

Artificial IntelligenceComputer Science
9
Article|0 citations·2026
Forecasting-based biomedical time-series data synthesis for open data and robust AI
Youngjoon Lee, S. Cho, Yehhyun Jo, Jinu Gong, Hyunjoo J. Lee, Joonhyuk Kang
SJR Q1Computers in Biology and MedicineOA

The limited data availability due to strict privacy regulations and significant resource demands severely constrains biomedical time-series AI development, which creates a critical gap between data requirements and accessibility. Synthetic data generation presents a promising solution by producing artificial datasets that maintain the statistical properties of real biomedical time-series data without compromising patient confidentiality. While GANs, VAEs, and diffusion models capture global data

Signal ProcessingComputer Science
10
Article|0 citations·2026
Beyond Fixed Rounds: Data-Free Early Stopping for Practical Federated Learning
Youngjoon Lee, Hyukjoon Lee, Seungrok Jung, Andy Luo, Jinu Gong, Yang Cao, Joonhyuk Kang
arXiv (Cornell University)OA

Federated Learning (FL) facilitates decentralized collaborative learning without transmitting raw data. However, reliance on fixed global rounds or validation data for hyperparameter tuning hinders practical deployment by incurring high computational costs and privacy risks. To address this, we propose a data-free early stopping framework that determines the optimal stopping point by monitoring the task vector's growth rate using only server-side parameters. The numerical results on skin lesion/

Artificial IntelligenceComputer Science
11
Article|0 citations·2025
Deceptive Synthetic Updates: Stealth Free-Rider Attack on Model Aggregation in Federated Learning
Youngjoon Lee, Jinu Gong, Joonhyuk Kang
OA

Federated Learning (FL) allows multiple clients to collaboratively train shared models without exchanging raw data, thereby preserving privacy. However, FL systems are vulnerable to malicious participants known as free-riders who exploit the collaborative nature without providing genuine data contributions. To expose this critical security threat, we introduce a novel stealth free-rider attack that leverages pre-trained forecasting models to generate highly realistic synthetic time-series data.

Artificial IntelligenceComputer Science

Research Areas

Artificial IntelligenceInstrumentationSignal Processing

Dive deeper into Young-Joon Lee's research on Nubint

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