← Search

Shitong Shao

20 accepted papers

2026

Accelerating Diffusion Model Training under Minimal Budgets: A Condensation-Based Perspective

CVPR 2026

Diffusion models have achieved remarkable performance on a wide range of generative tasks, yet training them from scratch is notoriously resource-intensive, typically requiring millions of training images and many GPU days. Motivated by a data-centric view of this bottleneck, we adopt a condensation

Cited by 0SourcecodeScholar
2026

CRAFT: Aligning Diffusion Models with Fine-Tuning Is Easier Than You Think

CVPR 2026

Aligning Diffusion models has achieved remarkable breakthroughs in generating high-quality, human preference-aligned images. Existing techniques, such as supervised fine-tuning (SFT) and DPO-style preference optimization, have become principled tools for fine-tuning diffusion models. However, SFT re

Cited by 0SourceScholar
2026

Exploring Data-Free LoRA Transferability for Video Diffusion Models

ICML 2026poster

Video diffusion models leveraging step distillation or causal distillation have achieved remarkable performance. However, adapting existing LoRAs to these variants remains a critical challenge due to weight space mismatches. We observe that direct application leads to style degradation and structura…

Cited by 0SourceScholar
2026

Guidance Matters: Rethinking the Evaluation Pitfall for Text-to-Image Generation

ICLR 2026poster

Classifier-free guidance (CFG) has helped diffusion models achieve great conditional generation in various fields. Recently, more diffusion guidance methods have emerged with improved generation quality and human preference. However, can these emerging diffusion guidance methods really achieve solid…

Cited by 0SourceScholar
2026

Late-to-Early Training: LET LLMs Learn Earlier, So Faster and Better

ICLR 2026poster

As Large Language Models (LLMs) achieve remarkable empirical success through scaling model and data size, pretraining has become increasingly critical yet computationally prohibitive, hindering rapid development. Despite the availability of numerous pretrained LLMs developed at significant computati…

Cited by 0SourceScholar
2026

Lightning Unified Video Editing via In-Context Sparse Attention

ICML 2026poster

Video editing has evolved toward In-Context Learning (ICL) paradigms, yet the resulting quadratic attention costs create a critical computational bottleneck. In this work, we propose **I**n-context **S**parse **A**ttention (**ISA**), the first experimentally lossless sparse framework tailored for IC…

Cited by 0SourceScholar
2026

OD$^3$: Optimization-free Dataset Distillation for Object Detection

ICLR 2026poster

Training large neural networks on large-scale datasets requires substantial computational resources, particularly for dense prediction tasks such as object detection. Although dataset distillation (DD) has been proposed to alleviate these demands by synthesizing compact datasets from larger ones, mo…

Cited by 0SourcecodeScholar
2026

Optimizing Few-Step Generation with Adaptive Matching Distillation

ICML 2026poster

Distribution Matching Distillation (DMD) is a powerful acceleration paradigm, yet its stability is often compromised in **Forbidden Zones**—regions where the real teacher provides unreliable guidance while the fake teacher exerts insufficient repulsive force. In this work, we propose a unified optim…

Cited by 0SourceScholar
2026

PosterCraft: Rethinking High-Quality Aesthetic Poster Generation in a Unified Framework

ICLR 2026poster

Generating aesthetic posters is more challenging than simple design images: it requires not only precise text rendering but also the seamless integration of abstract artistic content, striking layouts, and overall stylistic harmony. To address this, we propose PosterCraft, a unified framework that a…

Cited by 0SourcecodeScholar
2025

DELT: A Simple Diversity-driven EarlyLate Training for Dataset Distillation

CVPR 2025poster

Recent advances in dataset distillation have led to solutions in two main directions. The conventional batch-to-batch matching mechanism is ideal for small-scale datasets and includes bi-level optimization methods on models and syntheses, such as FRePo, RCIG, and RaT-BPTT, as well as other methods l…

2025

Golden Noise for Diffusion Models: A Learning Framework

ICCV 2025poster

Text-to-image diffusion model is a popular paradigm that synthesizes personalized images by providing a text prompt and a random Gaussian noise. While people observe that some noises are "golden noises" that can achieve better text-image alignment and higher human preference than others, we still la…

2025

IV-mixed Sampler: Leveraging Image Diffusion Models for Enhanced Video Synthesis

ICLR 2025poster

Exploring suitable solutions to improve performance by increasing the computational cost of inference in visual diffusion models is a highly promising direction. Sufficient prior studies have demonstrated that correctly scaling up computation in the sampling process can successfully lead to improved…

2025

Zigzag Diffusion Sampling: Diffusion Models Can Self-Improve via Self-Reflection

ICLR 2025poster

Diffusion models, the most popular generative paradigm so far, can inject conditional information into the generation path to guide the latent towards desired directions. However, existing text-to-image diffusion models often fail to maintain high image quality and high prompt-image alignment for th…

2024

Auto-DAS: Automated Proxy Discovery for Training-free Distillation-aware Architecture Search

ECCV 2024poster

"Distillation-aware Architecture Search (DAS) seeks to discover the ideal student architecture that delivers superior performance by distilling knowledge from a given teacher model. Previous DAS methods involve time-consuming training-based search processes. Recently, the training-free DAS method (,…

2024

Diffusion Models are Certifiably Robust Classifiers

NeurIPS 2024poster

Generative learning, recognized for its effective modeling of data distributions, offers inherent advantages in handling out-of-distribution instances, especially for enhancing robustness to adversarial attacks. Among these, diffusion classifiers, utilizing powerful diffusion models, have demonstrat…

2024

Elucidating the Design Space of Dataset Condensation

NeurIPS 2024poster

Dataset condensation, a concept within $\textit{data-centric learning}$, aims to efficiently transfer critical attributes from an original dataset to a synthetic version, meanwhile maintaining both diversity and realism of syntheses. This approach can significantly improve model training efficiency…

2024

Generalized Large-Scale Data Condensation via Various Backbone and Statistical Matching

CVPR 2024highlight

The lightweight "local-match-global" matching introduced by SRe2L successfully creates a distilled dataset with comprehensive information on the full 224x224 ImageNet-1k. However this one-sided approach is limited to a particular backbone layer and statistics which limits the improvement of the gene…

2024

Rethinking Centered Kernel Alignment in Knowledge Distillation

IJCAI 2024poster

Knowledge distillation has emerged as a highly effective method for bridging the representation discrepancy between large-scale models and lightweight models. Prevalent approaches involve leveraging appropriate metrics to minimize the divergence or distance between the knowledge extracted from the t…

2023

Teaching What You Should Teach: A Data-Based Distillation Method

IJCAI 2023poster

In real teaching scenarios, an excellent teacher always teaches what he (or she) is good at but the student is not. This gives the student the best assistance in making up for his (or her) weaknesses and becoming a good one overall. Enlightened by this, we introduce the "Teaching what you Should Tea…

Cited by 4SourcePDFScholar