D-OPSD  by vvvvvjdy

On-policy self-distillation for diffusion model tuning

Created 3 months ago
283 stars

Top 92.1% on SourcePulse

GitHubView on GitHub
Project Summary

Summary

D-OPSD addresses the challenge of continuously tuning step-distilled diffusion models, particularly those with LLM/VLM encoders. It offers a novel on-policy self-distillation framework for researchers and practitioners working with text-to-image diffusion models, enabling them to adapt models to new concepts, styles, or domains while preserving core capabilities and few-step inference efficiency.

How It Works

The core innovation lies in an on-policy self-distillation approach that leverages an emergent property of diffusion models with LLM/VLM encoders. D-OPSD assigns the same model two distinct roles within different contexts, facilitating supervised tuning on the model's own generated outputs. This method bypasses the need for external reward functions or additional modules, allowing for efficient adaptation and concept learning.

Quick Start & Requirements

  • Installation: Clone the repository, create and activate a Conda environment (conda create -n dopsd python=3.12 -y, conda activate dopsd), and install dependencies (pip install -r requirements.txt).
  • Prerequisites: Python 3.12. Specific training and evaluation guidance are provided in subdirectories like z-image-turbo_self-distill-vlm.
  • Links: Official Project Page: https://vvvvvjdy.github.io/d-opsd/

Highlighted Details

  • Enables continuous tuning of step-distilled diffusion models by exploiting emergent LLM/VLM encoder properties.
  • Facilitates learning new concepts, styles, and domain preferences while retaining original few-step inference capability and prior knowledge.
  • Successfully adapts models to target domains (e.g., anime) and learns new concepts from minimal data (few image-text pairs) via LoRA training.

Maintenance & Community

  • The project is associated with a recent arXiv preprint (2026). No specific community links (Discord, Slack) or contributor details are provided in the README.

Licensing & Compatibility

  • The README does not specify a license. Compatibility for commercial use or closed-source linking is undetermined without a clear license.

Limitations & Caveats

  • The project is presented as a research artifact with a recent arXiv publication; its production readiness or long-term maintenance status is not detailed. The absence of a specified license poses a significant adoption blocker for commercial or widespread use.
Health Check
Last Commit

1 month ago

Responsiveness

Inactive

Pull Requests (30d)
0
Issues (30d)
0
Star History
39 stars in the last 30 days

Explore Similar Projects

Starred by Junyang Lin Junyang Lin(Core Maintainer at Alibaba Qwen), Shizhe Diao Shizhe Diao(Author of LMFlow; Research Scientist at NVIDIA), and
1 more.

LMaaS-Papers by txsun1997

0%
545
Curated list of LMaaS research papers
Created 4 years ago
Updated 2 years ago
Starred by Eric Zhang Eric Zhang(Founding Engineer at Modal), Yineng Zhang Yineng Zhang(Inference Lead at SGLang; Research Scientist at Together AI), and
3 more.

tunix by google

0.3%
2k
JAX-native library for efficient LLM post-training
Created 1 year ago
Updated 20 hours ago
Feedback? Help us improve.