← Search

ZeKai Chen

8 accepted papers

2026

DANCE: Dynamic, Available, Neighbor-gated Condensation for Federated Text-Attributed Graphs

ICML 2026poster

Federated graph learning (FGL) enables collaborative training on graph data across multiple clients. With the rise of large language models (LLMs), textual attributes in FGL graphs are gaining attention. Text-attributed graph federated learning (TAG-FGL) improves FGL by explicitly leveraging LLMs to…

Cited by 0SourceScholar
2026

Think-at-Hard: Selective Latent Iterations to Improve Reasoning Language Models

ICML 2026poster

Improving reasoning abilities of Large Language Models (LLMs), especially under parameter constraints, is crucial for real-world applications. Looped transformers address this by performing multiple latent iterations to refine each token beyond a single forward pass. However, we identify a latent ov…

Cited by 0SourceScholar
2026

Towards Docking-oriented De Novo Ligand Design via Gradient Inversion

ICML 2026poster

De novo ligand design is a fundamental task that seeks to generate protein or molecule candidates that can effectively dock with protein receptors and achieve strong binding affinity entirely from scratch. It holds paramount significance for a wide spectrum of biomedical applications. However, most …

Cited by 0SourceScholar
2025

Differentially Private Fine-Tuning of Diffusion Models

ICCV 2025poster

Generative AI models, particularly diffusion models (DMs), have demonstrated exceptional capabilities in high-quality image synthesis. However, their large memorization capacity raises significant privacy concerns, especially when trained on sensitive datasets. This paper introduces DP-LoRA, a surpr…

2025

Rising from Ashes: Generalized Federated Learning via Dynamic Parameter Reset

NeurIPS 2025poster

Although Federated Learning (FL) is promising in privacy-preserving collaborative model training, it faces low inference performance due to heterogeneous data among clients. Due to heterogeneous data in each client, FL training easily learns the specific overfitting features. Existing FL methods ad…

Cited by 0SourceScholar
2025

SADA: Stability-guided Adaptive Diffusion Acceleration

ICML 2025poster

Diffusion models have achieved remarkable success in generative tasks but suffer from high computational costs due to their iterative sampling process and quadratic‐attention costs. Existing training-free acceleration strategies that reduce per-step computation cost, while effectively reducing samp…

2023

Boosting Transformers and Language Models for Clinical Prediction in Immunotherapy

ACL 2023industry

Clinical prediction is an essential task in the healthcare industry. However, the recent success of transformers, on which large language models are built, has not been extended to this domain. In this research, we explore the use of transformers and language models in prognostic prediction for immu…

Cited by 7SourcePDFScholar
2022

ASM2TV: An Adaptive Semi-supervised Multi-Task Multi-View Learning Framework for Human Activity Recognition

AAAI 2022technical

Many real-world scenarios, such as human activity recognition (HAR) in IoT, can be formalized as a multi-task multi-view learning problem. Each specific task consists of multiple shared feature views collected from multiple sources, either homogeneous or heterogeneous. Common among recent approaches…