Trainiere ein neu­ro­na­les Netzwerk

1. Vorbereitung

Stelle zuerst sicher, dass du keinen Ordner geöffnet hast. Um sicherzugehen, drücke einfach den Shortcut für »Ordner schließen«: StrgK und dann F. Drücke StrgShiftE, um die Seitenleiste mit dem Explorer-Tab zu öffnen. Dein Work­space sollte jetzt ungefähr so aussehen:

Für diese Anleitung brauchst du ein Repository, das du klonen kannst, indem du auf den blauen Button »Clone Repository« klickst. Gib die folgende URL ein und bestätige mit Enter:

https://github.com/specht/digits-experiment.git

Als nächstes musst du angeben, in welches Verzeichnis du das Repository klonen möchtest. Bestätige den Standardpfad /workspace/ mit Enter.

Beantworte die Frage »Would you like to open the cloned repository?« mit »Open«.

Du siehst nun auf der linken Seite ein paar Verzeichnisse und Da­tei­en, die wir für dieses Experiment verwenden werden.

Öffne nun die Datei 00-install-packages.ipynb, indem du darauf klickst. Es handelt sich dabei um ein Jupyter-Notebook, das du verwenden kannst, um die notwendigen Pakete zu installieren. Links neben dem ersten Codeblock siehst du ein Play-Symbol. Klicke darauf, um den Codeblock auszuführen.

Bevor der Code ausgeführt werden kann, müssen wir noch ein paar Pakete installieren. Bestätige die Meldung »Install / Enable suggested extensions« mit Enter:

Wenn das erledigt ist, musst du noch einen Python-Interpreter auswählen. Bestätige die Meldung »Python Environments« und »★ Python 3.10.12«:

Wenn du die Meldung »Pakete installiert« siehst, hast du den Vorbereitungsschritt erfolgreich abgeschlossen:

2. Trainingsmaterial herstellen

Du hast nun zwei Möglichkeiten: entweder verwendest du fertige Scans mit hand­ge­schrie­ben­en Ziffern, oder du schreibst selbst Ziffern – das ist natürlich viel spannender, weil du in diesem Fall dein neu­ro­na­les Netzwerk mit deinen eigenen Daten trainieren wirst. In jedem Fall brauchst du mindestens zwei Seiten mit hand­ge­schrie­ben­en Ziffern als Trainingsmaterial.

Variante 1: Fertige Scans verwenden

Du kannst diese Seiten im PNG-Format hier herunterladen und per Drag & Drop in deinem Work­space in das Verzeichnis 01-pages ziehen:

Variante 2: Ziffern selbst schreiben

Drucke diese PDF-Vorlage ohne zusätzliche Seitenränder zweimal aus und trage in jedes Feld die entsprechenden Ziffern ein. Scanne die Blätter anschließend mit einer Auflösung von 300 dpi, speichere sie im PNG-Format und ziehe sie per Drag & Drop in deinen Work­space in das Verzeichnis 01-pages.

3. Trainingsmaterial vorbereiten

Stelle sicher, dass du mindestens zwei Bilddateien im Verzeichnis 01-pages hast. Du kannst die Da­tei­en öffnen, um zu überprüfen, ob sie korrekt sind:

Öffne nun das zweite Jupyter-Notebook 01-prepare-training-data.ipynb, und führe den ersten Codeblock aus, indem du auf das Play-Symbol klickst.

Im nächsten Schritt siehst du den Effekt der Farbanpassung:

Wenn die Bilder korrekt angezeigt werden, führe den nächsten Codeblock aus. In diesem Schritt werden die gedruckten Rahmen in jeder Bilddatei gesucht:

Der nächste Schritt zeigt dir, wo die Rahmen gefunden wur­den:

Wenn das Ergebnis zufriedenstellend aussieht, führe den nächsten Codeblock aus. In diesem Schritt werden die einzelnen Ziffern aus den Bildern ausgeschnitten:

Im nächsten Schritt werden 10 Bilddateien generiert, in denen die einzelnen Ziffern gesammelt werden:

Du kannst dir die Bilder anschauen, indem du auf das Verzeichnis 04-sheets klickst:

Wenn alle Schritte erfolgreich durchgelaufen sind, können wir mit dem Training des Modells beginnen.

4. Modell trainieren

Öffne nun das dritte Juptyer-Notebook 02-train-model.ipynb, indem du darauf klickst. Es enthält den Code, um unser neu­ro­na­les Netzwerk zu trainieren, das die hand­ge­schrie­ben­en Ziffern erkennen soll. Im ersten Schritt wird die Menge aller Trainingsdaten in drei Teile aufgeteilt: 80% für das Training, 10% für die Validierung und 10% für den Test:

Im nächsten Schritt wird die Architektur des Modells festgelegt. Es handelt sich um ein einfaches neu­ro­na­les Netzwerk mit vier Schichten:

• die erste Schicht ist ein »Flatten Layer«, das die 28x28 Pixel großen Bilder in einen Vektor mit 784 Elementen umwandelt.
• die zweite und dritte Schicht sind jeweils ein »Dense Layer« mit 100 Neuronen und der Aktivierungsfunktion »ReLU«.
• die vierte Schicht ist ebenfalls ein »Dense Layer« mit 10 Neuronen, die die jeweiligen Ziffern repräsentieren, jedoch mit der Aktivierungsfunktion »Softmax«.

Dieses Modell verfügt über eine Gesamtzahl von 89.610 Parametern, die während des Trainings optimiert werden.

Wir sind jetzt bereit, das Modell zu trainieren. Klicke auf das Play-Symbol, um den Codeblock auszuführen. Nach 30 Trainingsepochen ist das Training beendet.

Im nächsten Schritt ermitteln wir, wie gut das Modell ist. Dazu verwenden wir die Testdaten, die wir zuvor beiseite gelegt haben. Die wichtigen Kennwerte sind der Verlust (loss) und die Genauigkeit (accuracy):

• der Verlust sollte möglichst klein sein, idealerweise nahe bei 0.
• die Genauigkeit sollte möglichst groß sein, idealerweise nahe bei 100%.

Um den vorherigen Schritt zu visualisieren, können wir uns nun ein paar Testbilder anzeigen lassen und die Vorhersagen des Modells überprüfen:

Die roten Ziffern markieren die falschen Vorhersagen. Das Modell ist noch nicht perfekt, aber es ist bereits in der Lage, handgeschriebene Ziffern zu erkennen.

5. Modell ausprobieren

Im nächsten Schritt wollen wir unser Modell interaktiv im Webbrowser testen. Dazu müssen wir das gespeicherte Modell zunächst in ein anderes Format konvertieren, das von TensorFlow.js gelesen werden kann. Führe dazu den nächsten Codeblock aus:

Wenn das Modell erfolgreich konvertiert wur­de, können wir es im Webbrowser testen. Klicke dazu rechts unten auf den Button »Go Live«. Du kannst nun links oben mit der Maus Ziffern schreiben und zuschauen, wie sich die Aktivierungen der Neuronen im Modell verändern. Du siehst die einzelnen Schichten und die Aktivierungen der Neuronen. Ganz rechts siehst du die Vorhersage des Modells.

Versuche, die Grenze zwischen zwei Ziffern zu finden, um zu sehen, wie das Modell reagiert. Mit der rechten Maustaste kannst du deine Zeichnung wieder wegradieren.

6. Modell verbessern

Um unser Modell zu verbessern, können wir ein Convolutional Neural Network (CNN) verwenden. Dabei werden die Pixel nicht mehr un­ab­hän­gig voneinander betrachtet, sondern unter Berücksichtigung ihrer Lage zueinander.

Um ein CNN zu verwenden, scrolle nach oben zum Abschnitt »Architektur des Modells« und setze die Variable use_model auf den Wert 2. Führe dann diesen und alle folgenden Codeblöcke erneut aus.

Das neue Modell hat eine etwas andere Architektur:

• die erste Schicht ist ein »Conv2D Layer«, das das 28x28 Pixel großen Eingabebild mit 32 3x3-Filtern in 32 26x26 Pixel große Bilder umwandelt.
• die zweite Schicht ist ein »MaxPooling2D Layer«, das die Größe der Bilder halbiert (13x13).
• die dritte Schicht ist ein weiteres »Conv2D Layer«, des die 13x13 großen Bilder mit 64 3x3-Filtern in 64 11x11 Pixel große Bilder umwandelt.
• die vierte Schicht ist ein weiteres »MaxPooling2D Layer«, des die Größe der Bilder halbiert (5x5).
• die fünfte Schicht ist ein »Flatten Layer«, das die 5x5 Pixel großen Bilder in einen Vektor mit 1600 Elementen umwandelt.
• die sechste Schicht ist ein »Dense Layer« mit 128 Neuronen und der Aktivierungsfunktion »ReLU«.
• die siebte Schicht ist ein weiteres »Dense Layer« mit 10 Neuronen, die die jeweiligen Ziffern repräsentieren, jedoch mit der Aktivierungsfunktion »Softmax«.

Das verbesserte Modell verfügt über eine Gesamtzahl von 225.034 Parametern, also ca. 2,5 mal so viele Parameter wie im vorherigen Modell:

Da das neue Modell mehr Parameter hat, dauert das Training auch länger. Nach 30 Epochen ist das Training beendet:

Wie du an den Werten für Verlust und Genauigkeit sehen kannst, ist das neue Modell deutlich besser als das vorherige.

Bei der Evaluation des neuen Modells siehst du, dass der Validation Loss und die Validation Accuracy zwar schlechter sind als beim Traning, aber immer noch deutlich besser als beim vorherigen Modell:

Die Testbilder sehen jetzt so aus:

Wenn du das Modell im Webbrowser testest, siehst du, wie die Pixel nicht mehr im ersten Schritt voneinander getrennt werden, sondern über mehrere Schichten hinweg im Quadrat angeordnet gemeinsam behandelt werden:

Falls du ein noch besseres Modell haben möchtest, kannst du die Anzahl der Epochen erhöhen oder die Architektur des Modells weiter anpassen. Du kannst auch die Anzahl der Trainingsdaten erhöhen, indem du mehr Blätter mit hand­ge­schrie­ben­en Ziffern verwendest. Hier findest du weitere Blätter, die du verwenden kannst:

7. Zusammenfassung

In diesem Experiment hast du gelernt, wie du ein neu­ro­na­les Netzwerk trai­nierst, das handgeschriebene Ziffern er­kennt. Du hast gesehen, wie du die Trainingsdaten vorbereitest, das Modell trai­nierst und es im Webbrowser testest. Du hast auch gesehen, wie du das Modell verbessern kannst, indem du ein Convolutional Neural Network verwendest.