← Search

Guohao Dai

23 accepted papers

2026

Cache-to-Cache: Direct Semantic Communication Between Large Language Models

ICLR 2026poster

Multi-LLM systems harness the complementary strengths of diverse Large Language Models, achieving performance and efficiency gains unattainable by a single model. In existing designs, LLMs communicate through text, forcing internal representations to be transformed into output token sequences. This…

Cited by 0SourcecodeScholar
2026

MoSs: Mixture of Scales for Efficient High-Resolution Autoregressive Image Generation

AAAI 2026technical

Since next-scale prediction was introduced as a new paradigm for autoregressive image generation, it has attracted extensive research interest. By progressively increasing resolution in a draft-to-refinement process, next-scale prediction demonstrates great potential in both generation quality and e

Cited by 0SourcePDFScholar
2026

PM-KVQ: Progressive Mixed-precision KV Cache Quantization for Long-CoT LLMs

ICLR 2026poster

Recently, significant progress has been made in developing reasoning-capable Large Language Models (LLMs) through long Chain-of-Thought (CoT) techniques. However, this long-CoT reasoning process imposes substantial memory overhead due to the large Key-Value (KV) Cache memory overhead. Post-training…

Cited by 0SourcecodeScholar
2026

STEP: Warm-Started Visuomotor Policies with Spatiotemporal Consistency Prediction

ICML 2026poster

Diffusion policies have recently been as a powerful paradigm for visuomotor control in robotic manipulation due to their ability to model the distribution of action sequences and capture multimodality. However, iterative denoising leads to substantial inference latency, limiting control frequency in…

Cited by 0SourceScholar
2026

SpecDiff: Accelerating Diffusion Model Inference with Self-Speculation

AAAI 2026technical

Feature caching has recently emerged as a promising method for diffusion model acceleration. It effectively alleviates the inefficiency problem caused by high computational requirements by caching similar features in the inference process of the diffusion model. In this paper, we analyze existing fe

Cited by 0SourcePDFScholar
2026

SpecPrune-VLA: Accelerating Vision-Language-Action Models via Action-Aware Self-Speculative Pruning

ICML 2026poster

Pruning is a typical acceleration technique for compute-bound models by removing computation on unimportant values. Recently, it has been applied to accelerate Vision-Language-Action (VLA) model inference. However, existing acceleration methods focus on local information from the current action step…

Cited by 0SourceScholar
2026

Think-at-Hard: Selective Latent Iterations to Improve Reasoning Language Models

ICML 2026poster

Improving reasoning abilities of Large Language Models (LLMs), especially under parameter constraints, is crucial for real-world applications. Looped transformers address this by performing multiple latent iterations to refine each token beyond a single forward pass. However, we identify a latent ov…

Cited by 0SourceScholar
2025

Accelerating Auto-regressive Text-to-Image Generation with Training-free Speculative Jacobi Decoding

ICLR 2025poster

The current large auto-regressive models can generate high-quality, high-resolution images, but these models require hundreds or even thousands of steps of next-token prediction during inference, resulting in substantial time consumption. In existing studies, Jacobi decoding, an iterative parallel d…

2025

DLFR-Gen: Diffusion-based Video Generation with Dynamic Latent Frame Rate

ICCV 2025poster

Diffusion Transformer (DiT)-based generation models have achieved remarkable success in video generation. However, their inherent computational demands pose significant efficiency challenges. In this paper, we exploit the inherent temporal non-uniformity of real-world videos, and observe that videos…

Cited by 0SourcePDFScholar
2025

DeepGate4: Efficient and Effective Representation Learning for Circuit Design at Scale

ICLR 2025poster

Circuit representation learning has become pivotal in electronic design automation, enabling critical tasks such as testability analysis, logic reasoning, power estimation, and SAT solving. However, existing models face significant challenges in scaling to large circuits due to limitations like over…

2025

DiTFastAttnV2: Head-wise Attention Compression for Multi-Modality Diffusion Transformers

ICCV 2025poster

Text-to-image generation models, especially Multimodal Diffusion Transformers (MMDiT), have shown remarkable progress in generating high-quality images. However, these models often face significant computational bottlenecks, particularly in attention mechanisms, which hinder their scalability and ef…

2025

Distilled Decoding 2: One-step Sampling of Image Auto-regressive Models with Conditional Score Distillation

NeurIPS 2025poster

Image Auto-regressive (AR) models have emerged as a powerful paradigm of visual generative models. Despite their promising performance, they suffer from slow generation speed due to the large number of sampling steps required. Although Distilled Decoding 1 (DD1) was recently proposed to enable few-s…

Cited by 0SourcecodeScholar
2025

FrameFusion: Combining Similarity and Importance for Video Token Reduction on Large Vision Language Models

ICCV 2025poster

The increasing demand to process long and high-resolution videos significantly burdens Large Vision-Language Models (LVLMs) due to the enormous number of visual tokens. Existing token reduction methods primarily prune tokens based on importance metrics, such as cumulative attention scores. However,…

2025

Linear Combination of Saved Checkpoints Makes Consistency and Diffusion Models Better

ICLR 2025poster

Diffusion Models (DM) and Consistency Models (CM) are two types of popular generative models with good generation quality on various tasks. When training DM and CM, intermediate weight checkpoints are not fully utilized and only the last converged checkpoint is used. In this work, we find proper che…

2025

MBQ: Modality-Balanced Quantization for Large Vision-Language Models

CVPR 2025poster

Vision-Language Models (VLMs) have already enabled a variety of real-world applications. The large parameter size of VLMs brings large memory and computation overhead which poses significant challenges for deployment. Post-Training Quantization (PTQ) is an effective technique to reduce the memory an…

2025

R2R: Efficiently Navigating Divergent Reasoning Paths with Small-Large Model Token Routing

NeurIPS 2025poster

Large Language Models (LLMs) achieve impressive reasoning capabilities at the cost of substantial inference overhead, posing substantial deployment challenges. Although distilled Small Language Models (SLMs) significantly enhance efficiency, their performance suffers as they fail to follow LLMs' rea…

Cited by 0SourcecodeScholar
2025

ViDiT-Q: Efficient and Accurate Quantization of Diffusion Transformers for Image and Video Generation

ICLR 2025poster

Diffusion transformers have demonstrated remarkable performance in visual generation tasks, such as generating realistic images or videos based on textual instructions. However, larger model sizes and multi-frame processing for video generation lead to increased computational and memory costs, posin…

2024

Can LLMs Learn by Teaching for Better Reasoning? A Preliminary Study

NeurIPS 2024poster

Teaching to improve student models (e.g., knowledge distillation) is an extensively studied methodology in LLMs. However, in human education, teaching enhances not only the students but also the teachers by fostering more rigorous and clearer reasoning, as well as deeper knowledge building. We ask:…

2024

DiTFastAttn: Attention Compression for Diffusion Transformer Models

NeurIPS 2024poster

Diffusion Transformers (DiT) excel at image and video generation but face computational challenges due to the quadratic complexity of self-attention operators. We propose DiTFastAttn, a post-training compression method to alleviate the computational bottleneck of DiT. We identify three key redundanc…

Cited by 19SourcePDFScholar
2024

Evaluating Quantized Large Language Models

ICML 2024poster

Post-training quantization (PTQ) has emerged as a promising technique to reduce the cost of large language models (LLMs). Specifically, PTQ can effectively mitigate memory consumption and reduce computational overhead in LLMs. To meet the requirements of both high efficiency and performance across d…

2024

FlashEval: Towards Fast and Accurate Evaluation of Text-to-image Diffusion Generative Models

CVPR 2024poster

In recent years there has been significant progress in the development of text-to-image generative models. Evaluating the quality of the generative models is one essential step in the development process. Unfortunately the evaluation process could consume a significant amount of computational resour…

2024

MixDQ: Memory-Efficient Few-Step Text-to-Image Diffusion Models with Metric-Decoupled Mixed Precision Quantization

ECCV 2024poster

"Few-step diffusion models, which enable high-quality text-to-image generation with only a few denoising steps, have substantially reduced inference time. However, considerable memory consumption (5-10GB) still poses limitations for practical deployment on mobile devices. Post-Training Quantization…

2023

Ada3D : Exploiting the Spatial Redundancy with Adaptive Inference for Efficient 3D Object Detection

ICCV 2023poster

Voxel-based methods have achieved state-of-the-art performance for 3D object detection in autonomous driving. However, their significant computational and memory costs pose a challenge for their application to resource-constrained vehicles. One reason for this high resource consumption is the presen…

Cited by 17PDFcodeScholar