activity
20232026
most citedFedAIoT: A Federated Learning Benchmark for Artificial Intelligence of Things

7 citations · 10 across the 16 of their papers we have counts for

collaborators
Showing cs.LGShow all

10 papers · 1 filter

cs.LG2026

How Linear Attention Remembers

Kichang Lee, JaeYeon Park, Songkuk Kim +1

Linear attention replaces the growing key--value (KV) cache of standard attention with a fixed-size recurrent state, substantially reducing memory growth with context length. This…

cs.LG2026

Metacognitive Selective Ensemble for Mobile Systems

Sungmin Lee, Kichang Lee, Joonhee Lee +3

Deep ensembles improve robustness in mobile sensing, but repeatedly executing many models over continuous sensor streams is costly. Selecting only a few members reduces this cost,…

cs.LG2026

Pushing the (Decision) Boundaries: Dynamically Calibrating Differentially Private Noise to Explainability in Federated Learning

Michael Khavkin, Kichang Lee, Jaeho Jin +2

Federated Learning (FL) with Differential Privacy (DP) is increasingly adopted to preserve data confidentiality in distributed machine learning. However, DP noise distorts learned…

cs.LG2026

Temperature Scaling Attack Disrupting Model Confidence in Federated Learning

Kichang Lee, Jaeho Jin, JaeYeon Park +2

Predictive confidence serves as a foundational control signal in mission-critical systems, directly governing risk-aware logic such as escalation, abstention, and conservative fall…

cs.LG2025

Toward Storage-Aware Learning with Compressed Data An Empirical Exploratory Study on JPEG

Kichang Lee, Songkuk Kim, JaeYeon Park +1

On-device machine learning is often constrained by limited storage, particularly in continuous data collection scenarios. This paper presents an empirical study on storage-aware le…

cs.LG2024

Tazza: Shuffling Neural Network Parameters for Secure and Private Federated Learning

Kichang Lee, Jaeho Jin, JaeYeon Park +2

Federated learning enables decentralized model training without sharing raw data, preserving data privacy. However, its vulnerability towards critical security threats, such as gra…

cs.LG2024★ 1 cited

DeTrigger: A Gradient-Centric Approach to Backdoor Attack Mitigation in Federated Learning

Kichang Lee, Yujin Shin, Jonghyuk Yun +3

Federated Learning (FL) enables collaborative model training across distributed devices while preserving local data privacy, making it ideal for mobile and embedded systems. Howeve…

cs.LG2024

Effective Heterogeneous Federated Learning via Efficient Hypernetwork-based Weight Generation

Yujin Shin, Kichang Lee, Sungmin Lee +3

While federated learning leverages distributed client resources, it faces challenges due to heterogeneous client capabilities. This necessitates allocating models suited to clients…

cs.LG2024

Improving Local Training in Federated Learning via Temperature Scaling

Kichang Lee, Pei Zhang, Songkuk Kim +1

Federated learning is inherently hampered by data heterogeneity: non-i.i.d. training data over local clients. We propose a novel model training approach for federated learning, FLe…

cs.LG2023★ 7 cited

FedAIoT: A Federated Learning Benchmark for Artificial Intelligence of Things

Samiul Alam, Tuo Zhang, Tiantian Feng +8

There is a significant relevance of federated learning (FL) in the realm of Artificial Intelligence of Things (AIoT). However, most existing FL works do not use datasets collected…