SAGE: Semantic-Aware Shared Sampling for Efficient Diffusion

Read full text
Authors: Haoran Zhao, Tong Bai, Lei Huang, Xiaoyu Liang
Cited by: 0
References: 21

Abstract

Diffusion models manifest evident benefits across diverse domains, yet their high sampling cost, requiring dozens of sequential model evaluations, remains a major limitation. Prior efforts mainly accelerate sampling via optimized solvers or distillation, which treat each query independently. In contrast, we reduce total number of steps by sharing early-stage sampling across semantically similar queries. To enable such efficiency gains without sacrificing quality, we propose SAGE, a semantic-aware shared sampling framework that integrates a shared sampling scheme for efficiency and a tailored training strategy for quality preservation. Extensive experiments show that SAGE reduces sampling cost by 25.5%, while improving generation quality with 5.0% lower FID, 5.4% higher CLIP, and 160% higher diversity over baselines.

Fields of Study & Subjects

Music and Audio Processing Advanced Clustering Algorithms Research AI in cancer detection

Publication details

Journal / source
ICASSP 2026 - 2026 IEEE International Conference on Acoustics, Speech and Signal Processing (ICASSP)
Publisher
IEEE
Pages
8762–8766
Publication date
21 April 2026
Language
EN