Yolo Vision Shenzhen
Shenzhen
Jetzt beitreten
Glossar

Feinabstimmung

Feinabstimmung von maschinellen Lernmodellen wie Ultralytics YOLO für bestimmte Aufgaben. Lernen Sie hier Methoden, Anwendungen und Best Practices kennen!

Feinabstimmung ist eine strategische Technik des maschinellen Lernens (ML), die dazu dient ein vorab trainiertes Modell für eine bestimmte Aufgabe oder einen bestimmten Datensatz zu spezialisieren. Anstatt von Grund auf zu trainieren, was große Mengen an markierten Daten und Rechenleistung erfordert, nutzt die Feinabstimmung das vorhandene Wissen eines Modells, das bereits Muster aus einem großen, allgemeinen Datensatz gelernt hat. Dieser Prozess ist eine praktische Anwendung des Transfer-Lernens und ermöglicht Nischenproblemen - wie der Erkennung seltener Fertigungsfehler oder der Klassifizierung medizinischer Bilder - mit deutlich reduziertem Aufwand eine hohe Leistung zu erzielen. medizinischer Bilder - mit deutlich reduzierten Ressourcen.

Wie Fine-Tuning funktioniert

Der Prozess beginnt mit einem Basismodell, wie wie ein Bildverarbeitungsmodell, das auf ImageNet oder ein Sprachmodell, das auf dem Internet-Korpus. Diese Modelle verfügen über ein solides Verständnis grundlegender Merkmale wie Kanten, Texturen und Formen in Bildern oder Grammatik und Semantik in Texten. Bei der Feinabstimmung wird das Modell mit einem neuen, kleineren benutzerdefinierten Datensatz, der für die Zielanwendung relevant ist.

Der Trainingsprozess beinhaltet die Anpassung der Modellgewichte leicht angepasst, um die Nuancen der neuen der neuen Daten anzupassen. Normalerweise geschieht dies mit einer niedrigeren Lernrate, um die wertvollen Merkmale zu erhalten zu erhalten, die in der anfänglichen Pre-Trainingsphase gelernt wurden, während das Modell sich noch anpassen kann. In vielen Computer Vision (CV)-Workflows können Ingenieure die frühen Schichten des Backbone, diegrundlegende Merkmale detect einfrieren Merkmale - einfrieren und nur eine Feinabstimmung der tieferen Schichten und Erkennungskopf, der für die endgültigen Vorhersagen.

Feinabstimmung mit Ultralytics YOLO

Die Anpassung eines modernen Modells wie Ultralytics YOLO11 an Ihre spezifischen Daten ist einfach. Die Bibliothek kümmert sich automatisch um die Komplexität des Ladens von vortrainierten Gewichte und die Konfiguration der Trainingsschleife.

Das folgende Beispiel zeigt, wie ein vortrainiertes YOLO11 geladen und anhand eines Beispieldatensatzes feinabgestimmt werden kann.

from ultralytics import YOLO

# Load a pre-trained YOLO11 model (uses COCO weights by default)
model = YOLO("yolo11n.pt")

# Fine-tune the model on a specific dataset configuration
# 'epochs' sets the training duration, and 'imgsz' defines input resolution
results = model.train(data="coco8.yaml", epochs=20, imgsz=640)

Anwendungsfälle in der Praxis

Die Feinabstimmung schließt die Lücke zwischen allgemeinen KI-Fähigkeiten und branchenspezifischen Anforderungen.

  • KI im Gesundheitswesen: Allgemein Bildverarbeitungsmodelle können Alltagsobjekte wie Autos oder Katzen identifizieren. Durch die Feinabstimmung dieser Modelle auf kommentierte medizinischen Bildanalysedaten wie Röntgenaufnahmen Röntgenaufnahmen oder MRT-Scans, können Ärzte spezialisierte Tools entwickeln, um bestimmte detect wie Lungenentzündung oder Hirntumoren mit hoher Genauigkeit erkennen.
  • Industrielle Fertigung: In Produktionslinien erkennt ein Standardmodell möglicherweise keine firmeneigenen Komponenten oder subtile Fehler. Durch die Feinabstimmung eines Modells auf Bilder der spezifischen Montagelinie können Hersteller die Qualitätskontrolle Qualitätskontrolle automatisieren, um Defekte wie Mikrorisse oder Ausrichtungsfehler zu erkennen und so die Effizienz erheblich zu steigern.

Feinabstimmung vs. verwandte Konzepte

Die Unterscheidung zwischen Feinabstimmung und anderen Anpassungsmethoden ist entscheidend für die Wahl des richtigen Ansatzes:

  • Merkmalsextraktion: Bei diesem Ansatz wird das vortrainierte Modell als fester Merkmalsextraktor behandelt. Die Gewichte werden vollständig eingefroren, und nur ein einfacher Klassifikator (wie ein lineares Regressionsmodell ) wird auf den Ausgaben den Ausgaben trainiert. Die Feinabstimmung geht einen Schritt weiter, indem die internen Parameter des Modells aktualisiert werden.
  • Parametereffiziente Feinabstimmung (PEFT): Während die Standard-Feinabstimmung alle oder die meisten Modellparameter aktualisiert, aktualisieren PEFT-Methoden wie LoRA (Low-Rank Adaptation) nur eine kleine kleine Teilmenge der Parameter. Dies macht den Prozess viel schneller und weniger speicherintensiv, insbesondere bei großen Große Sprachmodelle (LLMs).
  • Prompt Engineering: Diese umfasst die Erstellung spezifischer Eingaben (Prompts), um die Ausgabe eines eingefrorenen Modells zu steuern, ohne dessen interne Gewichte zu ändern oder es mit neuen Daten zu trainieren.

Tools und Frameworks

Um die Feinabstimmung zu implementieren, verlassen sich die Entwickler auf robuste Frameworks wie PyTorch und TensorFlowdie die notwendige Infrastruktur für Gradientenabstieg und Backpropagation bieten. Moderne Bibliotheken rationalisieren dies weiter; so ermöglicht das Ultralytics eine nahtlose Modelltraining und -validierung. Bei der Vorbereitung der Daten für die Feinabstimmung ist die Sicherstellung einer hochwertigen Datenbeschriftung wichtig, um eine um eine Verzerrung des Datensatzes zu verhindern, die die Leistung des Modells in realen Szenarien verzerren kann.

Werden Sie Mitglied der Ultralytics

Gestalten Sie die Zukunft der KI mit. Vernetzen Sie sich, arbeiten Sie zusammen und wachsen Sie mit globalen Innovatoren

Jetzt beitreten