← Search

Ruihao Gong

42 accepted papers

2026

Incremental BPE Tokenization

ICML 2026spotlight

We propose a novel algorithm for incremental Byte Pair Encoding (BPE) tokenization. The algorithm processes each input byte in **worst-case** $\mathcal{O}(\log^2 t)$ time, leading to an overall complexity of $\mathcal{O}(n \log^2 t)$, where $n$ is the input length and $t$ is the maximum token length…

Cited by 0SourceScholar
2026

LLMC+: Benchmarking Vision-Language Model Compression with a plug-and-play Toolkit

AAAI 2026technical

Large Vision-Language Models (VLMs) exhibit impressive multi-modal capabilities but suffer from prohibitive computational and memory demands, due to their long visual token sequences and massive parameter sizes. To address these issues, recent works have proposed training-free compression methods. H

Cited by 0SourcePDFScholar
2026

Light Forcing: Accelerating Autoregressive Video Diffusion via Sparse Attention

ICML 2026poster

Advanced autoregressive (AR) video generation models have improved visual fidelity and interactivity, but the quadratic complexity of attention remains a primary bottleneck for efficient deployment. While existing sparse attention solutions have shown promise on bidirectional models, we identify tha…

Cited by 0SourceScholar
2026

LinVideo: A Post-Training Framework towards O(n) Attention in Efficient Video Generation

CVPR 2026

Video diffusion models (DMs) have enabled high-quality video synthesis, but their computation costs scale quadratically with sequence length due to the nature of self-attention. While linear attention offers a more efficient alternative, fully replacing quadratic attention demands costly pretraining

Cited by 0SourceScholar
2026

MoDES: Accelerating Mixture-of-Experts Multimodal Large Language Models via Dynamic Expert Skipping

CVPR 2026

Mixture-of-Experts (MoE) Multimodal large language models (MLLMs) excel at vision-language tasks, but they suffer from high computational inefficiency. To reduce inference overhead, expert skipping methods have been proposed to deactivate redundant experts based on the current input tokens. However,

Cited by 0SourcecodeScholar
2026

OmniFit: Bridging Modalities via Layer-Adaptive Token Compression for Omnimodal Large Language Models

ICML 2026spotlight

Emerging Omni-modal Large Language Models (OmniLLMs) enable real-time interaction across video, audio, and text but suffer from prohibitive computational costs due to the quadratic complexity of processing continuous streaming inputs. Existing token compression strategies remain suboptimal as they t…

Cited by 0SourceScholar
2026

Phased DMD: Few-step Distribution Matching Distillation via Score Matching within Subintervals

CVPR 2026

Distribution Matching Distillation (DMD) distills score-based generative models into efficient one-step generators, without requiring a one-to-one correspondence with the sampling trajectories of their teachers. Yet, the limited capacity of one-step distilled models compromises generative diversity

Cited by 0SourcecodeScholar
2026

Post-Training Quantization for Video Matting

ICLR 2026poster

Video matting is crucial for applications such as film production and virtual reality, yet deploying its computationally intensive models on resource-constrained devices presents challenges. Quantization is a key technique for model compression and acceleration. As an efficient approach, Post-Traini…

Cited by 0SourceScholar
2026

QVGen: Pushing the Limit of Quantized Video Generative Models

ICLR 2026poster

Video diffusion models (DMs) have enabled high-quality video synthesis. Yet, their substantial computational and memory demands pose serious challenges to real-world deployment, even on high-end GPUs. As a commonly adopted solution, quantization has proven notable success in reducing cost for image…

Cited by 0SourcecodeScholar
2026

SGMD: Score Gradient Matching Distillation for Few-Step Video Diffusion Distillation

ICML 2026poster

Distribution Matching Distillation (DMD) is a widely used paradigm for accelerating inference in few-step video diffusion models. However, DMD-style training faces a structural bottleneck: the student-side auxiliary score network (the fake score) must closely track a continuously evolving generator.…

Cited by 0SourceScholar
2026

StructureBench: A Unified Benchmark Suite for Multi-Scenario Structured Generation Tasks with On-Device Models

IJCAI 2026

Structured output generation is increasingly critical for real-world AI systems, particularly in on-device settings where small language models (0.5B–8B parameters) must produce machine-executable outputs under strict latency and privacy constraints. Although constrained decoding provides formal gua

Cited by 0Scholar
2025

AtomNet: Designing Tiny Models from Operators Under Extreme MCU Constraints

AAAI 2025technical

Tiny machine learning (TinyML) has attracted heightened attention for its ability to provide low-cost and instantaneous performance on edge devices. Particularly, the commonly used microcontroller unit (MCU) imposes extreme constraints on peak memory (SRAM) and storage (Flash). Existing TinyML metho…

Cited by 0SourcePDFScholar
2025

DA-KD: Difficulty-Aware Knowledge Distillation for Efficient Large Language Models

ICML 2025poster

Although knowledge distillation (KD) is an effective approach to improve the performance of a smaller LLM (i.e., the student model) by transferring knowledge from a large LLM (i.e., the teacher model), it still suffers from high training cost. Existing LLM distillation methods ignore the difficulty…

Cited by 0SourcePDFScholar
2025

HarmoniCa: Harmonizing Training and Inference for Better Feature Caching in Diffusion Transformer Acceleration

ICML 2025poster

Diffusion Transformers (DiTs) excel in generative tasks but face practical deployment challenges due to high inference costs. Feature caching, which stores and retrieves redundant computations, offers the potential for acceleration. Existing learning-based caching, though adaptive, overlooks the imp…

2025

Hierachical Balance Packing: Towards Efficient Supervised Fine-tuning for Long-Context LLM

NeurIPS 2025poster

Training Long-Context Large Language Models (LLMs) is challenging, as hybrid training with long-context and short-context data often leads to workload imbalances. Existing works mainly use data packing to alleviate this issue, but fail to consider imbalanced attention computation and wasted communic…

Cited by 0SourcecodeScholar
2025

OmniBal: Towards Fast Instruction-Tuning for Vision-Language Models via Omniverse Computation Balance

ICML 2025poster

Vision-language instruction-tuning models have recently achieved significant performance improvements. In this work, we discover that large-scale 3D parallel training on those models leads to an imbalanced computation load across different devices. The vision and language parts are inherently hetero…

2025

Pre3: Enabling Deterministic Pushdown Automata for Faster Structured LLM Generation

ACL 2025long

Extensive LLM applications demand efficient structured generations, particularly for LR(1) grammars, to produce outputs in specified formats (e.g., JSON). Existing methods primarily parse LR(1) grammars into a pushdown automaton (PDA), leading to runtime execution overhead for context-dependent toke…

2025

Tool Playgrounds: A Comprehensive and Analyzable Benchmark for LLM Tool Invocation

ICASSP 2025accepted

The rapid advancement of large language models (LLMs) has paved the way for their use in solving real-world problems, which in turn has significantly driven the development of tool-assisted LLMs. This progress necessitates thorough evaluation methods. However, existing benchmarks typically only prov…

Cited by 0SourceScholar
2024

Compressing Large Language Models by Joint Sparsification and Quantization

ICML 2024poster

In this paper, we introduce a novel model compression technique named Joint Sparsification and Quantization (JSQ), explicitly tailored for large language models (LLMs). Traditional methods employ either sparsification or quantization individually to compress LLMs, leading to performance degradation…

2024

Fast and Controllable Post-training Sparsity: Learning Optimal Sparsity Allocation with Global Constraint in Minutes

AAAI 2024technical

Neural network sparsity has attracted many research interests due to its similarity to biological schemes and high energy efficiency. However, existing methods depend on long-time training or fine-tuning, which prevents large-scale applications. Recently, some works focusing on post-training sparsit…

2024

LLMC: Benchmarking Large Language Model Quantization with a Versatile Compression Toolkit

EMNLP 2024industry

Recent advancements in large language models (LLMs) are propelling us toward artificial general intelligence with their remarkable emergent abilities and reasoning capabilities. However, the substantial computational and memory requirements limit the widespread adoption. Quantization, a key compress…

2024

QLLM: Accurate and Efficient Low-Bitwidth Quantization for Large Language Models

ICLR 2024poster

Large Language Models (LLMs) have demonstrated unparalleled efficacy in natural language processing. However, their high computational demands and memory overheads hinder their broad deployment. To address this, two quantization strategies emerge, including Quantization-Aware Training (QAT) and Post…

2024

Selective Focus: Investigating Semantics Sensitivity in Post-training Quantization for Lane Detection

AAAI 2024technical

Lane detection (LD) plays a crucial role in enhancing the L2+ capabilities of autonomous driving, capturing widespread attention. The Post-Processing Quantization (PTQ) could facilitate the practical application of LD models, enabling fast speeds and limited memories without labeled data. However, p…

2024

TFMQ-DM: Temporal Feature Maintenance Quantization for Diffusion Models

CVPR 2024highlight

The Diffusion model a prevalent framework for image generation encounters significant challenges in terms of broad applicability due to its extended inference times and substantial memory requirements. Efficient Post-training Quantization (PTQ) is pivotal for addressing these issues in traditional m…

2023

Adaptive Contrastive Knowledge Distillation for BERT Compression

ACL 2023findings

In this paper, we propose a new knowledge distillation approach called adaptive contrastive knowledge distillation (ACKD) for BERT compression. Different from existing knowledge distillation methods for BERT that implicitly learn discriminative student features by mimicking the teacher features, we…

Cited by 10SourcePDFScholar
2023

Annealing-Based Label-Transfer Learning for Open World Object Detection

CVPR 2023poster

Open world object detection (OWOD) has attracted extensive attention due to its practicability in the real world. Previous OWOD works manually designed unknown-discover strategies to select unknown proposals from the background, suffering from uncertainties without appropriate priors. In this paper,…

2023

Exploring the Relationship Between Architectural Design and Adversarially Robust Generalization

CVPR 2023poster

Adversarial training has been demonstrated to be one of the most effective remedies for defending adversarial examples, yet it often suffers from the huge robustness generalization gap on unseen testing adversaries, deemed as the adversarially robust generalization problem. Despite the preliminary u…

Cited by 40SourcePDFScholar
2023

Lossy and Lossless (L2) Post-training Model Size Compression

ICCV 2023poster

Deep neural networks have delivered remarkable performance and have been widely used in various visual tasks. However, their huge sizes cause significant inconvenience for transmission and storage. Many previous studies have explored model size compression. However, these studies often approach vari…

Cited by 6PDFcodeScholar
2023

Outlier Suppression+: Accurate quantization of large language models by equivalent and effective shifting and scaling

EMNLP 2023long main

Post-training quantization (PTQ) of transformer language models faces significant challenges due to the existence of detrimental outliers in activations. We observe that these outliers are concentrated in specific channels and are asymmetric across channels. To address this issue, we propose the Out…

Cited by 0SourcecodeScholar
2022

Outlier Suppression: Pushing the Limit of Low-bit Transformer Language Models

NeurIPS 2022accept

Transformer architecture has become the fundamental element of the widespread natural language processing~(NLP) models. With the trends of large NLP models, the increasing memory and computation costs hinder their efficient deployment on resource-limited devices. Therefore, transformer quantization…

2022

QDrop: Randomly Dropping Quantization for Extremely Low-bit Post-Training Quantization

ICLR 2022poster

Recently, post-training quantization (PTQ) has driven much attention to produce efficient neural networks without long-time retraining. Despite the low cost, current PTQ works always fail under the extremely low-bit setting. In this study, we pioneeringly confirm that properly incorporating activati…

Cited by 192SourcePDFScholar
2021

A Free Lunch From ANN: Towards Efficient, Accurate Spiking Neural Networks Calibration

ICML 2021spotlight

Spiking Neural Network (SNN) has been recognized as one of the next generation of neural networks. Conventionally, SNN can be converted from a pre-trained ANN by only replacing the ReLU activation to spike activation while keeping the parameters intact. Perhaps surprisingly, in this work we show tha…

2021

BRECQ: Pushing the Limit of Post-Training Quantization by Block Reconstruction

ICLR 2021poster

We study the challenging task of neural network quantization without end-to-end retraining, called Post-training Quantization (PTQ). PTQ usually requires a small subset of training data but produces less powerful quantized models than Quantization-Aware Training (QAT). In this work, we propose a nov…

2021

Diversifying Sample Generation for Accurate Data-Free Quantization

CVPR 2021poster

Quantization has emerged as one of the most prevalent approaches to compress and accelerate neural networks. Recently, data-free quantization has been widely studied as a practical and promising solution. It synthesizes data for calibrating the quantized model according to the batch normalization (B…

Cited by 113PDFScholar
2021

MQBench: Towards Reproducible and Deployable Model Quantization Benchmark

NeurIPS 2021poster

Model quantization has emerged as an indispensable technique to accelerate deep learning inference. Although researchers continue to push the frontier of quantization algorithms, existing quantization work is often unreproducible and undeployable. This is because researchers do not choose consiste…

Cited by 62SourcecodeScholar
2021

MixMix: All You Need for Data-Free Compression Are Feature and Data Mixing

ICCV 2021poster

User data confidentiality protection is becoming a rising challenge in the present deep learning research. Without access to data, conventional data-driven model compression faces a higher risk of performance degradation. Recently, some works propose to generate images from a specific pretrained mod…

Cited by 40PDFScholar
2021

Once Quantization-Aware Training: High Performance Extremely Low-Bit Architecture Search

ICCV 2021poster

Quantization Neural Networks (QNN) have attracted a lot of attention due to their high efficiency. To enhance the quantization accuracy, prior works mainly focus on designing advanced quantization algorithms but still fail to achieve satisfactory results under the extremely low-bit case. In this wor…

Cited by 47PDFcodeScholar
2020

Forward and Backward Information Retention for Accurate Binary Neural Networks

CVPR 2020poster

Weight and activation binarization is an effective approach to deep neural network compression and can accelerate the inference by leveraging bitwise operations. Although many binarization methods have improved the accuracy of the model by minimizing the quantization error in forward propagation, th…

Cited by 431PDFcodeScholar
2020

Rotation Consistent Margin Loss for Efficient Low-Bit Face Recognition

CVPR 2020poster

In this paper, we consider the low-bit quantization problem of face recognition (FR) under the open-set protocol. Different from well explored low-bit quantization on closed-set image classification task, the open-set task is more sensitive to quantization errors (QEs). We redefine the QEs in angula…

Cited by 50PDFScholar
2020

Towards Unified INT8 Training for Convolutional Neural Network

CVPR 2020poster

Recently low-bit (e.g., 8-bit) network quantization has been extensively studied to accelerate the inference. Besides inference, low-bit training with quantized gradients can further bring more considerable acceleration, since the backward process is often computation-intensive. Unfortunately, the i…

Cited by 218PDFScholar
2019

Differentiable Soft Quantization: Bridging Full-Precision and Low-Bit Neural Networks

ICCV 2019poster

Hardware-friendly network quantization (e.g., binary/uniform quantization) can efficiently accelerate the inference and meanwhile reduce memory consumption of the deep neural networks, which is crucial for model deployment on resource-limited devices like mobile phones. However, due to the discreten…

Cited by 591PDFScholar