Diffusers documentation

Perencanaan berbasis nilai

You are viewing main version, which requires installation from source. If you'd like regular pip install, checkout the latest stable version (v0.42.1).
Hugging Face's logo
Join the Hugging Face community

and get access to the augmented documentation experience

to get started

Perencanaan berbasis nilai

🧪 Ini adalah pipeline eksperimental untuk reinforcement learning!

Pipeline ini didasarkan pada makalah Planning with Diffusion for Flexible Behavior Synthesis oleh Michael Janner, Yilun Du, Joshua B. Tenenbaum, Sergey Levine.

Abstrak dari makalah tersebut adalah:

Metode pembelajaran penguatan berbasis model sering kali hanya menggunakan pembelajaran untuk tujuan memperkirakan model dinamika yang mendekati, sambil menyerahkan sebagian besar pekerjaan pengambilan keputusan ke pengoptimal lintasan klasik. Meskipun secara konseptual sederhana, kombinasi ini memiliki sejumlah kekurangan empiris, yang menunjukkan bahwa model yang dipelajari mungkin tidak cocok untuk optimasi lintasan standar. Dalam makalah ini, kami mempertimbangkan bagaimana jika sebagian besar dari pipeline optimasi lintasan dimasukkan ke dalam masalah pemodelan, sehingga pengambilan sampel dari model dan perencanaan dengannya menjadi hampir identik. Inti dari pendekatan teknis kami terletak pada model probabilistik difusi yang merencanakan dengan cara mendenoising lintasan secara iteratif. Kami menunjukkan bagaimana pengambilan sampel yang dipandu klasifikator dan inpainting gambar dapat ditafsirkan ulang sebagai strategi perencanaan yang koheren, mengeksplorasi sifat-sifat yang tidak biasa dan berguna dari metode perencanaan berbasis difusi, serta mendemonstrasikan efektivitas kerangka kerja kami dalam pengaturan kontrol yang menekankan pengambilan keputusan jangka panjang dan fleksibilitas saat pengujian.

Anda dapat menemukan informasi tambahan tentang model di halaman proyek, kodebase asli, atau mencobanya di notebook demo.

Skrip untuk menjalankan model tersedia di sini.

Pastikan untuk melihat panduan Schedulers untuk mempelajari cara mengeksplorasi tradeoff antara kecepatan dan kualitas scheduler, serta lihat bagian reuse components across pipelines untuk mempelajari cara memuat komponen yang sama secara efisien ke dalam beberapa pipeline.

ValueGuidedRLPipeline

kelas diffusers.experimental.ValueGuidedRLPipeline

< >

( value_function: UNet1DModelunet: UNet1DModelscheduler: DDPMSchedulerenv )

Parameter

  • value_function (UNet1DModel) — UNet khusus untuk fine-tuning lintasan berdasarkan reward.
  • unet (UNet1DModel) — Arsitektur UNet untuk menghilangkan noise dari trajektori yang telah dienkod.
  • scheduler (SchedulerMixin) — Scheduler yang digunakan bersama unet untuk menghilangkan noise dari trajektori yang telah dienkod. Nilai default untuk aplikasi ini adalah DDPMScheduler.
  • env () — Lingkungan yang mengikuti API OpenAI gym untuk bertindak. Saat ini, hanya Hopper yang memiliki model terlatih sebelumnya.

Pipeline untuk pengambilan sampel yang dipandu nilai dari model difusi yang dilatih untuk memprediksi urutan keadaan.

Model ini mewarisi dari DiffusionPipeline. Periksa dokumentasi kelas induk untuk metode generik yang diimplementasikan untuk semua pipeline (mengunduh, menyimpan, menjalankan pada perangkat tertentu, dll.).

Update on GitHub