← Search

Yae Jee Cho

6 accepted papers

2026

Long-Context Modeling with Dynamic Hierarchical Sparse Attention for Memory-Constrained LLM Inference

ICML 2026spotlight

The quadratic cost of attention limits the scalability of long-context LLMs, especially under limited hardware memory budgets. While attention is often sparse, existing static sparse methods cannot adapt to task- or input-dependent variations, and recent dynamic approaches rely on predefined templat…

Cited by 0SourceScholar
2024

Heterogeneous LoRA for Federated Fine-tuning of On-Device Foundation Models

EMNLP 2024main

Foundation models (FMs) adapt surprisingly well to downstream tasks with fine-tuning. However, their colossal parameter space prohibits their training on resource-constrained edge-devices. For federated fine-tuning, we need to consider the smaller FMs of few billion parameters at most, namely on-dev…

Cited by 57SourcePDFScholar
2023

Local or Global: Selective Knowledge Assimilation for Federated Learning with Limited Labels

ICCV 2023poster

Many existing FL methods assume clients with fully-labeled data, while in realistic settings, clients have limited labels due to the expensive and laborious process of labeling. Limited labeled local data of the clients often leads to their local model having poor generalization abilities to their l…

Cited by 10PDFScholar
2023

On the Convergence of Federated Averaging with Cyclic Client Participation

ICML 2023poster

Federated Averaging (FedAvg) and its variants are the most popular optimization algorithms in federated learning (FL). Previous convergence analyses of FedAvg either assume full client participation or partial client participation where the clients can be uniformly sampled. However, in practical cro…

Cited by 37SourcePDFScholar
2022

Heterogeneous Ensemble Knowledge Transfer for Training Large Models in Federated Learning

IJCAI 2022poster

Federated learning (FL) enables edge-devices to collaboratively learn a model without disclosing their private data to a central aggregating server. Most existing FL algorithms require models of identical architecture to be deployed across the clients and server, making it infeasible to train large…

Cited by 159SourcePDFScholar