Young-Joon Lee
Korea Advanced Institute of Science and Technology · 情報科学
研究室紹介
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.
Research Overview
Research Output Trend
Figures are computed from collected data and may differ slightly.
Selected Papers
11Federated 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,
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
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
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
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
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/
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
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/
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.