Paper Reading AI Learner

Planned Diffusion

2025-10-20 20:27:48
Daniel Israel, Tian Jin, Ellie Cheng, Guy Van den Broeck, Aditya Grover, Suvinay Subramanian, Michael Carbin

Abstract

A central challenge in large language model inference is the trade-off between generation speed and output quality. Autoregressive models produce high-quality text but generate tokens sequentially. Diffusion models can generate tokens in parallel but often need many iterations to match the same quality. We propose planned diffusion, a hybrid method that combines the strengths of both paradigms. Planned diffusion works in two stages: first, the model creates a short autoregressive plan that breaks the output into smaller, independent spans. Second, the model generates these spans simultaneously using diffusion. This approach expands the speed-quality Pareto frontier and provides a practical path to faster, high-quality text generation. On AlpacaEval, a suite of 805 instruction-following prompts, planned diffusion achieves Pareto-optimal trade-off between quality and latency, achieving 1.27x to 1.81x speedup over autoregressive generation with only 0.87\% to 5.4\% drop in win rate, respectively. Our sensitivity analysis shows that the planning mechanism of planned diffusion is minimal and reliable, and simple runtime knobs exist to provide flexible control of the quality-latency trade-off.

Abstract (translated)

大型语言模型推理中一个核心挑战是在生成速度和输出质量之间的权衡。自回归模型能够产生高质量的文本,但其生成令牌的过程是顺序进行的;而扩散模型可以并行生成令牌,但却常常需要多次迭代才能达到相同的质量水平。我们提出了一种名为“计划式扩散”的混合方法,该方法结合了这两种范式的优点。 计划式扩散的工作流程分为两个阶段:首先,模型创建一个简短的自回归计划,将输出分解为较小且独立的小片段;其次,在这一规划的基础上,模型同时生成这些小片段。这种方法扩展了速度与质量之间的帕累托前沿,并提供了一条实现更快、高质量文本生成的实际路径。 在包含805个指令跟随提示的AlpacaEval测试集上,计划式扩散实现了质量和延迟之间的帕累托最优权衡,在自回归生成的基础上分别取得了1.27倍到1.81倍的速度提升,同时仅以质量胜率减少0.87%至5.4%为代价。我们的敏感性分析表明,计划式扩散的规划机制是微小且可靠的,并存在一些简单的运行时控制参数来灵活地调控质量和延迟之间的权衡。 通过这种方法,大型语言模型能够更高效地生成高质量文本,在确保输出质量的同时显著提高生成速度。

URL

https://arxiv.org/abs/2510.18087

PDF

https://arxiv.org/pdf/2510.18087.pdf


Tags
3D Action Action_Localization Action_Recognition Activity Adversarial Agent Attention Autonomous Bert Boundary_Detection Caption Chat Classification CNN Compressive_Sensing Contour Contrastive_Learning Deep_Learning Denoising Detection Dialog Diffusion Drone Dynamic_Memory_Network Edge_Detection Embedding Embodied Emotion Enhancement Face Face_Detection Face_Recognition Facial_Landmark Few-Shot Gait_Recognition GAN Gaze_Estimation Gesture Gradient_Descent Handwriting Human_Parsing Image_Caption Image_Classification Image_Compression Image_Enhancement Image_Generation Image_Matting Image_Retrieval Inference Inpainting Intelligent_Chip Knowledge Knowledge_Graph Language_Model LLM Matching Medical Memory_Networks Multi_Modal Multi_Task NAS NMT Object_Detection Object_Tracking OCR Ontology Optical_Character Optical_Flow Optimization Person_Re-identification Point_Cloud Portrait_Generation Pose Pose_Estimation Prediction QA Quantitative Quantitative_Finance Quantization Re-identification Recognition Recommendation Reconstruction Regularization Reinforcement_Learning Relation Relation_Extraction Represenation Represenation_Learning Restoration Review RNN Robot Salient Scene_Classification Scene_Generation Scene_Parsing Scene_Text Segmentation Self-Supervised Semantic_Instance_Segmentation Semantic_Segmentation Semi_Global Semi_Supervised Sence_graph Sentiment Sentiment_Classification Sketch SLAM Sparse Speech Speech_Recognition Style_Transfer Summarization Super_Resolution Surveillance Survey Text_Classification Text_Generation Time_Series Tracking Transfer_Learning Transformer Unsupervised Video_Caption Video_Classification Video_Indexing Video_Prediction Video_Retrieval Visual_Relation VQA Weakly_Supervised Zero-Shot