Diffusers documentation

Lập kế hoạch dựa trên giá trị

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

Lập kế hoạch dựa trên giá trị

🧪 Đây là một pipeline thử nghiệm cho học tăng cường!

Pipeline này dựa trên bài báo Planning with Diffusion for Flexible Behavior Synthesis của Michael Janner, Yilun Du, Joshua B. Tenenbaum, Sergey Levine.

Tóm tắt từ bài báo là:

Các phương pháp học tăng cường dựa trên mô hình thường chỉ sử dụng quá trình học để ước lượng một mô hình động lực học gần đúng, và giao phần còn lại của công việc ra quyết định cho các bộ tối ưu hóa quỹ đạo cổ điển. Mặc dù đơn giản về mặt khái niệm, sự kết hợp này có một số điểm yếu thực nghiệm, cho thấy các mô hình học được có thể không phù hợp với tối ưu hóa quỹ đạo tiêu chuẩn. Trong bài báo này, chúng tôi xem xét việc tích hợp càng nhiều càng tốt của quy trình tối ưu hóa quỹ đạo vào bài toán mô hình hóa, sao cho việc lấy mẫu từ mô hình và lập kế hoạch bằng mô hình trở nên gần như giống hệt nhau. Nền tảng của cách tiếp cận kỹ thuật của chúng tôi nằm ở một mô hình xác suất khuếch tán, lập kế hoạch bằng cách khử nhiễu các quỹ đạo một cách lặp đi lặp lại. Chúng tôi chỉ ra cách lấy mẫu có hướng dẫn bằng bộ phân loại và lấp đầy hình ảnh (image inpainting) có thể được diễn giải lại như các chiến lược lập kế hoạch nhất quán, khám phá các đặc tính bất thường và hữu ích của các phương pháp lập kế hoạch dựa trên khuếch tán, và chứng minh hiệu quả của khung làm việc của chúng tôi trong các thiết lập điều khiển nhấn mạnh vào ra quyết định tầm xa và tính linh hoạt tại thời điểm chạy.

Bạn có thể tìm thêm thông tin về mô hình trên trang dự án, mã nguồn gốc, hoặc thử nghiệm trong một notebook demo.

Script để chạy model có sẵn tại đây.

Hãy chắc chắn xem qua hướng dẫn về Schedulers để tìm hiểu cách khám phá sự đánh đổi giữa tốc độ và chất lượng của scheduler, và xem phần tái sử dụng các thành phần giữa các pipeline để tìm hiểu cách nạp các thành phần giống nhau vào nhiều pipeline một cách hiệu quả.

ValueGuidedRLPipeline

lớp diffusers.experimental.ValueGuidedRLPipeline

< >

( value_function: UNet1DModelunet: UNet1DModelscheduler: DDPMSchedulerenv )

Tham số

  • value_function (UNet1DModel) — UNet chuyên dụng để tinh chỉnh quỹ đạo dựa trên phần thưởng.
  • unet (UNet1DModel) — Kiến trúc UNet để khử nhiễu các quỹ đạo đã được mã hóa.
  • scheduler (SchedulerMixin) — Một scheduler được sử dụng kết hợp với unet để khử nhiễu các quỹ đạo đã mã hóa. Mặc định cho ứng dụng này là DDPMScheduler.
  • env () — Một môi trường tuân theo API OpenAI gym để thực hiện hành động. Hiện tại, chỉ có Hopper có các mô hình được huấn luyện trước.

Pipeline cho việc lấy mẫu có hướng dẫn giá trị từ một mô hình khuếch tán được huấn luyện để dự đoán các chuỗi trạng thái.

Mô hình này kế thừa từ DiffusionPipeline. Xem tài liệu của lớp cha để biết các phương thức chung được triển khai cho tất cả các pipeline (tải xuống, lưu, chạy trên một thiết bị cụ thể, v.v.).

Update on GitHub