Verbessere die Robustheit von KI-Modellen mit Datenaugmentierung

Das Testen ist ein entscheidender Teil beim Aufbau jeder technologischen Lösung. Es zeigt Teams, wie ein System tatsächlich funktioniert, bevor es live geht, und ermöglicht es ihnen, Probleme frühzeitig zu beheben. Dies gilt für viele Bereiche, einschließlich KI, wo Modelle nach der Bereitstellung unvorhersehbare reale Bedingungen bewältigen müssen.

Zum Beispiel ist computer vision ein Bereich der KI, der Maschinen beibringt, Bilder und Videos zu verstehen. Computer-Vision-Modelle wie Ultralytics YOLO26 unterstützen Aufgaben wie Objekterkennung, Instanzsegmentierung und Bildklassifizierung.

Sie können in vielen Branchen für Anwendungen wie Patientenüberwachung, Verkehrsanalyse, automatisierte Kassen und Qualitätsprüfung in der Fertigung eingesetzt werden. Doch selbst mit fortschrittlichen Modellen und qualitativ hochwertigen Trainingsdaten können Vision-KI-Lösungen Schwierigkeiten haben, wenn sie mit realen Schwankungen wie wechselnden Lichtverhältnissen, Bewegungen oder teilweise verdeckten Objekten konfrontiert werden.

Dies geschieht, weil Modelle von den Beispielen lernen, die ihnen während des Trainings gegeben werden. Wenn sie Bedingungen wie Blendung, Bewegungsunschärfe oder teilweise Sichtbarkeit zuvor nicht gesehen haben, ist die Wahrscheinlichkeit geringer, dass sie Objekte in diesen Szenarien korrekt erkennen.

Eine Möglichkeit, die Modellrobustheit zu verbessern, ist data augmentation. Anstatt große Mengen neuer Daten zu sammeln, können Ingenieure kleine und sinnvolle Änderungen an bestehenden Bildern vornehmen, wie etwa das Anpassen der Beleuchtung, Zuschneiden oder Mischen von Bildern. Dies hilft dem Modell zu lernen, dieselben Objekte in einer größeren Vielfalt von Situationen zu erkennen.

In diesem Artikel untersuchen wir, wie Data Augmentation die Robustheit von Modellen und die Zuverlässigkeit von Vision-KI-Systemen verbessert, wenn sie außerhalb kontrollierter Umgebungen eingesetzt werden. Lass uns anfangen!

Wie du die Robustheit eines Modells überprüfst#

Bevor wir in das Thema Datenaugmentation eintauchen, besprechen wir, woran du erkennst, ob ein computer vision model wirklich bereit für den realen Einsatz ist.

Ein robustes Modell funktioniert auch bei wechselnden Bedingungen weiterhin gut, anstatt nur bei sauberen, perfekt beschrifteten Bildern zu arbeiten. Hier sind einige praktische Faktoren, die du bei der Bewertung der KI-Modellrobustheit berücksichtigen solltest:

• Lichtveränderungen: Modelle können sich anders verhalten, wenn sie hellem Licht, schwachem Licht, Blendung oder Schatten ausgesetzt sind, was beeinflussen kann, wie sicher Objekte erkannt werden.
• Teilweise Verdeckung: In alltäglichen Szenen sind Objekte oft durch andere Gegenstände verdeckt oder nur teilweise sichtbar. Ein robusteres Modell ist in der Lage, sie auch bei fehlenden visuellen Informationen zu erkennen.
• Überfüllte Szenen: Umgebungen mit vielen überlappenden Objekten können die Erkennung erschweren. Modelle, die in diesen Fällen gut funktionieren, sind in komplexen Umgebungen in der Regel zuverlässiger.

Gute Ergebnisse bei sauberen, perfekt aufgenommenen Bildern führen nicht immer zu einer starken Leistung in der realen Welt. Regelmäßige Tests unter verschiedenen Bedingungen helfen zu zeigen, wie gut ein Modell nach der Bereitstellung besteht.

Was ist Data Augmentation?#

Die Art und Weise, wie ein Objekt auf einem Foto erscheint, kann sich je nach Beleuchtung, Winkel, Entfernung oder Hintergrund ändern. Wenn ein Computer-Vision-Modell trainiert wird, muss der Datensatz, von dem es lernt, diese Art von Variation enthalten, damit es in unvorhersehbaren Umgebungen gut funktionieren kann.

Data Augmentation erweitert einen Trainingsdatensatz, indem zusätzliche Beispiele aus den bereits vorhandenen Bildern erstellt werden. Dies geschieht durch gezielte Änderungen, wie etwa das Drehen oder Spiegeln eines Bildes, die Anpassung der Helligkeit oder das Zuschneiden eines Teils davon.

Stell dir zum Beispiel vor, du hast nur ein Foto einer Katze. Wenn du das Bild drehst oder die Helligkeit änderst, kannst du aus diesem einen Bild mehrere neue Versionen erstellen. Jede Version sieht etwas anders aus, ist aber immer noch ein Foto derselben Katze. Diese Variationen helfen dem Modell zu lernen, dass ein Objekt anders aussehen kann und trotzdem dasselbe ist.

Abb. 1. Einblick in die Augmentierung eines Katzenbildes (Source)

Wie Data Augmentation die Modellleistung verbessert#

Während des Modelltrainings kann Data Augmentation direkt in die Trainingspipeline eingebaut werden. Anstatt manuell neue Kopien von Bildern zu erstellen und zu speichern, können bei jedem Laden eines Bildes zufällige Transformationen angewendet werden.

Das bedeutet, dass das Modell bei jedem Durchgang eine leicht andere Version des Bildes sieht, sei es heller, gespiegelt oder teilweise verdeckt. Techniken wie Random Erasing können sogar kleine Bereiche des Bildes entfernen, um reale Situationen zu simulieren, in denen ein Objekt blockiert oder nur teilweise sichtbar ist.

Abb. 2. Beispiele für auf zufälligem Löschen basierende Augmentierung (Source)

Das Sehen vieler verschiedener Versionen desselben Bildes ermöglicht es dem Modell, zu lernen, welche Merkmale wichtig sind, anstatt sich auf ein einziges perfektes Beispiel zu verlassen. Diese Vielfalt baut die KI-Modellrobustheit auf, sodass es unter realen Bedingungen zuverlässiger funktionieren kann.

Gängige Data-Augmentation-Techniken#

Hier sind einige Data-Augmentation-Techniken, die verwendet werden, um Variationen in Trainingsbilder einzuführen:

• Geometrische Transformationen: Diese Techniken verändern das räumliche Erscheinungsbild eines Objekts in einem Bild. Durch Drehen, Spiegeln, Ändern der Größe, Zuschneiden oder Verschieben eines Bildes lernt das Modell, wie ein Objekt aus verschiedenen Blickwinkeln oder Entfernungen betrachtet werden kann.
• Farb- und Helligkeitsanpassungen: Reale Beleuchtung ist selten konsistent. Bilder können zu hell, zu dunkel oder farblich leicht verschoben sein, je nach Umgebung oder verwendeter Kamera. Das Anpassen von Helligkeit, Kontrast, Farbton und Sättigung ermöglicht es Modellen, diese visuellen Änderungen zu bewältigen und über verschiedene Szenen hinweg gut zu funktionieren.
• Variationen der Bildqualität: Unschärfe oder visuelles Rauschen können dazu führen, dass Bilder unklar aussehen. Das Hinzufügen von Unschärfe oder Rauschen während des Trainings hilft dem Modell, mit Bewegungsunschärfe, schlechten Lichtverhältnissen oder minderwertigen Kameraergebnissen umzugehen, sodass es weniger empfindlich gegenüber unvollkommenen Bildern wird.
• Verdeckungsbasierte Augmentierungen: In realen Umgebungen sind Objekte oft teilweise durch andere Objekte blockiert. Dies wird als Bildverdeckung bezeichnet. Das Ausblenden oder Maskieren kleiner Bereiche eines Bildes während des Trainings unterstützt das Modell dabei, Objekte auch dann zu erkennen, wenn nur ein Teil von ihnen sichtbar ist.
• Multi-Image-Augmentierungen: Diese Techniken kombinieren Teile mehrerer Bilder zu einem einzigen Trainingsbeispiel, was die Anzahl der sichtbaren Objekte erhöhen und die Fähigkeit des Modells verbessern kann, komplexe oder überfüllte Szenen zu bewältigen.

Abb. 3. Ein Beispiel für die Augmentierung mehrerer Bilder (Source)

Data Augmentation leicht gemacht mit dem Ultralytics Python Paket#

Das Verwalten von Datensätzen, Erstellen von Bildvariationen und Schreiben von Transformationscode kann zusätzliche Schritte bei der Erstellung einer Computer-Vision-Anwendung bedeuten. Das Ultralytics Python package hilft dabei, dies zu vereinfachen, indem es eine einzige Schnittstelle zum Trainieren, Ausführen und Bereitstellen von Ultralytics YOLO-Modellen wie YOLO26 bereitstellt. Als Teil dieses Bestrebens zur Optimierung von Trainings-Workflows enthält das Paket integrierte, von Ultralytics getestete Datenaugmentierungen, die für YOLO-Modelle optimiert sind.

Es unterstützt zudem nützliche Integrationen, die den Bedarf an separaten Tools oder benutzerdefiniertem Code überflüssig machen. Speziell für die Datenaugmentation lässt sich das Paket mit Albumentations integrieren, einer weit verbreiteten Bildaugmentierungsbibliothek. Diese Integration ermöglicht es, Augmentierungen während des Trainings automatisch anzuwenden, ohne zusätzliche Skripte oder benutzerdefinierten Code zu benötigen.

Verwaltung von Annotationen und augmentierten Datensätzen#

Ein weiterer Faktor, der die Modellrobustheit beeinflusst, ist die annotation-Qualität. Saubere, genaue Beschriftungen, die mit Annotationstools wie Roboflow erstellt und verwaltet werden, helfen dem Modell zu verstehen, wo sich Objekte befinden und wie sie aussehen.

Während des Trainings werden Data Augmentations wie Spiegeln, Zuschneiden und Drehen dynamisch angewendet und die Annotationen automatisch an diese Änderungen angepasst. Wenn die Labels präzise sind, funktioniert dieser Prozess reibungslos und liefert dem Modell viele realistische Beispiele derselben Szene.

Wenn Annotationen ungenau oder inkonsistent sind, können diese Fehler über die augmentierten Bilder hinweg wiederholt werden, was das Training weniger effektiv machen kann. Mit genauen Annotationen zu beginnen, verhindert, dass sich diese Fehler ausbreiten, und trägt zu einer besseren Modellrobustheit bei.

Verbesserung von Vision-KI-Anwendungen mit Data Augmentation#

Lass uns nun einige Beispiele durchgehen, wie Data Augmentation zur Robustheit von KI-Modellen in realen Anwendungen beiträgt.

Steigerung der Genauigkeit bei der Objekterkennung in realen Umgebungen#

Synthetische Bilder werden häufig zum Training von object detection-Systemen verwendet, wenn echte Daten rar, sensibel oder schwer zu erfassen sind. Sie ermöglichen es Teams, schnell Beispiele von Produkten, Umgebungen und Kameraperspektiven zu generieren, ohne jedes Szenario im echten Leben aufnehmen zu müssen.

Synthetische Datensätze können jedoch manchmal im Vergleich zu realem Filmmaterial zu sauber aussehen, wo sich Lichtverhältnisse ändern, Objekte überlappen und Szenen Hintergrundunordnung enthalten. Data Augmentation hilft, diese Lücke zu schließen, indem sie realistische Variationen einführt, wie z. B. unterschiedliche Beleuchtung, Rauschen oder Objektplatzierung, damit das Modell lernt, mit den Arten von Bedingungen umzugehen, denen es bei der Bereitstellung begegnen wird.

In einer aktuellen Studie wurde beispielsweise ein Ultralytics YOLO11-Modell vollständig mit synthetic images trainiert und Datenaugmentation hinzugefügt, um zusätzliche Variation einzubringen. Dies trug dazu bei, dass das Modell lernte, Objekte allgemeiner zu erkennen. Es schnitt bei Tests mit realen Bildern gut ab, obwohl es während des Trainings niemals reale Daten gesehen hatte.

Medizinische Bildgebungslösungen zuverlässiger machen#

Medizinische Bilddatensätze sind oft begrenzt, und die Scans selbst können je nach Gerätetyp, Bildeinstellungen oder klinischem Umfeld variieren. Unterschiede in der Anatomie des Patienten, Winkeln, Beleuchtung oder visuellem Rauschen können es für Computer-Vision-Modelle schwierig machen, Muster zu erlernen, die sich gut auf Patienten und Krankenhäuser übertragen lassen.

Data Augmentation hilft hierbei, indem sie während des Trainings mehrere Variationen desselben Scans erstellt, wie etwa das Hinzufügen von Rauschen, das leichte Verschieben des Bildes oder das Anwenden kleiner Verzerrungen. Diese Änderungen lassen die Trainingsdaten repräsentativer für reale klinische Bedingungen erscheinen.

Beispielsweise verwendeten Forscher in einer pediatric imaging-Studie Ultralytics YOLO11 für die anatomische Segmentierung und trainierten es mit augmentierten medizinischen Daten. Sie führten Variationen wie hinzugefügtes Rauschen, leichte Positionsverschiebungen und kleine Verzerrungen ein, um die Bilder realistischer zu gestalten.

Abb. 4. Originale und augmentierte medizinische pädiatrische Bilder (Source)

Durch das Lernen aus diesen Variationen konzentrierte sich das Modell auf bedeutungsvolle anatomische Merkmale anstatt auf oberflächliche Unterschiede. Dies machte seine Segmentierungsergebnisse über verschiedene Scans und Patienten hinweg stabiler.

Wichtige Erkenntnisse#

Das Sammeln vielfältiger Daten ist schwierig, aber Data Augmentation ermöglicht es Modellen, aus einem breiteren Spektrum visueller Bedingungen zu lernen. Dies führt zu einer stärkeren Modellrobustheit beim Umgang mit Verdeckungen, Lichtveränderungen und überfüllten Szenen. Insgesamt hilft dies ihnen, außerhalb kontrollierter Trainingsumgebungen zuverlässiger zu arbeiten.

Trete our community bei und entdecke das Neueste im Bereich Vision AI in unserem GitHub repository. Besuche unsere Lösungsseiten, um zu erfahren, wie Anwendungen wie AI in manufacturing und computer vision in healthcare den Fortschritt vorantreiben, und entdecke unsere licensing options, um deine nächste KI-Lösung zu unterstützen.