Glossar

Stochastischer Gradientenabstieg (SGD)

Entdecke, wie der stochastische Gradientenabstieg Modelle für maschinelles Lernen optimiert und effizientes Training für große Datensätze und Deep Learning-Aufgaben ermöglicht.

Trainiere YOLO Modelle einfach
mit Ultralytics HUB

Mehr erfahren

Der stochastische Gradientenabstieg (SGD) ist ein beliebter und effizienter Optimierungsalgorithmus, der häufig beim maschinellen Lernen (ML) und insbesondere beim Deep Learning (DL) eingesetzt wird. Er ist eine Abwandlung des Standard-Gradientenabstiegs, wurde aber speziell für Geschwindigkeit und Effizienz beim Umgang mit sehr großen Datensätzen entwickelt. Anstatt den Gradienten (die Richtung des steilsten Abstiegs für die Verlustfunktion) in jedem Schritt anhand des gesamten Datensatzes zu berechnen, nähert sich der SGD dem Gradienten anhand einer einzelnen, zufällig ausgewählten Datenprobe oder einer kleinen Teilmenge, die als Mini-Batch bezeichnet wird. Dieser Ansatz reduziert die Rechenkosten und den Speicherbedarf erheblich und macht es möglich, komplexe Modelle auf riesigen Datenmengen zu trainieren, wie sie z. B. in der Computer Vision vorkommen.

Relevanz beim maschinellen Lernen

SGD ist ein Eckpfeiler für das Training großer maschineller Lernmodelle, insbesondere der komplexen Neuronalen Netze (NN), die viele moderne KI-Anwendungen antreiben. Seine Effizienz macht es unverzichtbar, wenn du mit Datensätzen arbeitest, die zu groß sind, um in den Speicher zu passen, oder deren Verarbeitung mit dem traditionellen Batch Gradient Descent zu lange dauern würde. Modelle wie Ultralytics YOLO nutzen SGD oder seine Varianten häufig während des Trainingsprozesses, um Muster für Aufgaben wie Objekterkennung, Bildklassifizierung und Bildsegmentierung zu lernen. Große Deep Learning-Frameworks wie PyTorch und TensorFlow bieten robuste Implementierungen von SGD, was ihre grundlegende Rolle im KI-Ökosystem unterstreicht.

Schlüsselkonzepte

Um SGD zu verstehen, braucht es ein paar grundlegende Ideen:

  • Verlustfunktion: Ein Maß dafür, wie gut die Vorhersagen des Modells mit den tatsächlichen Zielwerten übereinstimmen. SGD zielt darauf ab, diese Funktion zu minimieren.
  • Lernrate: Ein Hyperparameter, der die Schrittgröße bei jeder Parameteraktualisierung bestimmt. Eine gute Lernrate zu finden, ist entscheidend für ein effektives Training. Lernratenpläne werden oft verwendet, um sie während des Trainings anzupassen.
  • Batch Size: Die Anzahl der Trainingsbeispiele, die in einer Iteration zur Schätzung des Gradienten verwendet werden. Beim reinen SGD ist die Stapelgröße 1. Wenn kleine Teilmengen verwendet werden, spricht man oft von Mini-Batch Gradient Descent.
  • Trainingsdaten: Der Datensatz, der zum Trainieren des Modells verwendet wird. SGD verarbeitet diese Daten Stichprobe für Stichprobe oder in Mini-Batches. Qualitativ hochwertige Daten sind wichtig und erfordern oft eine sorgfältige Datenerfassung und -beschriftung.
  • Steigung: Ein Vektor, der die Richtung des steilsten Anstiegs der Verlustfunktion angibt. SGD verschiebt Parameter in die entgegengesetzte Richtung des Gradienten, der aus einer Probe oder einem Mini-Batch berechnet wurde.
  • Epoche: Ein vollständiger Durchlauf durch den gesamten Trainingsdatensatz. Das Training umfasst normalerweise mehrere Epochen.

Unterschiede zu verwandten Konzepten

SGD ist einer von mehreren Optimierungsalgorithmen, und es ist wichtig, ihn von anderen zu unterscheiden:

  • Batch Gradient Descent (BGD): Berechnet den Gradienten in jedem Schritt anhand des gesamten Trainingsdatensatzes. Dies liefert eine genaue Gradientenschätzung, ist aber bei großen Datensätzen rechen- und speicherintensiv. Im Vergleich zu den verrauschten Aktualisierungen des SGD führt dies zu einer gleichmäßigeren Konvergenz.
  • Mini-Batch Gradient Descent: Ein Kompromiss zwischen BGD und SGD. Dabei wird der Gradient anhand einer kleinen, zufälligen Teilmenge (Mini-Batch) der Daten berechnet. Diese Methode vereint die Genauigkeit von BGD mit der Effizienz von SGD und ist in der Praxis am häufigsten anzutreffen. Die Leistung kann von der Stapelgröße abhängen.
  • Adam Optimierer: Ein Algorithmus zur Optimierung der adaptiven Lernrate, der individuelle adaptive Lernraten für verschiedene Parameter berechnet. Er konvergiert oft schneller als der Standard-SGD, kann aber manchmal weniger effektiv verallgemeinern, wie in Untersuchungen wie "The Marginal Value of Adaptive Gradient Methods in Machine Learning" diskutiert. Darüber hinaus gibt es viele Varianten des Gradientenabstiegs.

Anwendungen in der realen Welt

Die Effizienz von SGD ermöglicht den Einsatz in zahlreichen groß angelegten KI-Anwendungen:

Beispiel 1: Training großer Sprachmodelle (LLMs)

Für das Training von Modellen, wie sie in der natürlichen Sprachverarbeitung (NLP) verwendet werden, werden oft riesige Textdatensätze (Milliarden von Wörtern) benötigt. SGD und seine Varianten (wie Adam) sind unerlässlich, um diese Daten effizient zu iterieren und Modelle wie GPT-4 oder die auf Hugging Face Grammatik, Kontext und Semantik lernen können. Der stochastische Charakter hilft dabei, lokale Minima in der komplexen Verlustlandschaft zu vermeiden.

Beispiel 2: Training zur Objekterkennung in Echtzeit

Für Modelle wie Ultralytics YOLO , die für Echtzeit-Inferenzen entwickelt wurden, muss das Training effizient sein. Mit SGD können Entwickler diese Modelle auf großen Bilddatensätzen wie COCO oder benutzerdefinierten Datensätzen trainieren, die über Plattformen wie Ultralytics HUB verwaltet werden. Die schnellen Updates ermöglichen eine schnellere Konvergenz im Vergleich zu Batch GD, was für schnelle Iterationen bei der Modellentwicklung und der Abstimmung der Hyperparameter entscheidend ist. Diese Effizienz unterstützt Anwendungen in Bereichen wie autonome Fahrzeuge und Robotik.

Alles lesen