Sign In

BOLT-GAN: Bayes-Optimal Loss for Stable GAN Training

Created by
  • Haebom
Category
Empty

저자

Mohammadreza Tavasoli Naeini, Ali Bereyhi, Morteza Noshad, Ben Liang, Alfred O. Hero III

개요

BOLT-GAN은 Bayes Optimal Learning Threshold (BOLT)에서 영감을 얻은 WGAN 프레임워크의 간단하지만 효과적인 수정 사항을 소개합니다. 립시츠 연속 판별자를 사용하여 BOLT-GAN이 Earth Mover (Wasserstein) 거리와 다른 메트릭 거리를 암묵적으로 최소화하며 훈련 안정성이 향상됨을 보여줍니다. CIFAR-10, CelebA-64, LSUN Bedroom-64, LSUN Church-64의 네 가지 표준 이미지 생성 벤치마크에 대한 실험적 평가 결과, BOLT-GAN이 WGAN보다 일관되게 우수한 성능을 보이며 Frechet Inception Distance (FID)를 10-60% 더 낮게 달성했습니다. BOLT는 GAN 훈련을 향상시키는 데 널리 적용 가능한 원리임을 시사합니다.

시사점, 한계점

BOLT-GAN은 WGAN 프레임워크의 효과적인 수정 사항입니다.
BOLT-GAN은 WGAN보다 향상된 훈련 안정성을 보입니다.
BOLT-GAN은 네 가지 이미지 생성 벤치마크에서 WGAN보다 우수한 성능을 보입니다.
BOLT는 GAN 훈련을 향상시키기 위한 광범위하게 적용 가능한 원리입니다.
해당 논문에서 구체적인 BOLT-GAN의 구조적 세부 사항이나 이론적 증명은 명시되지 않았습니다.
실험은 4가지 이미지 벤치마크에 국한되어, 다양한 데이터셋에서의 일반화 성능을 추가적으로 검증할 필요가 있습니다.
👍