arXiv reaDer
EGSDE:エネルギー誘導確率微分方程式による対になっていない画像から画像への変換
EGSDE: Unpaired Image-to-Image Translation via Energy-Guided Stochastic Differential Equations
スコアベースの拡散生成モデル(SDGM)は、対になっていない画像間変換(I2I)でSOTAFIDの結果を達成しました。ただし、既存の方法ではソースドメインのトレーニングデータが完全に無視されるため、対になっていないI2Iのソリューションが最適ではないことがわかります。この目的のために、ソースドメインとターゲットドメインの両方で事前トレーニングされたエネルギー関数を使用して、現実的で忠実な不対I2Iの事前トレーニングされたSDEの推論プロセスをガイドするエネルギー誘導確率微分方程式(EGSDE)を提案します。 2つの特徴抽出器に基づいて、エネルギー関数を慎重に設計し、転送された画像がドメインに依存しない特徴を保持し、ドメイン固有の特徴を破棄するようにします。さらに、専門家の製品としてのEGSDEの代替説明を提供します。ここでは、3人の専門家(SDEと2つの特徴抽出器に対応)のそれぞれが忠実性またはリアリズムにのみ貢献します。経験的に、EGSDEを、4つのメトリックの下で広く採用されている3つの対になっていないI2Iタスクのベースラインの大規模なファミリと比較します。 EGSDEは、ほぼすべての設定で既存のSDGMベースの方法を一貫して上回っているだけでなく、忠実なパフォーマンスを損なうことなく、SOTAリアリズムの結果(たとえば、AFHQのCat to Dogで65.82のFID、Wild to Dogで59.75のFID)を達成します。
Score-based diffusion generative models (SDGMs) have achieved the SOTA FID results in unpaired image-to-image translation (I2I). However, we notice that existing methods totally ignore the training data in the source domain, leading to sub-optimal solutions for unpaired I2I. To this end, we propose energy-guided stochastic differential equations (EGSDE) that employs an energy function pretrained on both the source and target domains to guide the inference process of a pretrained SDE for realistic and faithful unpaired I2I. Building upon two feature extractors, we carefully design the energy function such that it encourages the transferred image to preserve the domain-independent features and discard domainspecific ones. Further, we provide an alternative explanation of the EGSDE as a product of experts, where each of the three experts (corresponding to the SDE and two feature extractors) solely contributes to faithfulness or realism. Empirically, we compare EGSDE to a large family of baselines on three widely-adopted unpaired I2I tasks under four metrics. EGSDE not only consistently outperforms existing SDGMs-based methods in almost all settings but also achieves the SOTA realism results (e.g., FID of 65.82 in Cat to Dog and FID of 59.75 in Wild to Dog on AFHQ) without harming the faithful performance.
updated: Thu Jul 14 2022 03:08:33 GMT+0000 (UTC)
published: Thu Jul 14 2022 03:08:33 GMT+0000 (UTC)
参考文献 (このサイトで利用可能なもの) / References (only if available on this site)
被参照文献 (このサイトで利用可能なものを新しい順に) / Citations (only if available on this site, in order of most recent)
Amazon.co.jpアソシエイト