← Search

Yuhui Xu

13 accepted papers

2026

Entropy-Based Block Pruning for Efficient Large Language Models

ICLR 2026poster

As large language models continue to scale, their growing computational and storage demands pose significant challenges for real-world deployment. In this work, we investigate redundancy within Transformer-based models and propose an entropy-based pruning strategy to enhance efficiency while maintai…

Cited by 0SourceScholar
2026

Learning to Reason over Continuous Tokens with Reinforcement Learning

ICLR 2026poster

Large Language Models (LLMs) have shown strong performance in complex reasoning tasks, especially when guided by Chain-of-Thought (CoT) prompting. However, conventional CoT reasoning in the discrete token space suffers from high computational and memory costs due to verbose intermediate steps. Recen…

Cited by 0SourceScholar
2026

Scalable Chain of Thoughts via Elastic Reasoning

ICLR 2026poster

Large reasoning models (LRMs) have achieved remarkable progress on complex tasks by generating extended chains of thought (CoT). However, their uncontrolled output lengths pose significant challenges for real-world deployment, where inference-time budgets on tokens, latency, or compute are strictly…

Cited by 0SourcecodeScholar
2025

One QuantLLM for ALL: Fine-tuning Quantized LLMs Once for Efficient Deployments

ACL 2025long

Large Language Models (LLMs) have advanced rapidly but face significant memory demands. While quantization has shown promise for LLMs, current methods typically require lengthy training to alleviate the performance degradation from quantization loss. However, deploying LLMs across diverse scenarios…

2025

Reward-Guided Speculative Decoding for Efficient LLM Reasoning

ICML 2025poster

We introduce Reward-Guided Speculative Decoding (RSD), a novel framework aimed at improving the efficiency of inference in large language models (LLMs). RSD synergistically combines a lightweight draft model with a more powerful target model, incorporating a controlled bias to prioritize high-reward…

Cited by 14SourcePDFScholar
2025

Scaling Computer-Use Grounding via User Interface Decomposition and Synthesis

NeurIPS 2025spotlight

Graphical user interface (GUI) grounding, the ability to map natural language instructions to specific actions on graphical user interfaces, remains a critical bottleneck in computer use agent development. Current benchmarks oversimplify grounding tasks as short referring expressions, failing to ca…

Cited by 0SourcecodeScholar
2025

ThinK: Thinner Key Cache by Query-Driven Pruning

ICLR 2025spotlight

Large Language Models (LLMs) have revolutionized the field of natural language processing, achieving unprecedented performance across a variety of applications. However, their increased computational and memory demands present significant challenges, especially when handling long sequences. This pa…

2024

Not All Experts are Equal: Efficient Expert Pruning and Skipping for Mixture-of-Experts Large Language Models

ACL 2024long

A pivotal advancement in the progress of large language models (LLMs) is the emergence of the Mixture-of-Experts (MoE) LLMs. Compared to traditional LLMs, MoE LLMs can achieve higher performance with fewer active parameters, but it is still hard to deploy them due to their immense parameter sizes. D…

2024

QA-LoRA: Quantization-Aware Low-Rank Adaptation of Large Language Models

ICLR 2024poster

Recently years have witnessed a rapid development of large language models (LLMs). Despite the strong ability in many language-understanding tasks, the heavy computational burden largely restricts the application of LLMs especially when one needs to deploy them onto edge devices. In this paper, we p…

2024

SPP: Sparsity-Preserved Parameter-Efficient Fine-Tuning for Large Language Models

ICML 2024poster

Large Language Models (LLMs) have become pivotal in advancing the field of artificial intelligence, yet their immense sizes pose significant challenges for both fine-tuning and deployment. Current post-training pruning methods, while reducing the sizes of LLMs, often fail to maintain their original…

2021

Fitting the Search Space of Weight-sharing NAS with Graph Convolutional Networks

AAAI 2021technical

Neural architecture search has attracted wide attentions in both academia and industry. To accelerate it, researchers proposed weight-sharing methods which first train a super-network to reuse computation among different operators, from which exponentially many sub-networks can be sampled and effici…

Cited by 20SourcePDFScholar
2020

PC-DARTS: Partial Channel Connections for Memory-Efficient Architecture Search

ICLR 2020spotlight

Differentiable architecture search (DARTS) provided a fast solution in finding effective network architectures, but suffered from large memory and computing overheads in jointly training a super-net and searching for an optimal architecture. In this paper, we present a novel approach, namely Partia…

Cited by 920SourcecodeScholar
2020

TRP: Trained Rank Pruning for Efficient Deep Neural Networks

IJCAI 2020poster

To enable DNNs on edge devices like mobile phones, low-rank approximation has been widely adopted because of its solid theoretical rationale and efficient implementations. Several previous works attempted to directly approximate a pre-trained model by low-rank decomposition; however, small approxima…

Cited by 0SourcePDFScholar