← Search

Trung Tuan Dao

4 accepted papers

2025

Improved Training Technique for Shortcut Models

NeurIPS 2025poster

Shortcut models represent a promising, non-adversarial paradigm for generative modeling, uniquely supporting one-step, few-step, and multi-step sampling from a single trained network. However, their widespread adoption has been stymied by critical performance bottlenecks. This paper tackles the five…

Cited by 0SourceScholar
2025

Self-Corrected Flow Distillation for Consistent One-Step and Few-Step Image Generation

AAAI 2025technical

Flow matching has emerged as a promising framework for training generative models, demonstrating impressive empirical performance while offering relative ease of training compared to diffusion-based models. However, this method still requires numerous function evaluations in the sampling process. To…

2024

DiMSUM: Diffusion Mamba - A Scalable and Unified Spatial-Frequency Method for Image Generation

NeurIPS 2024poster

We introduce a novel state-space architecture for diffusion models, effectively harnessing spatial and frequency information to enhance the inductive bias towards local features in input images for image generation tasks. While state-space networks, including Mamba, a revolutionary advancement in re…