Remonter du bruit prédit à l'image

Partie 3 · ★★★

À la génération, le réseau reçoit l'image abîmée x(t) et prédit le bruit ajouté. Si la prédiction était parfaite, on pourrait inverser la formule d'abîmage pour retrouver l'image : x(0) = (x(t) − √(1 − ᾱ) × ε) / √ᾱ. Inverser, c'est isoler x(0) en déplaçant d'abord le terme de bruit, puis en divisant par √ᾱ.

Écris retrouver. Données : x(t) = [1,028 ; 0,194], ᾱ = 0,81, et le bruit prédit ε = [0,5 ; −1]. Compare ensuite avec un réseau qui n'aurait rien appris (il prédit [0 ; 0]).

def retrouver(xt, eps_pred, ab):
    return [(x - math.sqrt(1 - ab) * ___) / math.sqrt(___) for x, e in zip(xt, eps_pred)]

xt = [1.028, 0.194]
ab = 0.81
x0_estime = retrouver(xt, [0.5, -1], ab)
x0_sans_reseau = retrouver(xt, [0, 0], ab)

Ctrl/⌘+Entrée

Exercice suivant → La perte de la diffusion : prédire le bruit