Classification multi-classe (10 chiffres)

Idée

Au lieu de répondre oui/non à une seule question, le modèle doit choisir entre dix réponses possibles : est-ce un 0, un 1, un 2, …, ou un 9 ?

Pourquoi

Au lieu de répondre oui/non à une seule question, le modèle doit choisir entre dix réponses possibles : est-ce un 0, un 1, un 2, …, ou un 9 ? À chaque entrée correspond une étiquette parmi dix, et le réseau doit apprendre à associer la bonne étiquette à chaque image.

Outil

Comme un test à choix multiple : la sortie est une fonction $\hat{y} : \text{image} \to {0, 1, \dots, 9}$ — une fonction caractéristique d'appartenance à une classe parmi dix.

Formule

Cell 7 du notebook construit le réseau via initialisation(...). Sans regarder, donner les valeurs.

Piège

Le binaire chats/chiens utilise sigmoïde + BCELoss. Tentation : copier-coller pour MNIST avec 10 sorties → catastrophe. Sigmoïde donné 10 probabilités INDÉPENDANTES (somme arbitraire), au lieu d'une distribution propre. Il faut Softmax (somme = 1) + CrossEntropyLoss. Le notebook Cell 9-13 le fait correctement, mais l'erreur de copier-coller est très tentante.