Mabilis na High-Resolution Image Synthesis na may Latent Adversarial Diffusion Distillation

Mabilis na High-Resolution Image Synthesis na may Latent Adversarial Diffusion Distillation

Arxiv Link

ika-18 ng Marso, 2024

Ang "Fast High-Resolution Image Synthesis na may Latent Adversarial Diffusion Distillation" ay nagpapakita ng isang nobelang diskarte sa distillation na kilala bilang Latent Adversarial Diffusion Distillation (LADD). Ang diskarte na ito ay idinisenyo upang tugunan ang mga limitasyon ng umiiral na mga modelo ng pagsasabog, partikular na ang hamon ng mabagal na bilis ng hinuha, na humahadlang sa mga real-time na aplikasyon. Ang LADD ay nagbibigay-daan sa high-resolution, multi-aspect ratio na synthesis ng larawan sa pamamagitan ng mahusay na paglilinis ng malalaking latent diffusion models (LDMs), na makabuluhang pinapasimple ang proseso ng pagsasanay at pagpapahusay ng performance kumpara sa mga nakaraang pamamaraan .

Ibubuod namin ang mga pangunahing takeaways mula sa papel na ito.

Panimula

Ang mga modelo ng pagsasabog ay lumitaw bilang isang mahusay na tool para sa synthesis at pag-edit ng larawan at video, na nag-aalok ng mga resultang may mataas na kalidad. Gayunpaman, ang kanilang likas na umuulit, na nangangailangan ng maraming pagsusuri sa network upang baguhin ang ingay sa magkakaugnay na mga imahe, ay limitado ang kanilang pagiging praktikal para sa mga real-time na aplikasyon. Iba't ibang mga diskarte ang iminungkahi upang mapabilis ang mga modelo ng pagsasabog. Ipinakilala ng LADD ang isang bagong diskarte, na gumagamit ng mga generative na feature mula sa mga pretrained na LDM, na nagbibigay-daan para sa mahusay na high-resolution na synthesis ng imahe sa isang bahagi ng mga hakbang na kinakailangan ng mga tradisyonal na pamamaraan.

Background

Nagsisimula ang papel sa pamamagitan ng pagbibigay ng pangkalahatang-ideya ng mga modelo ng diffusion at ang kanilang distillation. Gumagana ang mga tradisyonal na modelo ng pagsasabog sa pamamagitan ng unti-unting pag-denoising ng isang imahe sa pamamagitan ng maraming umuulit na hakbang, na ginagawang mabagal at magastos sa computation ang proseso. Ang mga pamamaraan ng distillation, kabilang ang Adversarial Diffusion Distillation (ADD), ay nagsikap na i-streamline ang prosesong ito sa pamamagitan ng pagbawas sa bilang ng mga hakbang na kailangan. Gayunpaman, ang ADD ay nahaharap sa mga limitasyon tulad ng isang nakapirming resolution ng pagsasanay, at ang pangangailangan ng pag-decode sa RGB space para sa distilling latent diffusion models, na maaaring limitahan ang high-resolution na pagsasanay.

Pamamaraan

Tinutugunan ng LADD ang mga isyung ito sa pamamagitan ng direktang pag-distill sa latent space, sa gayo'y naiiwas ang pangangailangang mag-decode sa pixel space, at nagbibigay-daan para sa pagsasanay sa mas matataas na resolution. Hindi tulad ng ADD, na umaasa sa isang pretrained na discriminator na tumatakbo sa pixel space, ang LADD ay gumagamit ng isang bagong diskarte kung saan ang discriminator at modelo ng guro ay pinag-isa, na direktang gumagana sa mga nakatago. Ang pamamaraang ito ay hindi lamang pinapasimple ang proseso ng pagsasanay, ngunit nagbibigay din ng ilang mga pakinabang, kabilang ang kahusayan, ang kakayahang magbigay ng tukoy na feedback sa antas ng ingay, at ang kapasidad para sa pagsasanay sa Multi-Aspect Ratio (MAR).

Mga Eksperimento at Resulta

Malawakang sinusuri ng papel ang LADD sa pamamagitan ng iba't ibang mga eksperimento, na nagpapakita ng mahusay na pagganap nito sa pag-synthesize ng mga larawang may mataas na resolution na may ilang hakbang lamang. Kapansin-pansin, kapag inilapat sa Stable Diffusion 3 (SD3), ang LADD ay nagreresulta sa isang modelong tinatawag na SD3-Turbo, na nakakakuha ng maihahambing na kalidad ng imahe sa estado- of-the-art na text-to-image generators sa apat na hakbang lang. Tinutuklasan din ng mga eksperimento ang epekto ng iba't ibang distribusyon ng ingay ng guro, ang paggamit ng synthetic na data, latent distillation approach, at ang scaling na gawi ng LADD.

Paghahambing sa State-of-the-Art

Ang pagiging epektibo ng LADD ay higit na binibigyang-diin sa pamamagitan ng paghahambing sa kasalukuyang nangungunang mga pamamaraan sa text-to-image at image-to-image synthesis. Ang SD3-Turbo ay hindi lamang tumutugma sa pagganap ng modelo ng guro nito (SD3) sa kalidad ng larawan, ngunit nagpapakita rin ng mga makabuluhang pagpapabuti sa iba pang mga baseline sa mga tuntunin ng bilis ng hinuha at pagkakahanay ng teksto ng imahe.

Mga Limitasyon at Mga Direksyon sa Hinaharap

Sa kabila ng mga pagsulong nito, ang LADD ay walang limitasyon. Napansin ng mga may-akda ang isang trade-off sa pagitan ng kapasidad ng modelo, agarang pag-align, at bilis ng hinuha, na maaaring makaapekto sa kakayahan ng modelo na pangasiwaan ang ilang mga hamon sa synthesis ng text-to-image. Kasama sa mga direksyon ng pananaliksik sa hinaharap ang pagtuklas sa trade-off na ito nang mas malalim at pagbuo ng mga diskarte upang mapahusay ang kontrol sa mga lakas ng paggabay sa larawan at teksto.

Konklusyon

Ang "Fast High-Resolution Image Synthesis with Latent Adversarial Diffusion Distillation" ay nagpapakilala ng bagong diskarte sa image/video synthesis na makabuluhang nagpapabilis sa pagbuo ng mga de-kalidad na larawan mula sa mga text prompt. Sa pamamagitan ng pag-distill ng malalaking diffusion model sa latent space, binibigyang-daan ng LADD ang mga real-time na application at nagtatakda ng bagong pamantayan para sa kahusayan at pagganap sa synthesis ng imahe.

Code Labs Academy © 2024 Lahat ng karapatan ay nakalaan.