← Search

Dong-Jun Han

22 accepted papers

2026

Bridging On-Device and Cloud LLMs for Collaborative Reasoning: A Unified Methodology for Local Routing and Post-Training

ICML 2026poster

Device-cloud collaboration holds promise for deploying large language models (LLMs), leveraging lightweight on-device models for efficiency while relying on powerful cloud models for superior reasoning. A central challenge in this setting is determining, for each incoming query, whether it should be…

Cited by 0SourceScholar
2026

Federated Sketching LoRA: A Flexible Framework for Heterogeneous Collaborative Fine-Tuning of LLMs

ICML 2026poster

Fine-tuning large language models (LLMs) on resource-constrained clients remains a challenging problem. Recent works have fused low-rank adaptation (LoRA) techniques with federated fine-tuning to mitigate challenges associated with client model sizes and data scarcity. Still, the heterogeneity of re…

Cited by 0SourceScholar
2026

Identifying Robust Neural Pathways: Few-Shot Adversarial Mask Tuning for Vision-Language Models

ICLR 2026poster

Recent vision-language models (VLMs), such as CLIP, have demonstrated remarkable transferability across a wide range of downstream tasks by effectively leveraging the joint text-image embedding space, even with only a few data samples. Despite their impressive performance, these models remain vulner…

Cited by 0SourceScholar
2026

ProLoG: Hybrid Prompt and LoRA Based Adaptation of Vision-Language Models for OOD Generalization

AAAI 2026technical

While vision-language foundation models (VLMs) achieve remarkable performance when fine-tuned on downstream in-distribution (ID) data, this process compromises their generalization ability on out-of-distribution (OOD) data that deviate from the downstream tasks due to overfitting. To address this, w

Cited by 0SourcePDFScholar
2025

Adaptive Energy Alignment for Accelerating Test-Time Adaptation

ICLR 2025poster

In response to the increasing demand for tackling out-of-domain (OOD) scenarios, test-time adaptation (TTA) has garnered significant research attention in recent years. To adapt a source pre-trained model to target samples without getting access to their labels, existing approaches have typically em…

Cited by 0SourcePDFScholar
2025

Asynchronous Federated Reinforcement Learning with Policy Gradient Updates: Algorithm Design and Convergence Analysis

ICLR 2025poster

To improve the efficiency of reinforcement learning (RL), we propose a novel asynchronous federated reinforcement learning (FedRL) framework termed AFedPG, which constructs a global model through collaboration among $N$ agents using policy gradient (PG) updates. To address the challenge of lagged po…

Cited by 19SourcePDFScholar
2025

Decentralized Sporadic Federated Learning: A Unified Algorithmic Framework with Convergence Guarantees

ICLR 2025spotlight

Decentralized federated learning (DFL) captures FL settings where both (i) model updates and (ii) model aggregations are exclusively carried out by the clients without a central server. Existing DFL works have mostly focused on settings where clients conduct a fixed number of local updates between l…

2025

LLMAP: LLM-Assisted Multi-Objective Route Planning with User Preferences

EMNLP 2025

The rise of large language models (LLMs) has made natural language-driven route planning an emerging research area that encompasses rich user objectives. Current research exhibits two distinct approaches: direct route planning using LLM-as-Agent and graph-based searching strategies. However, LLMs in

Cited by 8SourcePDFScholar
2025

PRISM: Privacy-Preserving Improved Stochastic Masking for Federated Generative Models

ICLR 2025poster

Despite recent advancements in federated learning (FL), the integration of generative models into FL has been limited due to challenges such as high communication costs and unstable training in heterogeneous data environments. To address these issues, we propose PRISM, a FL framework tailored for ge…

2025

Rethinking the Starting Point: Collaborative Pre-Training for Federated Downstream Tasks

AAAI 2025technical

A few recent studies have shown the benefits of using centrally pre-trained models to initialize federated learning (FL). However, existing methods do not generalize well when faced with an arbitrary set of downstream FL tasks. Specifically, they often (i) achieve limited accuracy, especially with u…

Cited by 0SourcePDFScholar
2025

Unlocking the Potential of Model Calibration in Federated Learning

ICLR 2025poster

Over the past several years, various federated learning (FL) methodologies have been developed to improve model accuracy, a primary performance metric in machine learning. However, to utilize FL in practical decision-making scenarios, beyond considering accuracy, the trained model must also have a r…

Cited by 3SourcePDFScholar
2024

Achieving Lossless Gradient Sparsification via Mapping to Alternative Space in Federated Learning

ICML 2024poster

Handling the substantial communication burden in federated learning (FL) still remains a significant challenge. Although recent studies have attempted to compress the local gradients to address this issue, they typically perform compression only within the original parameter space, which may potenti…

Cited by 4SourcePDFScholar
2024

Consistency-Guided Temperature Scaling Using Style and Content Information for Out-of-Domain Calibration

AAAI 2024technical

Research interests in the robustness of deep neural networks against domain shifts have been rapidly increasing in recent years. Most existing works, however, focus on improving the accuracy of the model, not the calibration performance which is another important requirement for trustworthy AI syst…

2024

Hierarchical Federated Learning with Multi-Timescale Gradient Correction

NeurIPS 2024poster

While traditional federated learning (FL) typically focuses on a star topology where clients are directly connected to a central server, real-world distributed systems often exhibit hierarchical architectures. Hierarchical FL (HFL) has emerged as a promising solution to bridge this gap, leveraging a…

2023

Active Learning for Object Detection with Evidential Deep Learning and Hierarchical Uncertainty Aggregation

ICLR 2023poster

Despite the huge success of object detection, the training process still requires an immense amount of labeled data. Although various active learning solutions for object detection have been proposed, most existing works do not take advantage of epistemic uncertainty, which is an important metric fo…

Cited by 39SourcePDFScholar
2023

NEO-KD: Knowledge-Distillation-Based Adversarial Training for Robust Multi-Exit Neural Networks

NeurIPS 2023poster

While multi-exit neural networks are regarded as a promising solution for making efficient inference via early exits, combating adversarial attacks remains a challenging problem. In multi-exit networks, due to the high dependency among different submodels, an adversarial example targeting a specific…

Cited by 6SourcePDFScholar
2023

StableFDG: Style and Attention Based Learning for Federated Domain Generalization

NeurIPS 2023poster

Traditional federated learning (FL) algorithms operate under the assumption that the data distributions at training (source domains) and testing (target domain) are the same. The fact that domain shifts often occur in practice necessitates equipping FL methods with a domain generalization (DG) capab…

Cited by 16SourcePDFScholar
2023

Test-Time Style Shifting: Handling Arbitrary Styles in Domain Generalization

ICML 2023poster

In domain generalization (DG), the target domain is unknown when the model is being trained, and the trained model should successfully work on an arbitrary (and possibly unseen) target domain during inference. This is a difficult problem, and despite active studies in recent years, it remains a grea…

Cited by 10SourcePDFScholar
2023

Warping the Space: Weight Space Rotation for Class-Incremental Few-Shot Learning

ICLR 2023top-25%

Class-incremental few-shot learning, where new sets of classes are provided sequentially with only a few training samples, presents a great challenge due to catastrophic forgetting of old knowledge and overfitting caused by lack of data. During finetuning on new classes, the performance on previous…

Cited by 63SourcePDFScholar
2021

Sageflow: Robust Federated Learning against Both Stragglers and Adversaries

NeurIPS 2021poster

While federated learning (FL) allows efficient model training with local data at edge devices, among major issues still to be resolved are: slow devices known as stragglers and malicious attacks launched by adversaries. While the presence of both of these issues raises serious concerns in practica…

Cited by 126SourcePDFScholar
2020

Election Coding for Distributed Learning: Protecting SignSGD against Byzantine Attacks

NeurIPS 2020poster

Current distributed learning systems suffer from serious performance degradation under Byzantine attacks. This paper proposes Election Coding, a coding-theoretic framework to guarantee Byzantine-robustness for distributed learning algorithms based on signed stochastic gradient descent (SignSGD) that…

Cited by 44SourcePDFScholar