O Google postou em seu blog um estudo da equipe interna do cérebro intitulado “Criando imagens de alta fidelidade usando modelos de difusão”. Neste artigo, os pesquisadores falam sobre os novos avanços que fizeram no dimensionamento de imagens digitais sem perder qualidade.
A equipe do Google Brain treinou um modelo de aprendizado de máquina para transformar fotos de baixa resolução em imagens detalhadas de alta resolução sem praticamente nenhuma perda de qualidade. Os especialistas acreditam que seu desenvolvimento pode ser usado para uma variedade de propósitos, desde melhorar fotos antigas de família até melhorar a qualidade de imagens médicas.
O conceito de modelos de difusão tem sido estudado pelo Google desde 2015, mas até recentemente, o gigante das buscas preferia outra família de métodos de treinamento de IA – modelos gerativos profundos. A empresa descobriu que os resultados da nova abordagem eram nitidamente superiores às tecnologias existentes.
A nova abordagem foi designada SR3. O Google diz que SR3 é um modelo de difusão de ultra-alta resolução que cria uma imagem de alta resolução a partir de ruído puro com base na imagem original de baixa resolução. O modelo é treinado no processo de distorção da imagem, no qual o ruído é gradualmente adicionado à imagem até que apenas o ruído puro permaneça. O algoritmo então inverte o processo, removendo gradativamente o ruído da imagem, guiado pela imagem original de baixa resolução.
O SR3 apresentou melhor desempenho ao dimensionar retratos e fotos da natureza. O algoritmo permite que você obtenha imagens fotorrealísticas enquanto aumenta a resolução de retratos em até dezesseis vezes.
Depois que o Google se convenceu da eficácia do SR3, a empresa foi ainda mais longe com outra abordagem chamada CDM, que é um modelo de difusão de classes condicionalmente. O CDM é treinado em dados da ImageNet, que contém mais de 14 milhões de imagens de alta resolução. O CDM propõe uma abordagem em cascata que primeiro gera uma imagem de baixa resolução, seguida pelo trabalho do SR3 para criar imagens de alta resolução que são gradualmente aumentadas para o mais alto possível. De acordo com o Google, uma imagem com resolução de 32 × 32 pixels pode ser ampliada para 256 × 256 pixels sem perda perceptível, oito vezes. Uma imagem com resolução de 64 × 64 pixels foi completamente dimensionada para uma resolução de 1024 × 1024 pixels, 16 vezes.
Os resultados do trabalho de IA são realmente impressionantes. As imagens finais, apesar de pequenas falhas, parecem muito boas e são percebidas pela maioria dos usuários como as imagens originais.