← Search

Zhenheng Tang

22 accepted papers

2026

Ghost in the Cloud: Your Geo-Distributed Large Language Models Training is Easily Manipulated

ICLR 2026poster

Geo-distributed training and Federated Learning (FL) provide viable solutions to address the substantial data and computational resource needs associated with training large language models (LLMs). However, we empirically demonstrate that a single adversarial participant can significantly compromise…

Cited by 0SourceScholar
2026

GitTaskBench: A Benchmark for Code Agents Solving Real-World Tasks Through Code Repository Leveraging

AAAI 2026technical

Beyond scratch coding, exploiting large-scale code repositories (e.g., GitHub) for practical tasks is vital in real-world software development, yet current benchmarks rarely evaluate code agents in such authentic, workflow-driven scenarios. To bridge this gap, we introduce GitTaskBench, a benchmark

Cited by 0SourcePDFScholar
2026

Identifying and Mitigating Errors in Gradient Aggregation of Distributed Data Parallel Training

ICML 2026poster

Hardware-related silent data corruptions during gradient aggregation pose significant challenges to fault-tolerant distributed training, often leading to slow or failed convergence. To address this, we first mathematically formulate these errors as gradient inconsistency and theoretically analyze ho…

Cited by 0SourceScholar
2026

JailbreakLoRA: Your Downloaded LoRA from Sharing Platforms might be Unsafe

ICLR 2026poster

Low-Rank Adaptation (LoRA) benefits from its plug-and-play nature, enabling large language models (LLMs) to achieve significant performance gains at low cost, has driven the development of LoRA-sharing platforms. However, the jailbreak and backdoor concerns associated with LoRA-sharing platforms rem…

Cited by 0SourceScholar
2026

Reasoning Language Model Inference Serving Unveiled: An Empirical Study

ICLR 2026poster

The reasoning large language model (RLLM) has been proven competitive in solving complex reasoning tasks such as mathematics, coding, compared to traditional LLM. However, the serving performance and behavior of RLLM remains \textit{unexplored}, which may undermine the deployment and utilization of…

Cited by 0SourcecodeScholar
2026

Semantic Integrity Matters: Benchmarking and Preserving High-Density Reasoning in KV Cache Compression

ICML 2026poster

While Key-Value (KV) cache compression is essential for efficient LLM inference, current evaluations disproportionately focus on \textbf{sparse retrieval} tasks, potentially masking the degradation of High-Density Reasoning where Chain-of-Thought (CoT) coherence is critical. We introduce KVFundaBenc…

Cited by 0SourceScholar
2026

VCG-Bench: Towards A Unified Visual-Centric Benchmark for Structured Generation and Editing

ICML 2026poster

Despite the rapid advancements in Vision-Language Models (VLMs), a critical gap remains in their ability to handle structured, controllable diagrammatic tasks essential for professional workflows, as existing methods predominantly rely on pixel-based synthesis which operates in probabilistic pixel s…

Cited by 0SourceScholar
2025

Can Compressed LLMs Truly Act? An Empirical Evaluation of Agentic Capabilities in LLM Compression

ICML 2025poster

Post-training compression reduces the computational and memory costs of large language models (LLMs), enabling resource-efficient deployment. However, existing compression benchmarks focus narrowly on language modeling (e.g., perplexity) and natural language understanding tasks (e.g., GLUE accuracy)…

2025

ChunkKV: Semantic-Preserving KV Cache Compression for Efficient Long-Context LLM Inference

NeurIPS 2025poster

Large Language Models (LLMs) require significant GPU memory when processing long texts, with the key value (KV) cache consuming up to 70\% of total memory during inference. Although existing compression methods reduce memory by evaluating the importance of individual tokens, they overlook critical s…

Cited by 0SourcecodeScholar
2025

Hot-pluggable Federated Learning: Bridging General and Personalized FL via Dynamic Selection

ICLR 2025poster

Personalized federated learning (PFL) achieves high performance by assuming clients only meet test data locally, which does not meet many generic federated learning (GFL) scenarios. In this work, we theoretically show that PMs can be used to enhance GFL with a new learning problem named Selective FL…

Cited by 0SourcePDFScholar
2025

MegaAgent: A Large-Scale Autonomous LLM-based Multi-Agent System Without Predefined SOPs

ACL 2025finding

LLM-based multi-agent systems (MAS) have shown promise in tackling complex tasks. However, existing solutions often suffer from limited agent coordination and heavy reliance on predefined Standard Operating Procedures (SOPs), which demand extensive human input. To address these limitations, we propo…

2025

One-shot Federated Learning Methods: A Practical Guide

IJCAI 2025

One-shot Federated Learning (OFL) is a distributed machine learning paradigm that constrains client-server communication to a single round, addressing privacy and communication overhead issues associated with multiple rounds of data exchange in traditional Federated Learning (FL). OFL demonstrates t

Cited by 0SourcePDFScholar
2025

ParZC: Parametric Zero-Cost Proxies for Efficient NAS

AAAI 2025technical

Recent advancements in Zero-shot Neural Architecture Search (NAS) highlight the ability of zero-cost proxies in identifying superior architecture. However, we identify a critical issue with current zero-cost proxies: they aggregate node-wise zero-cost statistics without considering that not all node…

Cited by 7SourcePDFScholar
2025

STBLLM: Breaking the 1-Bit Barrier with Structured Binary LLMs

ICLR 2025poster

In this paper, we present the first structural binarization method for LLM compression to less than 1-bit precision. Although LLMs have achieved remarkable performance, their memory-bound nature during the inference stage hinders the adoption of resource-constrained devices. Reducing weights to 1-bi…

2024

Discovering Sparsity Allocation for Layer-wise Pruning of Large Language Models

NeurIPS 2024poster

In this paper, we present DSA, the first automated framework for discovering sparsity allocation schemes for layer-wise pruning in Large Language Models (LLMs). LLMs have become increasingly powerful, but their large parameter counts make them computationally expensive. Existing pruning methods fo…

Cited by 10SourcePDFScholar
2024

FedImpro: Measuring and Improving Client Update in Federated Learning

ICLR 2024poster

Federated Learning (FL) models often experience client drift caused by heterogeneous data, where the distribution of data differs across clients. To address this issue, advanced research primarily focuses on manipulating the existing gradients to achieve more consistent client models. In this paper,…

Cited by 21SourcePDFScholar
2024

FuseFL: One-Shot Federated Learning through the Lens of Causality with Progressive Model Fusion

NeurIPS 2024spotlight

One-shot Federated Learning (OFL) significantly reduces communication costs in FL by aggregating trained models only once. However, the performance of advanced OFL methods is far behind the normal FL. In this work, we provide a causal view to find that this performance drop of OFL methods comes from…

2024

LPZero: Language Model Zero-cost Proxy Search from Zero

EMNLP 2024finding

Despite the outstanding performance, Neural Architecture Search (NAS) is criticized for massive computation. Recently, Zero-shot NAS has emerged as a promising approach by exploiting Zero-cost (ZC) proxies, which markedly reduce computational demands. Despite this, existing ZC proxies heavily rely o…

Cited by 2SourcePDFScholar
2024

Pruner-Zero: Evolving Symbolic Pruning Metric From Scratch for Large Language Models

ICML 2024poster

Despite the remarkable capabilities, Large Language Models (LLMs) face deployment challenges due to their extensive size. Pruning methods drop a subset of weights to accelerate, but many of them require retraining, which is prohibitively expensive and computationally demanding. Recently, post-traini…

2024

Should We Really Edit Language Models? On the Evaluation of Edited Language Models

NeurIPS 2024poster

Model editing has become an increasingly popular alternative for efficiently updating knowledge within language models. Current methods mainly focus on reliability, generalization, and locality, with many methods excelling across these criteria. Some recent works disclose the pitfalls of these ed…

2023

NAS-LID: Efficient Neural Architecture Search with Local Intrinsic Dimension

AAAI 2023technical

One-shot neural architecture search (NAS) substantially improves the search efficiency by training one supernet to estimate the performance of every possible child architecture (i.e., subnet). However, the inconsistency of characteristics among subnets incurs serious interference in the optimization…

2022

Virtual Homogeneity Learning: Defending against Data Heterogeneity in Federated Learning

ICML 2022spotlight

In federated learning (FL), model performance typically suffers from client drift induced by data heterogeneity, and mainstream works focus on correcting client drift. We propose a different approach named virtual homogeneity learning (VHL) to directly “rectify” the data heterogeneity. In particular…