← Search

Zhihui Lu

6 accepted papers

2026

Bridging Draft Policy Misalignment: Group Tree Optimization for Speculative Decoding

ICLR 2026poster

Speculative decoding accelerates large language model (LLM) inference by letting a lightweight draft model propose multiple tokens that the target model verifies in parallel. Yet existing training objectives optimize only a single greedy draft path, while decoding follows a tree policy that re-ranks…

Cited by 0SourceScholar
2026

InfoDecom: Decomposing Information for Defending Against Privacy Leakage in Split Inference

AAAI 2026technical

Split inference (SI) enables users to access deep learning (DL) services without directly transmitting raw data. However, recent studies reveal that data reconstruction attacks (DRAs) can recover the original inputs from the smashed data sent from the client to the server, leading to significant pri

Cited by 0SourcePDFScholar
2026

MOC: Multi-Order Communication in LLM-based Multi-Agent Systems

ICML 2026poster

Despite the remarkable progress of Large Language Model (LLM) based Multi-Agent Systems, most research focuses on optimizing coordination topology while largely underexploring the equally critical problem: how to transmit and optimize messages among agents effectively? Current communication schemes …

Cited by 0SourceScholar
2025

Backdoor Attack on Vertical Federated Graph Neural Network Learning

IJCAI 2025

Federated Graph Neural Network (FedGNN) integrate federated learning (FL) with graph neural networks (GNNs) to enable privacy-preserving training on distributed graph data. Vertical Federated Graph Neural Network (VFGNN), a key branch of FedGNN, handles scenarios where data features and labels are d

Cited by 0SourcePDFScholar
2025

GRIFFIN: Effective Token Alignment for Faster Speculative Decoding

NeurIPS 2025poster

Speculative decoding accelerates inference in large language models (LLMs) by generating multiple draft tokens simultaneously. However, existing methods often struggle with token misalignment between the training and decoding phases, limiting their performance. To address this, we propose GRIFFIN, a…

Cited by 0SourcecodeScholar
2025

Universal Backdoor Defense via Label Consistency in Vertical Federated Learning

IJCAI 2025

Backdoor attacks in vertical federated learning (VFL) are particularly concerning as they can covertly compromise VFL decision-making, posing a severe threat to critical applications of VFL. Existing defense mechanisms typically involve either label obfuscation during training or model pruning durin

Cited by 0SourcePDFScholar