← Search

Wenxuan Zeng

8 accepted papers

2026

Attention Sink Forges Native MoE in Attention Layers: Sink-Aware Training to Address Head Collapse

ICML 2026poster

Large Language Models (LLMs) often assign disproportionate attention to the first token, a phenomenon known as the attention sink. Several recent approaches aim to address this issue, including Sink Attention in GPT-OSS and Gated Attention in Qwen3-Next. However, a comprehensive analysis of the rela…

Cited by 0SourceScholar
2026

Jupiter: Enhancing LLM Data Analysis Capabilities via Notebook and Inference-Time Value-Guided Search

AAAI 2026technical

Large language models (LLMs) have shown great promise in automating data science workflows. However, existing models still struggle with multi-step reasoning and tool use, limiting their effectiveness on complex data analysis tasks. To address this limitation, we propose a scalable pipeline that ext

Cited by 0SourcePDFScholar
2025

MPCache: MPC-Friendly KV Cache Eviction for Efficient Private LLM Inference

NeurIPS 2025poster

Private large language model (LLM) inference based on secure multi-party computation (MPC) achieves formal data privacy protection but suffers from significant latency overhead, especially for long input sequences. While key-value (KV) cache eviction and sparse attention algorithms have been propose…

Cited by 0SourceScholar
2024

BAT: Behavior-Aware Human-Like Trajectory Prediction for Autonomous Driving

AAAI 2024technical

The ability to accurately predict the trajectory of surrounding vehicles is a critical hurdle to overcome on the journey to fully autonomous vehicles. To address this challenge, we pioneer a novel behavior-aware trajectory prediction model (BAT) that incorporates insights and findings from traffic p…

2024

FaiMA: Feature-aware In-context Learning for Multi-domain Aspect-based Sentiment Analysis

COLING 2024main

Multi-domain aspect-based sentiment analysis (ABSA) seeks to capture fine-grained sentiment across diverse domains. While existing research narrowly focuses on single-domain applications constrained by methodological limitations and data scarcity, the reality is that sentiment naturally traverses mu…

2023

CoPriv: Network/Protocol Co-Optimization for Communication-Efficient Private Inference

NeurIPS 2023poster

Deep neural network (DNN) inference based on secure 2-party computation (2PC) can offer cryptographically-secure privacy protection but suffers from orders of magnitude latency overhead due to enormous communication. Previous works heavily rely on a proxy metric of ReLU counts to approximate the com…

Cited by 6SourcePDFScholar
2023

Converge to the Truth: Factual Error Correction via Iterative Constrained Editing

AAAI 2023technical

Given a possibly false claim sentence, how can we automatically correct it with minimal editing? Existing methods either require a large number of pairs of false and corrected claims for supervised training or do not handle well errors spanning over multiple tokens within an utterance. In this paper…

2023

MPCViT: Searching for Accurate and Efficient MPC-Friendly Vision Transformer with Heterogeneous Attention

ICCV 2023poster

Secure multi-party computation (MPC) enables computation directly on encrypted data and protects both data and model privacy in deep learning inference. However, existing neural network architectures, including Vision Transformers (ViTs), are not designed or optimized for MPC and incur significant l…

Cited by 23PDFcodeScholar