4 papers
SpecForge: A Flexible and Efficient Open-Source Training Framework for Speculative Decoding
Shenggui Li, Chao Wang, Yikai Zhu +14
Large language models incur high inference latency due to sequential autoregressive decoding. Speculative decoding alleviates this bottleneck by using a lightweight draft model to…
Integrating Reweighted Least Squares with Plug-and-Play Diffusion Priors for Noisy Image Restoration
Ji Li, Chao Wang
Existing plug-and-play image restoration methods typically employ off-the-shelf Gaussian denoisers as proximal operators within classical optimization frameworks based on variable…
Image Restoration via Primal Dual Hybrid Gradient and Flow Generative Model
Ji Li, Chao Wang
Regularized optimization has been a classical approach to solving imaging inverse problems, where the regularization term enforces desirable properties of the unknown image. Recent…
Efficient Diffusion Posterior Sampling for Noisy Inverse Problems
Ji Li, Chao Wang
The pretrained diffusion model as a strong prior has been leveraged to address inverse problems in a zero-shot manner without task-specific retraining. Different from the unconditi…