はとむぎ研究室

大学生の備忘録でした

GANの仕組み

GANの仕組み

 与えられた画像が本物か偽物かを見分ける識別器 vs 識別器を騙す画像を作る生成器

 

識別器の学習

 下図の通り、[0,1]のノイズを与えられた生成器が作成した偽物の画像と、学習させたい本物の画像を用意する。<本物の画像, 正解ラベル(全て1)>と<偽物の画像, 正解ラベル(全て0)>の2組を識別器に与えて、識別器を学習させる。

f:id:heart_mugi:20200914230939j:plain

識別器の学習の様子

 

生成器の学習

 下図の通り、ノイズから作成した偽物の画像と全て1(つまり全て本物とする)正解ラベルを識別器に与え、生成器を学習させる。つまり、生成器は作成した画像が全て本物だと誤認されるよう学習している。注意点として、このとき識別器は学習していない。

f:id:heart_mugi:20200914231348j:plain

生成器の学習の様子

 

参照

 

実践GAN ~敵対的生成ネットワークによる深層学習~ (Compass Booksシリーズ)

実践GAN ~敵対的生成ネットワークによる深層学習~ (Compass Booksシリーズ)