← Search

Zewen Jin

4 accepted papers

2026

AdaCluster: Adaptive Query-Key Clustering for Sparse Attention in Video Generation

CVPR 2026

Video diffusion transformers (DiTs) suffer from prohibitive inference latency due to quadratic attention complexity. Existing sparse attention methods either overlook semantic similarity, or fail to adapt to heterogeneous token distributions across layers, leading to model performance degradation. W

Cited by 0SourcecodeScholar
2026

SMIDT: High-Performance Inference Framework for MoE Models with Dynamic Top-K Routing

AAAI 2026technical

To accelerate Mixture-of-Experts (MoE) inference, the hybrid parallelism paradigm is first applying pipeline parallelism (PP) to vertically divide the model into stages, with each stage further divided horizontally using tensor or expert parallelism. On the algorithm side, dynamic Top-K routing redu

Cited by 0SourcePDFScholar
2025

BigMac: A Communication-Efficient Mixture-of-Experts Model Structure for Fast Training and Inference

AAAI 2025technical

The Mixture-of-Experts (MoE) structure scales the Transformer-based large language models (LLMs) and improves their performance with only the sub-linear increase in computation resources. Recently, a fine-grained DeepSeekMoE structure is proposed, which can further improve the computing efficiency o…

2025

HATA: Trainable and Hardware-Efficient Hash-Aware Top-k Attention for Scalable Large Model Inference

ACL 2025finding

Large Language Models (LLMs) have emerged as a pivotal research area, yet the attention module remains a critical bottleneck in LLM inference, even with techniques like KVCache to mitigate redundant computations. While various top-k attention mechanisms have been proposed to accelerate LLM inference…