← Search

Yehui Tang

55 accepted papers

2026

GenVidBench: A 6-Million Benchmark for AI-Generated Video Detection

AAAI 2026technical

The rapid advancement of video generation models has made it increasingly challenging to distinguish AI-generated videos from real ones. This issue underscores the urgent need for effective AI-generated video detectors to prevent the dissemination of false information via such videos. However, the d

Cited by 0SourcePDFScholar
2026

PocketLLM: Ultimate Compression of Large Language Models via Meta Networks

AAAI 2026technical

As Large Language Models (LLMs) continue to grow in size, storing and transmitting them on edge devices becomes increasingly challenging. Traditional methods like quantization and pruning struggle to achieve extreme compression of LLMs without sacrificing accuracy. In this paper, we introduce Pocket

Cited by 0SourcePDFScholar
2026

Towards Proprioception-Aware Embodied Planning for Dual-Arm Humanoid Robots

ICRA 2026poster

In recent years, Multimodal Large Language Models (MLLMs) have demonstrated the ability to serve as high-level planners, enabling robots to follow complex human instructions. However, their effectiveness, especially in long-horizon tasks involving dual-arm humanoid robots, remains limited. This limi…

2025

CBQ: Cross-Block Quantization for Large Language Models

ICLR 2025spotlight

Post-training quantization (PTQ) has played a pivotal role in compressing large language models (LLMs) at ultra-low costs. Although current PTQ methods have achieved promising results by addressing outliers and employing layer- or block-wise loss optimization techniques, they still suffer from signi…

Cited by 13SourcePDFScholar
2025

DenseSSM: State Space Models with Dense Hidden Connection for Efficient Large Language Models

NAACL 2025long

Large language models (LLMs) face a significant challenge due to the excessive computational and memory requirements of the commonly used Transformer architecture. While state space model (SSM) is a new type of foundational network architecture offering lower computational complexity, their performa…

Cited by 0SourcePDFScholar
2025

EMS-SD: Efficient Multi-sample Speculative Decoding for Accelerating Large Language Models

NAACL 2025long

Speculative decoding emerges as a pivotal technique for enhancing the inference speed of Large Language Models (LLMs). Despite recent research aiming to improve prediction efficiency, multi-sample speculative decoding has been overlooked due to varying numbers of accepted tokens within a batch in th…

2025

Eve: Efficient Multimodal Vision Language Models with Elastic Visual Experts

AAAI 2025technical

Multimodal vision language models (VLMs) have made significant progress with the support of continuously increasing model sizes and data volumes. Running VLMs on edge devices has become a challenge for their widespread application. There are several efficient VLM efforts, but they often sacrifice li…

2025

Forest-of-Thought: Scaling Test-Time Compute for Enhancing LLM Reasoning

ICML 2025poster

Large Language Models (LLMs) have demonstrated remarkable abilities across various language tasks, but solving complex reasoning problems remains a significant challenge. While existing methods, such as Chain-of-Thought (CoT) and Tree-of-Thought (ToT), enhance reasoning by decomposing problems or st…

2025

LLM Data Selection and Utilization via Dynamic Bi-level Optimization

ICML 2025poster

While large-scale training data is fundamental for developing capable large language models (LLMs), strategically selecting high-quality data has emerged as a critical approach to enhance training efficiency and reduce computational costs. Current data selection methodologies predominantly rely on s…

Cited by 0SourcePDFScholar
2025

QEM-Bench: Benchmarking Learning-based Quantum Error Mitigation and QEMFormer as a Multi-ranged Context Learning Baseline

ICML 2025poster

Quantum Error Mitigation (QEM) has emerged as a pivotal technique for enhancing the reliability of noisy quantum devices in the *Noisy Intermediate-Scale Quantum* (NISQ) era. Recently, machine learning (ML)-based QEM approaches have demonstrated strong generalization capabilities without sampling ov…

Cited by 0SourcePDFScholar
2025

SpeCache: Speculative Key-Value Caching for Efficient Generation of LLMs

ICML 2025poster

Transformer-based large language models (LLMs) have already achieved remarkable results on long-text tasks, but the limited GPU memory (VRAM) resources struggle to accommodate the linearly growing demand for key-value (KV) cache as the sequence length increases, which has become a bottleneck for the…

Cited by 0SourcePDFScholar
2025

Tensor Network: from the Perspective of AI4Science and Science4AI

IJCAI 2025

Tensor network has been a promising numerical tool for computational problems across science and AI. For their emerging and fast development especially in the intersection between AI and science, this paper tries to present a compact review, regarding both their applications and its own recent techn

Cited by 0SourcePDFScholar
2025

TinySAM: Pushing the Envelope for Efficient Segment Anything Model

AAAI 2025technical

Recently segment anything model (SAM) has shown powerful segmentation capability and has drawn great attention in computer vision fields. Massive following works have developed various applications based on the pre-trained SAM and achieved impressive performance on downstream vision tasks. However,…

2024

Adapt without Forgetting: Distill Proximity from Dual Teachers in Vision-Language Models

ECCV 2024poster

"Multi-modal models such as CLIP possess remarkable zero-shot transfer capabilities, making them highly effective in continual learning tasks. However, this advantage is severely compromised by catastrophic forgetting, which undermines the valuable zero-shot learning abilities of these models. Exist…

2024

Circuit Design and Efficient Simulation of Quantum Inner Product and Empirical Studies of Its Effect on Near-Term Hybrid Quantum-Classic Machine Learning

CVPR 2024poster

For the essential operation namely inner product (IP) as widely adopted in classic computing e.g. matrix multiplication its quantum counterpart: quantum inner product (QIP) has also been recently theoretically explored with a verifiable lower complexity on quantum computers. However it remains uncle…

2024

Context-Guided Spatial Feature Reconstruction for Efficient Semantic Segmentation

ECCV 2024poster

"Semantic segmentation is an important task for numerous applications but it is still quite challenging to achieve advanced performance with limited computational costs. In this paper, we present CGRSeg, an efficient yet competitive segmentation framework based on context-guided spatial feature reco…

2024

Data-efficient Large Vision Models through Sequential Autoregression

ICML 2024poster

Training general-purpose vision models on purely sequential visual data, eschewing linguistic inputs, has heralded a new frontier in visual understanding. These models are intended to not only comprehend but also seamlessly transit to out-of-domain tasks. However, current endeavors are hamstrung by…

2024

ExCP: Extreme LLM Checkpoint Compression via Weight-Momentum Joint Shrinking

ICML 2024oral

Large language models (LLM) have recently attracted significant attention in the field of artificial intelligence. However, the training process of these models poses significant challenges in terms of computational and storage capacities, thus compressing checkpoints has become an urgent problem. I…

2024

Kangaroo: Lossless Self-Speculative Decoding for Accelerating LLMs via Double Early Exiting

NeurIPS 2024poster

Speculative decoding has demonstrated its effectiveness in accelerating the inference of large language models (LLMs) while maintaining an identical sampling distribution. However, the conventional approach of training separate draft model to achieve a satisfactory token acceptance rate can be costl…

2024

Memory-Space Visual Prompting for Efficient Vision-Language Fine-Tuning

ICML 2024poster

Current solutions for efficiently constructing large vision-language (VL) models follow a two-step paradigm: projecting the output of pre-trained vision encoders to the input space of pre-trained language models as visual prompts; and then transferring the models to downstream VL tasks via end-to-en…

2024

MemoryFormer : Minimize Transformer Computation by Removing Fully-Connected Layers

NeurIPS 2024poster

In order to reduce the computational complexity of large language models, great efforts have been made to to improve the efficiency of transformer models such as linear attention and flash-attention. However, the model size and corresponding computational complexity are constantly scaled up in pursu…

Cited by 0SourcePDFScholar
2024

Node2ket: Efficient High-Dimensional Network Embedding in Quantum Hilbert Space

ICLR 2024poster

Network embedding (NE) is a prominent technique for network analysis where the nodes are represented as vectorized embeddings in a continuous space. Existing works tend to resort to the low-dimensional embedding space for efficiency and less risk of over-fitting. In this paper, we explore a new NE p…

Cited by 2SourcePDFScholar
2024

Rethinking Optimization and Architecture for Tiny Language Models

ICML 2024poster

The power of large language models (LLMs) has been demonstrated through numerous data and computing resources. However, the application of language models on mobile devices is facing huge challenge on the computation and memory costs, that is, tiny language models with high performance are urgently…

2024

SLAB: Efficient Transformers with Simplified Linear Attention and Progressive Re-parameterized Batch Normalization

ICML 2024poster

Transformers have become foundational architectures for both natural language and computer vision tasks. However, the high computational cost makes it quite challenging to deploy on resource-constraint devices. This paper investigates the computational bottleneck modules of efficient transformer, *i…

2024

SSL4Q: Semi-Supervised Learning of Quantum Data with Application to Quantum State Classification

ICML 2024poster

The accurate classification of quantum states is crucial for advancing quantum computing, as it allows for the effective analysis and correct functioning of quantum devices by analyzing the statistics of the data from quantum measurements. Traditional supervised methods, which rely on extensive labe…

Cited by 2SourcePDFScholar
2024

Star-Agents: Automatic Data Optimization with LLM Agents for Instruction Tuning

NeurIPS 2024poster

The efficacy of large language models (LLMs) on downstream tasks usually hinges on instruction tuning, which relies critically on the quality of training data. Unfortunately, collecting high-quality and diverse data is both expensive and time-consuming. To mitigate this issue, we propose a novel St…

2024

Token Compensator: Altering Inference Cost of Vision Transformer without Re-Tuning

ECCV 2024poster

"Token compression expedites the training and inference of Vision Transformers (ViTs) by reducing the number of the redundant tokens, , pruning inattentive tokens or merging similar tokens. However, when applied to downstream tasks, these approaches suffer from significant performance drop when the…

2024

Towards LLM4QPE: Unsupervised Pretraining of Quantum Property Estimation and A Benchmark

ICLR 2024spotlight

Estimating the properties of quantum systems such as quantum phase has been critical in addressing the essential quantum many-body problems in physics and chemistry. Deep learning models have been recently introduced to property estimation, surpassing conventional statistical approaches. However, t…

Cited by 3SourcePDFScholar
2023

Masked Image Modeling With Local Multi-Scale Reconstruction

CVPR 2023highlight

Masked Image Modeling (MIM) achieves outstanding success in self-supervised representation learning. Unfortunately, MIM models typically have huge computational burden and slow learning process, which is an inevitable obstacle for their industrial applications. Although the lower layers play the key…

2023

Network Expansion for Practical Training Acceleration

CVPR 2023poster

Recently, the sizes of deep neural networks and training datasets both increase drastically to pursue better performance in a practical sense. With the prevalence of transformer-based models in vision tasks, even more pressure is laid on the GPU platforms to train these heavy models, which consumes…

2023

One-for-All: Bridge the Gap Between Heterogeneous Architectures in Knowledge Distillation

NeurIPS 2023poster

Knowledge distillation (KD) has proven to be a highly effective approach for enhancing model performance through a teacher-student training scheme. However, most existing distillation methods are designed under the assumption that the teacher and student models belong to the same model family, parti…

2022

Bridge the Gap Between Architecture Spaces via A Cross-Domain Predictor

NeurIPS 2022accept

Neural Architecture Search (NAS) can automatically design promising neural architectures without artificial experience. Though it achieves great success, prohibitively high search cost is required to find a high-performance architecture, which blocks its practical implementation. Neural predictor ca…

Cited by 7SourcePDFScholar
2022

CMT: Convolutional Neural Networks Meet Vision Transformers

CVPR 2022poster

Vision transformers have been successfully applied to image recognition tasks due to their ability to capture long-range dependencies within an image. However, there are still gaps in both performance and computational cost between transformers and existing convolutional neural networks (CNNs). In t…

Cited by 954PDFcodeScholar
2022

GhostNetV2: Enhance Cheap Operation with Long-Range Attention

NeurIPS 2022accept

Light-weight convolutional neural networks (CNNs) are specially designed for applications on mobile devices with faster inference speed. The convolutional operation can only capture local information in a window region, which prevents performance from being further improved. Introducing self-attent…

2022

Hire-MLP: Vision MLP via Hierarchical Rearrangement

CVPR 2022poster

Previous vision MLPs such as MLP-Mixer and ResMLP accept linearly flattened image patches as input, making them inflexible for different input sizes and hard to capture spatial information. Such approach withholds MLPs from getting comparable performance with their transformer-based counterparts and…

Cited by 140PDFcodeScholar
2022

Learning Efficient Vision Transformers via Fine-Grained Manifold Distillation

NeurIPS 2022accept

In the past few years, transformers have achieved promising performance on various computer vision tasks. Unfortunately, the immense inference overhead of most existing vision transformers withholds them from being deployed on edge devices such as cell phones and smart watches. Knowledge distillatio…

Cited by 71SourcePDFScholar
2022

Source-Free Domain Adaptation via Distribution Estimation

CVPR 2022poster

Domain Adaptation aims to transfer the knowledge learned from a labeled source domain to an unlabeled target domain whose data distributions are different. However, the training data in source domain required by most of the existing methods is usually unavailable in real-world applications due to pr…

Cited by 161PDFScholar
2022

Spatial-Channel Token Distillation for Vision MLPs

ICML 2022spotlight

Recently, neural architectures with all Multi-layer Perceptrons (MLPs) have attracted great research interest from the computer vision community. However, the inefficient mixing of spatial-channel information causes MLP-like vision models to demand tremendous pre-training on large-scale datasets. Th…

Cited by 6SourcePDFScholar
2022

Vision GNN: An Image is Worth Graph of Nodes

NeurIPS 2022accept

Network architecture plays a key role in the deep learning-based computer vision system. The widely-used convolutional neural network and transformer treat the image as a grid or sequence structure, which is not flexible to capture irregular and complex objects. In this paper, we propose to represen…

2021

Augmented Shortcuts for Vision Transformers

NeurIPS 2021poster

Transformer models have achieved great progress on computer vision tasks recently. The rapid development of vision transformers is mainly contributed by their high representation ability for extracting informative features from input images. However, the mainstream transformer models are designed wi…

2021

Learning Frequency Domain Approximation for Binary Neural Networks

NeurIPS 2021oral

Binary neural networks (BNNs) represent original full-precision weights and activations into 1-bit with sign function. Since the gradient of the conventional sign function is almost zero everywhere which cannot be used for back-propagation, several attempts have been proposed to alleviate the optimi…

Cited by 71SourcePDFScholar
2021

ReNAS: Relativistic Evaluation of Neural Architecture Search

CVPR 2021poster

An effective and efficient architecture performance evaluation scheme is essential for the success of Neural Architecture Search (NAS). To save computational cost, most of existing NAS algorithms often train and evaluate intermediate neural architectures on a small proxy dataset with limited trainin…

Cited by 106PDFcodeScholar
2020

Frequency Domain Compact 3D Convolutional Neural Networks

CVPR 2020poster

This paper studies the compression and acceleration of 3-dimensional convolutional neural networks (3D CNNs). To reduce the memory cost and computational complexity of deep neural networks, a number of algorithms have been explored by discovering redundant parameters in pre-trained networks. However…

Cited by 32PDFScholar
2020

SCOP: Scientific Control for Reliable Neural Network Pruning

NeurIPS 2020poster

This paper proposes a reliable neural network pruning algorithm by setting up a scientific control. Existing pruning methods have developed various hypotheses to approximate the importance of filters to the network and then execute filter pruning accordingly. To increase the reliability of the resul…