A seguirPróximo guia
Super-resolução ESRGAN e GAN
IA visual
GUIA visual de IA
Wasserstein GAN (WGAN) é um redesenho do objetivo de treinamento GAN que usa a distância de Wasserstein em vez da perda min-max original.
Ouça este guiaNarrado por IA · ~2 min
Isso torna o treinamento GAN notoriamente instável muito mais confiável e fornece um valor de perda que realmente se correlaciona com a qualidade da imagem.
Os GANs originais treinam duas redes em um cabo de guerra: um gerador cria imagens falsas e um discriminador tenta identificá-las. Isto muitas vezes entra em colapso ou estagna porque a perda do discriminador não diz nada de útil sobre o progresso. O WGAN, introduzido por Arjovsky, Chintala e Bottou em 2017, substitui o discriminador por um 'crítico' que avalia o quão real uma imagem parece em uma escala contínua, em vez de classificar o real versus o falso. O alvo de treinamento passa a ser a distância de Wasserstein (motorista de terra) entre as distribuições de dados reais e geradas. Essa distância fornece gradientes mais suaves e significativos, mesmo quando as duas distribuições quase não se sobrepõem, reduzindo drasticamente o colapso do modo e tornando a curva de perda um sinal de qualidade genuína.
Gerando faces e texturas fotorrealistas onde os GANs vanilla foram reduzidos a algumas saídas repetidas
Produzir imagens médicas sintéticas, como ressonância magnética ou patches histológicos, para aumentar os escassos conjuntos de dados rotulados
Modelagem de eventos de colisão de partículas em simulações de física de alta energia onde o treinamento estável é crítico
Servindo como referência básica na pesquisa de ML porque sua perda rastreia a qualidade da amostra em relação ao treinamento
A visão central do WGAN, de que a escolha da distância de distribuição molda a qualidade do gradiente, ainda ecoa na modelagem generativa. Embora os modelos de difusão agora dominem a síntese de imagens, as ideias de transporte ideal do WGAN reaparecem na correspondência de fluxo, nos métodos da ponte de Schrodinger e na destilação de modelos de difusão em geradores rápidos de poucas etapas. Espere que os objetivos do estilo Wasserstein continuem informando as abordagens híbridas onde o treinamento estável e uma métrica de perda significativa são importantes, especialmente em domínios científicos e com poucos dados.
Boletim informativo gratuito
Três histórias verificadas de IA a cada manhã de dia útil, escritas em linguagem simples. Gratuito para sempre, sem anúncios.
Um e-mail a cada dia útil. Cancele a inscrição com um clique. Nunca vendemos ou compartilhamos seu endereço.
Teste você mesmo
Feedback instantâneo sobre cada resposta e um certificado compartilhável com um ID verificável assim que você concluir um curso.
Apoie a educação gratuita em IA. O AI Understanding é uma organização sem fins lucrativos 501(c)(3) — sem anúncios, sem paywall, nunca. Faça uma doação
Wasserstein GAN (WGAN) é um redesenho do objetivo de treinamento GAN que usa a distância de Wasserstein em vez da perda min-max original. Isso torna o treinamento GAN notoriamente instável muito mais confiável e fornece um valor de perda que realmente se correlaciona com a qualidade da imagem.
WGAN substitui o objetivo original baseado em Jensen-Shannon pela distância de Wasserstein, que fornece gradientes mais suaves mesmo quando as distribuições quase não se sobrepõem.
O crítico classifica as imagens em uma escala contínua em vez de classificar o real versus o falso, que é o que faz o objetivo de Wasserstein funcionar.
A dualidade que permite ao WGAN estimar a distância de Wasserstein vale apenas para funções 1-Lipschitz, portanto os gradientes do crítico devem ser limitados.
O primeiro artigo WGAN usou recorte bruto de gramatura; Mais tarde, o WGAN-GP o substituiu por uma penalidade de gradiente mais suave.
Os gradientes mais suaves do WGAN restringem o colapso do modo e produzem uma perda que realmente se correlaciona com a qualidade da amostra.
Continue aprendendo
Mais guias escolhidos para este tópico
A seguirPróximo guia
Super-resolução ESRGAN e GAN
IA visual