SEIKO is a novel reinforcement learning method to efficiently fine-tune diffusion models in an online setting. Our methods outperform all baselines (PPO, classifier-based guidance, direct reward backpropagation) for fine-tuning Stable Diffusion.
☆31Jul 18, 2024Updated 2 years ago
Alternatives and similar repositories for SEIKO
Users that are interested in SEIKO are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- Derivative-Free, Training-Free, Guidance in Diffusion Models☆16Sep 30, 2024Updated 2 years ago
- [ICML 2024] Code for the paper "Confronting Reward Overoptimization for Diffusion Models: A Perspective of Inductive and Primacy Biases"☆38Jul 12, 2024Updated 2 years ago
- Code for paper: "Fine-Tuning Discrete Diffusion Models via Reward Optimization with Applications to DNA and Protein Design"☆78May 10, 2025Updated last year
- [NeurIPS 2023] Reward-Directed Conditional Diffusion: Provable Distribution Estimation and Reward Improvement☆18Oct 30, 2024Updated last year
- Code for the tutorial/review paper for RL-based-fine-tuniing. In this code, we especially focus on the design of biological sequences li…☆159Sep 15, 2024Updated 2 years ago
- Deploy to Railway using AI coding agents - Free Credits Offer • AdUse Claude Code, Codex, OpenCode, and more. Autonomous software development now has the infrastructure to match with Railway.
- Code for the paper "Molecule Design by Latent Space Energy-based Modeling and Gradual Distribution Shifting" in UAI 2023☆15Nov 15, 2023Updated 2 years ago
- [ICML 24] Robust Optimization in Protein Fitness Landscapes Using Reinforcement Learning in Latent Space☆18Aug 9, 2024Updated 2 years ago
- AlignProp uses direct reward backpropogation for the alignment of large-scale text-to-image diffusion models. Our method is 25x more samp…☆329Nov 1, 2024Updated last year
- Inference-Time Alignment in Protein Diffusion Models☆59Jan 7, 2026Updated 9 months ago
- Computational analysis of nucleic acids structures using graph neural networks☆16Mar 25, 2024Updated 2 years ago
- Official code repository for the paper Exploring Chemical Space with Score-based Out-of-distribution Generation (ICML 2023)☆42May 19, 2024Updated 2 years ago
- The official implementation of Diffusion-KTO: Aligning Diffusion Models by Optimizing Human Utility☆67Aug 16, 2025Updated last year
- ☆34Jul 10, 2023Updated 3 years ago
- Code base for paper: Reparameterized Policy Learning for Multimodal Trajectory Optimization☆28Jul 19, 2023Updated 3 years ago
- Wordpress hosting with auto-scaling - Free Trial Offer • AdFully Managed hosting for WordPress and WooCommerce businesses that need reliable, auto-scalable performance. Cloudways SafeUpdates now available.
- Subtask-Aware Visual Reward Learning from Segmented Demonstrations (ICLR 2025 accepted)☆20Apr 11, 2025Updated last year
- ☆41Oct 8, 2026Updated last week
- DDPO for finetuning diffusion models, implemented in PyTorch with LoRA support☆775Mar 22, 2024Updated 2 years ago
- Hierarchical Hidden Markov Model☆12Dec 11, 2022Updated 3 years ago
- Official Code for Local Search GFlowNets (ICLR 2024 Spotlight)☆26Feb 27, 2025Updated last year
- Official Code for Guided Trajectory Generation with Diffusion Models for Offline Model-based Optimization (NIPS 2024)☆23Aug 15, 2024Updated 2 years ago
- A general framework for inference-time scaling and steering of diffusion models with arbitrary rewards.☆237Jun 26, 2025Updated last year
- Benchmarks for Model-Based Optimization☆96Apr 21, 2024Updated 2 years ago
- Official codebase for Exact Energy-Guided Diffusion Sampling via Contrastive Energy Prediction☆36Nov 3, 2023Updated 2 years ago
- Managed Kubernetes at scale on DigitalOcean • AdDigitalOcean Kubernetes includes the control plane, bandwidth allowance, container registry, automatic updates, and more for free.
- Code of the paper "Universal Morphology Control via Contextual Modulation" at ICML 2023☆17Aug 3, 2023Updated 3 years ago
- An official 're'-implementation of Physics-induced graph neural network: An application to wind-farm power estimation (PGNN).☆29Jul 27, 2021Updated 5 years ago
- Official codebase for GTA: Generative Trajectory Augmentation with Guidance for Offline Reinforcement Learning.☆32Nov 12, 2024Updated last year
- ☆63Apr 7, 2026Updated 6 months ago
- Reverse Engineering Imperceptible Backdoor Attacks on Deep Neural Networks for Detection and Training Set Cleansing☆15Feb 18, 2021Updated 5 years ago
- ☆15May 18, 2025Updated last year
- Source code for "A Dense Reward View on Aligning Text-to-Image Diffusion with Preference" (ICML'24).☆38May 9, 2024Updated 2 years ago
- Diffusion Reinforcement Learning Library☆192Feb 13, 2024Updated 2 years ago
- Output-Constrained Regression Trees (OCRT)☆14Sep 14, 2026Updated last month
- Deploy on Railway without the complexity - Free Credits Offer • AdConnect your repo and Railway handles the rest with instant previews. Quickly provision container image services, databases, and storage volumes.
- Official repository for GFACS☆38May 17, 2024Updated 2 years ago
- ☆38Jul 2, 2023Updated 3 years ago
- Course Website for "AI618: Generative Model and Unsupervised Learning"☆36May 23, 2023Updated 3 years ago
- The official implementation of MaskGRPO: Consolidating Reinforcement Learning for Multimodal Discrete Diffusion Models. (ICLR 2026, arxiv…☆19Jan 27, 2026Updated 8 months ago
- ☆22Oct 20, 2023Updated 2 years ago
- Contains implementation of the DoubIL and ResiduIL algorithms from the ICML '22 paper Causal Imitation Learning under Temporally Correlat…☆11Dec 9, 2022Updated 3 years ago
- Official codebase for Exact Energy-Guided Diffusion Sampling via Contrastive Energy Prediction （ICML 2023）☆55Aug 26, 2023Updated 3 years ago