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
- Derivative-Free Guidance in Diffusion Models with Soft Value-Based Decoding. For controlled generation in DNA, RNA, proteins, molecules (…☆41Jan 7, 2026Updated 10 months ago
- Code for paper: "Fine-Tuning Discrete Diffusion Models via Reward Optimization with Applications to DNA and Protein Design"☆78May 10, 2025Updated 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…☆160Sep 15, 2024Updated 2 years ago
- GPU virtual machines on DigitalOcean Gradient AI • AdGet to production fast with high-performance AMD and NVIDIA GPUs you can spin up in seconds. The definition of operational simplicity.
- 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
- Inference-Time Alignment in Protein Diffusion Models☆59Jan 7, 2026Updated 10 months ago
- AlignProp uses direct reward backpropogation for the alignment of large-scale text-to-image diffusion models. Our method is 25x more samp…☆330Nov 1, 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
- [MLSys'23] Exploiting Hardware Utilization and Adaptive Dataflow for Efficient Sparse Convolution in 3D Point Clouds.☆13Jun 22, 2023Updated 3 years ago
- The collection of my research papers' illustrations.☆21Oct 19, 2026Updated 2 weeks ago
- ☆63Apr 7, 2026Updated 6 months ago
- Code for the paper "Training Diffusion Models with Reinforcement Learning"☆579Jul 5, 2023Updated 3 years ago
- DDPO for finetuning diffusion models, implemented in PyTorch with LoRA support☆775Mar 22, 2024Updated 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.
- ☆34Jul 10, 2023Updated 3 years ago
- Subtask-Aware Visual Reward Learning from Segmented Demonstrations (ICLR 2025 accepted)☆20Apr 11, 2025Updated last year
- Code base for paper: Reparameterized Policy Learning for Multimodal Trajectory Optimization☆28Jul 19, 2023Updated 3 years ago
- ☆10Jan 27, 2026Updated 9 months ago
- A general framework for inference-time scaling and steering of diffusion models with arbitrary rewards.☆237Jun 26, 2025Updated last year
- Official Code for Local Search GFlowNets (ICLR 2024 Spotlight)☆26Feb 27, 2025Updated last year
- Code for QuantSparse: Comprehensively Compressing Video Diffusion Transformer with Model Quantization and Attention Sparsification☆12Oct 8, 2025Updated last year
- Evolutionary Algorithm with Diffusion Models for Protein Design☆31Jan 7, 2026Updated 10 months ago
- Tracking literature and additional online resources on transformers for sequential decision making including RL and beyond.☆54Dec 21, 2022Updated 3 years ago
- 1-Click AI Models by DigitalOcean Gradient • AdDeploy popular AI models on DigitalOcean Gradient GPU virtual machines with just a single click. Zero configuration with optimized deployments.
- DNN for spatio-temporal process interpolation and forecasting.☆12Jun 10, 2025Updated last year
- Generative models and other stuff too, maybe, perhaps even probably☆16Dec 12, 2015Updated 10 years ago
- ☆25Apr 16, 2025Updated last year
- kinematic inference in hierarchical models, intrinsic and extrinsic intentions, human kinematics, obstacle avoidance☆13Nov 7, 2024Updated 2 years ago
- Benchmarks for Model-Based Optimization☆96Apr 21, 2024Updated 2 years ago
- An official 're'-implementation of Physics-induced graph neural network: An application to wind-farm power estimation (PGNN).☆30Jul 27, 2021Updated 5 years ago
- Official codebase for GTA: Generative Trajectory Augmentation with Guidance for Offline Reinforcement Learning.☆32Nov 12, 2024Updated last year
- Official codebase for Exact Energy-Guided Diffusion Sampling via Contrastive Energy Prediction☆36Nov 3, 2023Updated 3 years ago
- Faster Inference of Flow-Based Generative Models via Improved Data-Noise Coupling. In ICLR, 2025.☆15Mar 4, 2025Updated last year
- 1-Click AI Models by DigitalOcean Gradient • AdDeploy popular AI models on DigitalOcean Gradient GPU virtual machines with just a single click. Zero configuration with optimized deployments.
- wip☆11Jan 9, 2026Updated 9 months ago
- Variational Reinforcement Learning☆18Jul 25, 2024Updated 2 years ago
- Diffusion Reinforcement Learning Library☆192Feb 13, 2024Updated 2 years ago
- Allosteric design of proteins using allosteric pathways, stability calculations, and normal mode analysis☆12Jan 11, 2026Updated 9 months ago
- [CVPR 2025] Accelerating Diffusion Transformer via Increment-Calibrated with Channel-Aware Singular Value Decomposition☆14May 14, 2025Updated last year
- dynamic planning, hybrid models, hierarchical active inference, tool use☆15Jun 13, 2025Updated last year
- Flexible modulation of sequence generation in the entorhinal-hippocampal system.☆13Apr 5, 2022Updated 4 years ago