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.
☆30Jul 18, 2024Updated last year
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 Guidance in Diffusion Models with Soft Value-Based Decoding. For controlled generation in DNA, RNA, proteins, molecules (…☆37Jan 7, 2026Updated 2 months ago
- 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
- Code for the tutorial/review paper for RL-based-fine-tuniing. In this code, we especially focus on the design of biological sequences li…☆158Sep 15, 2024Updated last year
- [ICML 24] Robust Optimization in Protein Fitness Landscapes Using Reinforcement Learning in Latent Space☆16Aug 9, 2024Updated last year
- Subtask-Aware Visual Reward Learning from Segmented Demonstrations (ICLR 2025 accepted)☆18Apr 11, 2025Updated 11 months ago
- Managed hosting for WordPress and PHP on Cloudways • AdManaged hosting with the flexibility to host WordPress, Magento, Laravel, or PHP apps, on multiple cloud providers. Cloudways by DigitalOcean.
- Code for paper: "Fine-Tuning Discrete Diffusion Models via Reward Optimization with Applications to DNA and Protein Design"☆65May 10, 2025Updated 10 months ago
- Official implementation for Diffusion Alignment as Sampling (DAS), ICLR'25, Spotlight☆60Feb 12, 2025Updated last year
- Computational analysis of nucleic acids structures using graph neural networks☆15Mar 25, 2024Updated 2 years ago
- Official Code for Guided Trajectory Generation with Diffusion Models for Offline Model-based Optimization (NIPS 2024)☆22Aug 15, 2024Updated last year
- ☆32Jul 10, 2023Updated 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 last year
- Official code repository for the paper Exploring Chemical Space with Score-based Out-of-distribution Generation (ICML 2023)☆40May 19, 2024Updated last year
- AlignProp uses direct reward backpropogation for the alignment of large-scale text-to-image diffusion models. Our method is 25x more samp…☆314Nov 1, 2024Updated last year
- SimKO: Simple Pass@K Policy Optimization☆28Oct 24, 2025Updated 5 months ago
- DigitalOcean Gradient AI Platform • AdBuild production-ready AI agents using customizable tools or access multiple LLMs through a single endpoint. Create custom knowledge bases or connect external data.
- Tracking literature and additional online resources on transformers for sequential decision making including RL and beyond.☆50Dec 21, 2022Updated 3 years ago
- Official Code for Local Search GFlowNets (ICLR 2024 Spotlight)☆24Feb 27, 2025Updated last year
- Benchmarks for Model-Based Optimization☆96Apr 21, 2024Updated last year
- Generative models and other stuff too, maybe, perhaps even probably☆16Dec 12, 2015Updated 10 years ago
- Official repository for GFACS☆34May 17, 2024Updated last year
- An official 're'-implementation of Physics-induced graph neural network: An application to wind-farm power estimation (PGNN).☆29Jul 27, 2021Updated 4 years ago
- Official codebase for Exact Energy-Guided Diffusion Sampling via Contrastive Energy Prediction☆35Nov 3, 2023Updated 2 years ago
- Offline Reinforcement Learning Reading Group☆28Nov 22, 2022Updated 3 years ago
- Source code for "A Dense Reward View on Aligning Text-to-Image Diffusion with Preference" (ICML'24).☆40May 9, 2024Updated last year
- Wordpress hosting with auto-scaling on Cloudways • AdFully Managed hosting built for WordPress-powered businesses that need reliable, auto-scalable hosting. Cloudways SafeUpdates now available.
- Official repository for the paper "Peptide design through binding interface mimicry with PepMimic" accepted by Nature Biomedical Engineer…☆29Oct 20, 2025Updated 5 months ago
- Flexible modulation of sequence generation in the entorhinal-hippocampal system.☆13Apr 5, 2022Updated 3 years ago
- Hierarchical Hidden Markov Model☆11Dec 11, 2022Updated 3 years ago
- ☆11Feb 26, 2024Updated 2 years ago
- AbBiBench (Antibody Binding Benchmarking), a benchmarking framework for optimizing antibody binding affinity.☆36Oct 13, 2025Updated 5 months ago
- [NeurIPS 2021] World modelling and action learning using a contrastive formulation of the active inference framework, for reaching visual…☆14Jan 22, 2024Updated 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
- DDPO for finetuning diffusion models, implemented in PyTorch with LoRA support☆750Mar 22, 2024Updated 2 years ago
- Anti exploration in offline reinforcement learning☆11May 17, 2021Updated 4 years ago
- Managed Kubernetes at scale on DigitalOcean • AdDigitalOcean Kubernetes includes the control plane, bandwidth allowance, container registry, automatic updates, and more for free.
- ☆39Jul 2, 2023Updated 2 years ago
- Implement PPO algorithm on mujoco environment，such as Ant-v2, Humanoid-v2, Hopper-v2, Halfcheeth-v2.☆57Jun 30, 2020Updated 5 years ago
- ☆13Aug 14, 2022Updated 3 years ago
- Official codebase for Exact Energy-Guided Diffusion Sampling via Contrastive Energy Prediction （ICML 2023）☆54Aug 26, 2023Updated 2 years ago
- A benchmark for 3D biomolecular structure prediction models☆69May 11, 2025Updated 10 months ago
- Inference-Time Alignment in Protein Diffusion Models☆56Jan 7, 2026Updated 2 months ago
- A general framework for inference-time scaling and steering of diffusion models with arbitrary rewards.☆219Jun 26, 2025Updated 9 months ago