arXiv reaDer
スタイル転送による絵画分類におけるデータ バイアスへの取り組み
Tackling Data Bias in Painting Classification with Style Transfer
ドメインのギャップによるモデルの偏りと、芸術スタイルの不均一な分布によるデータの偏りにより、絵画コレクションで分類器をトレーニングすることは困難です。データ蒸留、従来のデータ拡張、スタイル転送などの以前の手法は、タスク固有のトレーニング データセットまたはドメイン適応を使用して、分類器のトレーニングを改善します。 Kaokore データセットのような小さな絵画データセットのデータ バイアスを処理すると同時に、実世界の画像でトレーニングされたモデルを微調整する際にドメイン適応を考慮するシステムを提案します。私たちのシステムは、スタイルの転送と分類の 2 つの段階で構成されています。スタイル転送段階では、均一にサンプリングされたコンテンツとスタイル イメージを使用して、クラスごとにスタイル化されたトレーニング サンプルを生成し、ドメインごとにスタイル変換ネットワークをトレーニングします。分類段階では、元のトレーニング データセットと定型化された画像でトレーニングするときに、アテンション レイヤーでスタイル レイヤーとコンテンツ レイヤーの有効性を解釈できます。多数派クラスと少数派クラスの拡張サンプルの割合を動的に変更することで、モデルのパフォーマンスと収束をトレードオフできます。より少ないトレーニング エポックとより少ないトレーニング パラメーターで分類子を使用して、SOTA に匹敵する結果を達成します。
It is difficult to train classifiers on paintings collections due to model bias from domain gaps and data bias from the uneven distribution of artistic styles. Previous techniques like data distillation, traditional data augmentation and style transfer improve classifier training using task specific training datasets or domain adaptation. We propose a system to handle data bias in small paintings datasets like the Kaokore dataset while simultaneously accounting for domain adaptation in fine-tuning a model trained on real world images. Our system consists of two stages which are style transfer and classification. In the style transfer stage, we generate the stylized training samples per class with uniformly sampled content and style images and train the style transformation network per domain. In the classification stage, we can interpret the effectiveness of the style and content layers at the attention layers when training on the original training dataset and the stylized images. We can tradeoff the model performance and convergence by dynamically varying the proportion of augmented samples in the majority and minority classes. We achieve comparable results to the SOTA with fewer training epochs and a classifier with fewer training parameters.
updated: Fri Jan 06 2023 14:33:53 GMT+0000 (UTC)
published: Fri Jan 06 2023 14:33:53 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アソシエイト