arXiv reaDer
StyleGAN画像の自動セグメンテーションのための自己教師ありワンショット学習
Self-Supervised One-Shot Learning for Automatic Segmentation of StyleGAN Images
StyleGANによって生成された合成画像の自動ワンショットセグメンテーションのフレームワークを提案します。私たちのフレームワークは、GANジェネレーターのマルチスケールの隠れた機能が、生成された画像の自動オンザフライセグメンテーションに利用できる有用なセマンティック情報を保持しているという観察に基づいています。これらの機能を使用して、私たちのフレームワークは、ピクセルごとの分類のために隠された機能をコンパクトなスペースに投影する自己教師あり対照的クラスタリング アルゴリズムを使用して、合成画像をセグメント化することを学習します。この新しい対照的学習器は、画像セグメンテーションにピクセル単位で交換された予測損失を使用することに基づいており、ワンショット セグメンテーションの特徴ベクトルの学習を高速化します。多くの標準的なベンチマークで実装をテストして、セグメンテーションのパフォーマンスが得られました。このセグメンテーション パフォーマンスは、半教師付きベースライン メソッドを平均 1.02% のマージンで上回るだけでなく、推論速度を 4.5 倍向上させます。最後に、提案されたワンショット学習器を使用して、BagGAN を実装した結果も示します。BagGAN は、脅威検出用の注釈付きの合成手荷物 X 線スキャンを生成するためのフレームワークです。このフレームワークは、PIDRay バゲージ ベンチマークでトレーニングおよびテストされ、手動注釈に基づくベースライン セグメンターに匹敵するパフォーマンスが得られました。
We propose a framework for the automatic one-shot segmentation of synthetic images generated by a StyleGAN. Our framework is based on the observation that the multi-scale hidden features in the GAN generator hold useful semantic information that can be utilized for automatic on-the-fly segmentation of the generated images. Using these features, our framework learns to segment synthetic images using a self-supervised contrastive clustering algorithm that projects the hidden features into a compact space for per-pixel classification. This novel contrastive learner is based on using a pixel-wise swapped prediction loss for image segmentation that leads to faster learning of the feature vectors for one-shot segmentation. We have tested our implementation on a number of standard benchmarks to yield a segmentation performance that not only outperforms the semi-supervised baseline methods by an average wIoU margin of 1.02% but also improves the inference speeds by a factor of 4.5. Finally, we also show the results of using the proposed one-shot learner in implementing BagGAN, a framework for producing annotated synthetic baggage X-ray scans for threat detection. This framework was trained and tested on the PIDRay baggage benchmark to yield a performance comparable to its baseline segmenter based on manual annotations.
updated: Fri Mar 17 2023 18:50:19 GMT+0000 (UTC)
published: Fri Mar 10 2023 01:04:27 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アソシエイト