← Search

Ao Sun

5 accepted papers

2026

DSSA: Dense-Sparse Switchable Attention for Seamless Short-to-Long Adaptation

ICLR 2026poster

Long-sequence processing is a critical capability for modern large language models. However, the self-attention mechanism in the standard Transformer architecture faces severe computational and memory bottlenecks when processing long sequences. While trainable sparse attention methods offer a promis…

Cited by 0SourceScholar
2025

APB: Accelerating Distributed Long-Context Inference by Passing Compressed Context Blocks across GPUs

ACL 2025long

While long-context inference is crucial for advancing large language model (LLM) applications, its prefill speed remains a significant bottleneck. Current approaches, including sequence parallelism strategies and compute reduction through approximate attention mechanisms, still fall short of deliver…

2025

FR-Spec: Accelerating Large-Vocabulary Language Models via Frequency-Ranked Speculative Sampling

ACL 2025long

Speculative sampling has emerged as an important technique for accelerating the auto-regressive generation process of large language models (LLMs) by utilizing a draft-then-verify mechanism to produce multiple tokens per forward pass. While state-of-the-art speculative sampling methods use only a si…

2025

NavigateDiff: Visual Predictors are Zero-Shot Navigation Assistants

ICRA 2025

Navigating unfamiliar environments presents significant challenges for household robots, requiring the ability to recognize and reason about novel decoration and layout. Existing reinforcement learning methods cannot be directly transferred to new environments, as they typically rely on extensive ma

Cited by 11SourcecodeScholar
2023

Explain Any Concept: Segment Anything Meets Concept-Based Explanation

NeurIPS 2023poster

EXplainable AI (XAI) is an essential topic to improve human understanding of deep neural networks (DNNs) given their black-box internals. For computer vision tasks, mainstream pixel-based XAI methods explain DNN decisions by identifying important pixels, and emerging concept-based XAI explore formin…