arXiv reaDer
適応型対照学習による統一任意スタイル転送フレームワーク
A Unified Arbitrary Style Transfer Framework via Adaptive Contrastive Learning
Unified Contrastive Arbitrary Style Transfer (UCAST) は、CNN ベース、ViT ベース、フローベースの方法など、既存のほとんどの任意の画像スタイル転送モデルに適合する新しいスタイル表現学習および転送フレームワークです。画像スタイル転送タスクの重要なコンポーネントとして、満足のいく結果を得るには適切なスタイル表現が不可欠です。ディープ ニューラル ネットワークに基づく既存のアプローチは、通常、二次統計を使用して出力を生成します。ただし、単一の画像から計算されたこれらの手作りの機能では、スタイル情報を十分に活用できず、局所的な歪みやスタイルの不一致などのアーティファクトが発生します。これらの問題に対処するために、特定のスタイルと全体的なスタイル分布との関係を考慮して、対照学習に基づいて大量の画像からスタイル表現を直接学習することを提案します。具体的には、入力依存の温度を導入することにより、スタイル転送のための適応型コントラスト学習スキームを提示します。私たちのフレームワークは、スタイル表現とスタイル転送のための並列対照学習スキーム、スタイル分布の効果的な学習のためのドメイン拡張モジュール、およびスタイル転送のための生成ネットワークの 3 つの主要なコンポーネントで構成されています。定性的および定量的な評価を行い、最先端の方法よりも優れた結果が得られることを示します。
We present Unified Contrastive Arbitrary Style Transfer (UCAST), a novel style representation learning and transfer framework, which can fit in most existing arbitrary image style transfer models, e.g., CNN-based, ViT-based, and flow-based methods. As the key component in image style transfer tasks, a suitable style representation is essential to achieve satisfactory results. Existing approaches based on deep neural network typically use second-order statistics to generate the output. However, these hand-crafted features computed from a single image cannot leverage style information sufficiently, which leads to artifacts such as local distortions and style inconsistency. To address these issues, we propose to learn style representation directly from a large amount of images based on contrastive learning, by taking the relationships between specific styles and the holistic style distribution into account. Specifically, we present an adaptive contrastive learning scheme for style transfer by introducing an input-dependent temperature. Our framework consists of three key components, i.e., a parallel contrastive learning scheme for style representation and style transfer, a domain enhancement module for effective learning of style distribution, and a generative network for style transfer. We carry out qualitative and quantitative evaluations to show that our approach produces superior results than those obtained via state-of-the-art methods.
updated: Thu Mar 23 2023 14:13:05 GMT+0000 (UTC)
published: Thu Mar 09 2023 04:35:00 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アソシエイト