Die Bilddaten für diese Studie wurden aus den Betriebsumgebungen der Stromversorgung mit den erforderlichen Genehmigungen erhoben. Alle Bilder wurden anonymisiert, ohne dass personenbezogene Daten gespeichert wurden. Diese Studie konzentriert sich darauf, persönliche Schutzausrüstung zu erkennen, anstatt Personen zu identifizieren. Da die Forschung ausschließlich die Entwicklung von Algorithmen unter Verwendung anonymisierter Daten umfasst, war keine ethische Genehmigung erforderlich.
Das folgende Protokoll beschreibt ein umfassendes Verfahren zur Entwurf, Schulung und Bewertung eines leichten, leistungsstarken und von Natur aus interpretierbaren Objekterkennungsmodells namens WTLS-YOLOv11n zur Erkennung von persönlicher Schutzausrüstung (PSA). Die in dieser Studie verwendete Software ist in der Materialtabelle aufgeführt.
Gesamtarchitektonischer Rahmen
Das vorgeschlagene WTLS-YOLOv11n-Modell basiert auf dem YOLOv11n-Basismodell. Die Kernmethodik besteht darin, zentrale Module im Rückgrat und im Kopf systematisch zu ersetzen oder zu verbessern, um Leistung, Effizienz und Interpretierbarkeit zu verbessern.
Das vorgeschlagene WTLS-YOLOv11n-Modell basiert auf dem YOLOv11n-Basismodell. Die Kernmethodik besteht darin, zentrale Module im Rückgrat und im Kopf systematisch zu ersetzen oder zu verbessern, um Leistung, Effizienz und Interpretierbarkeit zu verbessern. Das übergeordnete architektonische Paradigma von YOLOv11n, das aus Rückgrat, Nacken und Kopf besteht, wurde im vorgeschlagenen Modell beibehalten. Im Backbone wurden spezifische C3K2-Module durch die vorgeschlagenen C3K2-WTConv-Module ersetzt, um die Feature-Extraktion zu verbessern. Im Kopf wurde der ursprüngliche Detektionskopf durch den vorgeschlagenen Lightweight Shared Composite Detection Head (LSCD) ersetzt, um die Modellkomplexität zu reduzieren. Die vollständige Architektur des vorgeschlagenen WTLS-YOLOv11n-Modells, im Gegensatz zur Basislinie, ist in Abbildung 1 dargestellt.
C3K2-WTConv-Moduldesign
Dieses Modul ist darauf ausgelegt, die Standard-Faltungsmodule im YOLOv11n-Backbone zu ersetzen. Sein Design wird von zwei Hauptzielen geleitet: (a) das rezeptive Feld des Modells effizient zu erweitern, um kontextuelle Informationen in mehreren Maßstäben zu erfassen, ohne die Parameteranzahl oder Rechenkomplexität wesentlich zu erhöhen, und (b) robustere und grundsätzlich interpretierbare Merkmalsrepräsentationen zu lernen, indem Merkmalsabbildungen explizit in den Frequenzbereich zerlegt werden.
Design der Core WTConv-Schicht
Die grundlegende WTConv-Schicht (Wavelet Transform Convolution) ist darauf ausgelegt, die 2D-diskrete Haar-Wavelet-Transformation umzusetzen. Dieser Prozess wird durch eine Reihe spezifischer Wavelet-Kernel durchgeführt:
Wavelet-Kerne: Die Transformation wird über vier feste, nicht trainierbare Tiefenfaltungskerne (F_LL, F_LH, F_HL, F_HH) implementiert, die den Haar-Wavelet-Basisfunktionen entsprechen. Diese Kernel sind dafür verantwortlich, die Eingabefeature-Abbildung in ihre Nieder- und Hochfrequenzkomponenten zu zerlegen.
Dekomposition und Downsampling: Diese Kernel werden auf die Eingabefeature-Abbildung (X) mit einem Schritt 2 angewendet. Diese einzelne Operation führt effizient sowohl die Zerlegung der Merkmale als auch das räumliche Downsampling durch und teilt die Eingabe in vier verschiedene Unterbänder auf.
Physikalische Interpretation der Ausgangskomponenten
Die vier durch die Wavelet-Zerlegung erzeugten Feature-Unterbänder besitzen eine klare physikalische Interpretation. Wie in Abbildung 2 dargestellt, zerlegt die WTConv-Schicht die Eingabe rekursiv in folgende Komponenten:
Niederfrequente Komponente (LL): Diese Komponente bewahrt die Gesamtstruktur, Kontur und andere globale Informationen des Ziels bei halber räumlicher Auflösung. Sie dient als Grundlage für das Modell, um die "Form" des Ziels zu verstehen.
Hochfrequenzkomponenten (LH, HL, HH): Diese drei Komponenten erfassen feinkörnige Details wie horizontale, vertikale und diagonale Kanten bzw. Texturen. Sie ermöglichen es dem Modell, sich auf die "Details" des Ziels zu konzentrieren.
Integration in die C3K2-Modulstruktur
Die entworfene WTConv-Schicht ist nahtlos in die C3K2-Flaschenhalsstruktur von YOLOv11n integriert.
Ersatzstrategie: Im ursprünglichen C3K2-Modul wird die Standard-3x3-Faltungsschicht durch die WTConv-Schicht ersetzt. Dieser Ansatz bewahrt den effizienten Mechanismus der Wiederverwendung von Funktionen der C3K2-Architektur und führt dabei die Vorteile der Wavelet-Transformation ein, was zum endgültigen C3K2-WTConv-Modul führt (detaillierte Struktur in Abbildung 3).
Kaskadierte rezeptive Felderweiterung: Der WTConv-Prozess kann rekursiv auf den Niederfrequenzausgang (LL) der vorhergehenden Stufe angewendet werden. Dieser kaskadierte Zerlegungsmechanismus ermöglicht es dem Modell, Merkmale über einem exponentiell wachsenden rezeptiven Feld mit minimalem Rechenaufwand zu analysieren und so einen effizienten Multi-Skala-Frequenzdekompositionspfad zu schaffen.
Lightweight Shared Composite Detection Head (LSCD) Design
Dieses Modul wurde entwickelt, um den ursprünglichen YOLOv11n-Detektionskopf zu ersetzen, mit dem Hauptziel, die Modellkomplexität und den Rechenaufwand drastisch zu reduzieren, um eine effiziente Bereitstellung auf ressourcenbegrenzten Edge-Geräten zu ermöglichen. Das Design berücksichtigt die signifikante Parameterredundanz, die in Standard-Multi-Scale-Detektionsköpfen gefunden wird (detaillierte Struktur in Abbildung 4 dargestellt.
Begründung und Entwurfsziele
Die unabhängigen Vorhersagezweige für jede Merkmalsskala (P3-P5) im ursprünglichen YOLOv11-Kopf führen zu einer hohen Parameteranzahl. Das LSCD führt eine hybride Parameter-Sharing-Strategie ein, um eine überlegene Parametereffizienz zu erreichen und dabei kritische Multi-Scale-Feature-Fusionsfähigkeiten zu erhalten.
Parameter-Sharing und Mechanismus zur Funktionsfusion
Der Kern des LSCD liegt in seiner zweistufigen Feature-Processing-Pipeline:
Skalenspezifische Vorverarbeitung: Für jede Eingabefeature-Map vom Hals (P3, P4, P5) wird eine nicht geteilte 1x1-Faltung gefolgt von einer Group Normalization (GN)-Schicht angewendet. Dieser erste Schritt ermöglicht es dem Netzwerk, skalierungsspezifische Kanaltransformationen zu erlernen und sicherzustellen, dass die einzigartigen Eigenschaften jedes Merkmalsniveaus vor der Fusion erhalten bleiben.
Effiziente Kreuz-Skala-Fusion: Nach der Vorverarbeitung werden eine Reihe gemeinsamer 3x3-Faltungs-GN-Module eingesetzt, um die Kernfunktionsfusion über verschiedene Skalen hinweg durchzuführen. Durch das Teilen von Gewichten lernen diese Module ein verallgemeinertes Merkmalfusionsmuster, das die Hauptquelle für die Parameterreduktion ist. Dieses Design zwingt das Modell dazu, robustere und universellere Fusionsdarstellungen zu lernen.
Dynamische Skalenanpassung und Verzweigungsoptimierung
Um möglichen Informationsverlust durch Gewichtsteilung auszugleichen und die Vorhersagegenauigkeit zu verfeinern, werden zwei zusätzliche Mechanismen eingeführt:
Learnable Scale Layer: Für jede Detektionsskala wird eine learnable Scale-Ebene hinzugefügt. Diese Schicht führt einen per-maßstabsfähigen lernbaren Skalar ein, der die fusionierten Merkmale dynamisch neu gewichtet, sodass das Modell Merkmale adaptiv betonen oder unterdrücken kann, basierend auf den auf diesem Maßstab vorherrschenden Zielobjektgrößen.
Optimierter Klassifikationszweig: Der Klassifikationszweig wird durch die Verwendung der Softmax-Aktivierungsfunktion für die Wahrscheinlichkeitsverteilung und die Einbindung einer Gruppennormalisierungsschicht (GN) erweitert. Dies stabilisiert das Training der Klassifikationsaufgabe und verbessert die Robustheit der Konfidenzvorhersagen, eine Technik, die sich in Architekturen wie FCOS als wirksam erwiesen hat.
Verbessertes Design der Verlustfunktion (MPDIoU)
Die Verlustfunktion wurde speziell neu entwickelt, um die Herausforderungen der genauen Begrenzungsbox-Regression für kleine, überladene und unregelmäßig geformte Ziele zu adressieren, die in PSA-Detektionsszenarien häufig vorkommen.
Motivation und Einschränkungen des traditionellen IoU-Verlusts
Standard-IoU-basierte Verluste leiden in diesem Zusammenhang unter mehreren kritischen Nachteilen:
Verschwindende Gradienten: Wenn die vorhergesagten und die Ground-Truth-Boxen keine Überschneidung haben, ist das IoU null und der Verlustgradient verschwindet, wodurch der Lernprozess ins Stocken kommt.
Unempfindlichkeit gegenüber der Ausrichtung: Mehrere Begrenzungsbox-Konfigurationen können denselben IoU-Wert ergeben, wodurch die Verlustfunktion unempfindlich gegenüber der Qualität der Ausrichtung ist (z. B. Mittelpunktabweichung vs. Formungleichheit).
Schlechte Leistung bei kleinen Objekten: Bei kleinen Zielen können selbst geringfügige Pixelabweichungen erheblich sein, führen aber oft zu vernachlässigbaren Änderungen des IoU-Werts, was zu ungenauer Lokalisierung führt.
Einführung des Verlusts der Mindestpunktentfernung IoU (MPDIoU)
Um die oben genannten Einschränkungen zu überwinden, ersetzt das Protokoll den Standardverlust durch den Minimum Point Distance IoU (MPDIoU)-Verlust. MPDIoU erweitert die Standard-IoU-Metrik, indem es einen Strafterm einbaut, der direkt den Abstand zwischen den vorhergesagten und den Ground-Truth-Boxen bestraft, selbst wenn sie sich nicht überlappen.
Kernmechanismus: Der Strafterm leitet sich aus dem euklidischen Abstand zwischen den entsprechenden Eckpunkten der vorhergesagten Box (pred) und der Ground-Truth-Box (gt) ab. Konkret werden die quadrierten Abstände für die oben linken Ecken
) und die unten rechten Ecken (
) berechnet:
(1)
(2)
Um sicherzustellen, dass die Strafe skaleninvariant ist, wird dieser Abstand durch die Maße der kleinsten umschließenden Box normalisiert, die sowohl die vorhergesagte als auch die Ground-Truth-Box abdeckt. Sei w und h die Breite bzw. Höhe dieser umschließenden Kiste. Die MPDIoU-Metrik wird dann wie folgt formuliert:
(3)
Schließlich ist die MPDIoU-Verlustfunktion (LMPDIoU) definiert als:
LMPDIoU = 1 - MPDIoU (4)
Wichtige Vorteile: Dieser geometrische Strafbegriff adressiert die oben genannten Probleme direkt: i) einen aussagekräftigen, von null verschiedenen Gradienten zu schaffen, auch wenn Boxen sich nicht überlappen, wodurch das Verschwinden des Gradienten verhindert und eine kontinuierliche Modelloptimierung sichergestellt wird, ii) eine erhöhte Empfindlichkeit gegenüber Abweichungen von Position, Größe und Seitenverhältnis bietet, was für die präzise Lokalisierung kleiner und vielfältiger PSA-Elemente entscheidend ist, iii) stabilere und konsistentere Gradientensignale während des gesamten Trainings liefern, was eine schnellere Konvergenz fördert und zu einer überlegenen Lokalisierungsgenauigkeit führt.
Datensätze und experimentelle Einrichtung
Selbsterstellter PSA-Erkennungsdatensatz
Wir haben einen PSA-Detektionsdatensatz mit 5.000 hochauflösenden Bildern von Stromübertragungsleitungen erstellt, mit LabelImg annotiert und in das YOLO-TXT-Format konvertiert. Der Datensatz umfasst fünf Kategorien: Schutzhelme (1.245 Fälle), Sicherheitsgurte (1.128 Fälle), Armbänder (892 Fälle), Arbeitsstatus in Höhe (1.056 Fälle) und Status auf Bodenniveau (1.124 Fälle). Die Daten werden in Trainingssätze (4.000 Bilder), Validierung (500 Bilder) und Testsätze (500 Bilder) aufgeteilt. Die Bilder zeigen herausfordernde Bedingungen, darunter extreme Beleuchtung, starke Verdeckung (23 % mit >50 % Abdeckung) und komplexe industrielle Hintergründe. Standard-Augmentationstechniken (zufälliges Umdrehen, Drehen, Farbjitter, Mosaik) wurden während des Trainings angewendet.
PASCAL VOC-Datensatz für Generalisierungstests
Um die Verallgemeinerungsfähigkeit des Modells über Energiebetriebsszenarien hinaus zu bewerten, verwendeten wir den PASCAL VOC-Datensatz, der VOC2007 und VOC2012 kombiniert und insgesamt 21.503 Bilder umfasst. Gemäß der Standardpraxis verwendeten wir 11.540 Bilder von VOC2012 für Schulung und Validierung sowie 9.963 Bilder von VOC2007 für Tests. Der VOC-Datensatz enthält 20 Objektkategorien in verschiedenen realen Szenarien. Obwohl die spezifischen Objekte von unseren PSA-Kategorien abweichen, sind die Erkennungsherausforderungen (Skalenvariation, Okklusion, komplexe Hintergründe) ähnlich, was es geeignet macht, die allgemeinen Detektionsfähigkeiten und Übertragbarkeit des Modells zu bewerten.
Implementierungsdetails
Dieses Protokoll beschreibt das Verfahren zum Training und zur Bewertung des Modells. Es wird angenommen, dass der Benutzer Zugriff auf den Quellcode, eine geeignete Python-Umgebung und die vorbereiteten Datensätze hat.
System- und Umweltvorbereitung
Eine Workstation mit einer NVIDIA-GPU mit mindestens 24 GB VRAM wurde verwendet, um ausreichend Speicher für das Training sicherzustellen (z. B. NVIDIA GeForce RTX 4090). Das PyTorch-Deep-Learning-Framework (Version 1.12.0 oder höher) zusammen mit dem entsprechenden CUDA-Toolkit wurde auf der Workstation installiert. Die Installation wurde überprüft, indem ein Terminal geöffnet und der Befehl "python -c'import torch; print(torch.cuda.is_available())'", was voraussichtlich "True" zurückgeben würde. Erforderliche Python-Pakete wurden installiert, indem man zum Projekt-Root-Verzeichnis navigierte und den Befehl "pip install -r requirements.txt" ausführte. Dieser Befehl installierte automatisch Abhängigkeiten wie numpy, opencv-python, pyyaml, tensorboard und torchvision. Die Datensatzvorbereitung wurde überprüft, um sicherzustellen, dass Trainingsbilder in /data/train/images/ und Annotationsdateien in /data/train/labels/ im YOLO-Format (Klasse x_center y_center Breite Höhe) lagen. Der gleiche Verifikationsprozess wurde auf die Validierungs- (/data/val/) und Testdatensätze (/data/test/) angewendet.
Ausbildungskonfiguration
Die Konfigurationsdatei config/wtls_yolov11n.yaml wurde mit einem Texteditor geöffnet, um die Trainingsparameter einzurichten. Datenpfade wurden konfiguriert, indem der Parameter 'path' gefunden und in das Root-Verzeichnis des Datensatzes gesetzt wurde, während die Parameter 'train', 'val' und 'test' überprüft wurden, um auf die richtigen Unterverzeichnisse zu verweisen. Der 'nc'-Parameter (Anzahl der Klassen) wurde bestätigt, um mit dem Datensatz übereinzustimmen, der für die PPE-Erkennung auf 5 gesetzt wurde. Die Trainingshyperparameter wurden so konfiguriert, dass die Anzahl der Epochen auf 300 für das vollständige Training und die Batchgröße auf 16 gesetzt war, mit dem Verständnis, dass dieser Wert basierend auf der GPU-Speicherverfügbarkeit angepasst und bei Ausfallfehlern auf 8 reduziert werden konnte. Die Eingabebildgröße (imgsz) wurde auf 640 gesetzt, und der Optimierer wurde als SGD mit einer anfänglichen Lernrate (lr0) von 0,01 bestätigt. Der Gewichtsabfall wurde auf 0,0005 bestätigt und der Impuls auf 0,937 gesetzt. Datenaugmentationstechniken wurden mit den standardmäßig konfigurierten Einstellungen aktiviert, die Mosaik-Augmentation (Mosaik: 1,0), zufälligen horizontalen Flip mit 50 % Wahrscheinlichkeit und Farbjitter mit den Parametern HSVH: 0,015, HSVS: 0,7 und HSVV : 0,4 umfassten. Die Hardware-Auslastungsparameter wurden so konfiguriert, dass die Anzahl der Arbeiter für CPU-Threads beim Datenladen auf 8 gesetzt wurde und der Geräteparameter auf 0 gesetzt wurde, um die erste GPU zu verwenden, mit der Möglichkeit, ihn bei Bedarf für Multi-GPU-Training auf "0,1" zu setzen.
Modelltrainingsausführung
Das Modelltraining wurde über die Kommandozeile initialisiert, indem der Befehl "python train.py --cfg config/wtls_yolov11n.yaml --weights ''--data data.yaml" ausgeführt wurde, wobei der --cfg-Parameter die Modellkonfigurationsdatei angab, der --weights-Parameter auf eine leere Zeichenkette gesetzt wurde, um von Grund auf zu trainieren (alternativ konnte yolov11n.pt für Transfer Learning verwendet werden), und der --data-Parameter die Datensatzkonfiguration spezifizierte. Während des Trainings wurden Konvergenzindikatoren beobachtet, um die Modellleistung zu verfolgen. In den ersten 50 Epochen wurde ein rascher Rückgang des Verlusts beobachtet, wobei box_loss von etwa 1,5 auf 0,8 absank. Zwischen Epochen 50 und 150 wurde eine stetige Verbesserung festgestellt, da box_loss von etwa 0,8 auf 0,5 abnahm. In den Epochen 150 bis 300 fand die Feinabstimmungsphase statt, wobei box_loss sich um 0,4 bis 0,5 stabilisierte. Die Validierungsmetrik mAP@0,5 sollte bis Epoche 200 über 85 % erreichen. Das Checkpoint-Speichern wurde überprüft, um sicherzustellen, dass das Training Checkpoints alle 10 Epochen automatisch im Verzeichnis runs/train/exp/weights/speicherte. Das Vorhandensein von last.pt (letzter Checkpoint) und best.pt (höchster mAP-Checkpoint) wurde bestätigt, wobei jede Checkpoint-Datei etwa 5 bis 6 MB groß sein sollte. Der Trainingsfortschritt konnte optional mit TensorBoard überwacht werden, indem ein neues Terminal geöffnet und der Befehl "tensorboard --logdir runs/train" ausgeführt wurde, und dann durch einen http://localhost:6006 Browser navigiert wurde, um Echtzeitdiagramme von Verlustkurven, Lernratenplan und mAP-Trends anzuzeigen. Häufige Probleme wurden durch Fehlerbehebungsverfahren behoben, bei denen CUDA-Out-of-Memory-Fehler durch Reduzierung der Batchgröße auf 8 oder 4 behoben wurden, NaN-Verlustwerte durch Reduzierung der Lernrate auf 0,005 und mAP-Plateaus unter 80 % durch Überprüfung der Annotationskorrektheit und Erhöhung der Trainingsepochen auf 400 untersucht wurden.
Modellbewertungsprotokoll
Ein vielschichtiges Bewertungsverfahren wurde durchgeführt, um die Wirksamkeit, Effizienz und Interpretierbarkeit des vorgeschlagenen Modells umfassend zu validieren.
Quantitative Leistungsbewertung
Leistungskennzahlen
Bewerten Sie die Modellgenauigkeit anhand der mittleren Durchschnittsgenauigkeit (mAP) bei einem IoU-Schwellenwert von 0,5 (mAP@0,5), Präzision und Abruf. Bewertung der Modelleffizienz durch Messung der Gesamtzahl der Parameter (M) und Gleitkommaoperationen (FLOPs, G). Messen Sie die Inferenzgeschwindigkeit in Frames Per Sekunde (FPS) sowohl auf einer Server-GPU als auch auf einem Edge-Gerät. Für Edge-Geräte-Tests stellen Sie den Energiemodus auf maximale Leistung ein und erwärmen Sie das Modell mit 100 Dummy-Inferenzen, bevor Sie FPS-Aufnahmen von über 500 Testbildern aufnehmen.
Vergleich mit modernen Modellen: Vergleichen Sie das vorgeschlagene WTLS-YOLOv11n-Modell mit seinem direkten Baseline (YOLOv11n) und einer vielfältigen Palette von Detektoren, darunter gängige CNN-basierte YOLO-Modelle (YOLOv5n, YOLOv8n, YOLOv9s), einen zweistufigen Detektor (Faster R-CNN) und einen transformatorbasierten Detektor (RT-DETR). Trainiere alle Modelle für 300 Epochen mit ihren empfohlenen Standard-Hyperparametern. Erfassen Sie alle im Abschnitt Leistungskennzahlen für jedes Modell definierten Kennzahlen sowohl im selbsterstellten PPE-Datensatz als auch im PASCAL-VOC-Datensatz. Für die VOC-Bewertung trainieren Sie auf der kombinierten VOC2012 Trainval- und VOC2007 Trainval-Sets und testen Sie dann auf VOC2007 Testsatz gemäß dem Standardprotokoll.
Ablationsstudien: Führen Sie eine systematische Ablationsstudie durch, um die individuellen Beiträge der vorgeschlagenen Komponenten zu analysieren.
Analyse der Komponentenwirksamkeit: Ausgehend vom Basismodell YOLOv11n integrieren Sie schrittweise das C3K2-WTConv-Modul, den LSCD-Kopf und den MPDIoU-Verlust. Für jede Konfiguration wird das Modell für 300 Epochen mit identischen Hyperparametern (Batchgröße 16, Lernrate 0,01, SGD-Optimierer) neu trainiert. Zeichnen Sie die Änderungen in mAP, Präzision, Rückruf, Parametern, FLOPs und FPS auf, um die Wirksamkeit jeder Komponente zu überprüfen. Berechnen Sie prozentuale Verbesserungen im Verhältnis zum Ausgangswert für jede Kennzahl.
Platzierungsanalyse: Um die optimale Platzierung des C3K2-WTConv-Moduls zu bestimmen, integrieren Sie es in verschiedene Teile der Netzwerkarchitektur. Testen Sie drei Konfigurationen: (i) Nur-Backbone-Integration, (ii) Nacken-Only-Integration und (iii) vollständige Integration sowohl in Backbone als auch in Nacken. Trainieren Sie jede Konfiguration über 300 Epochen und vergleichen Sie die resultierenden mAP@0,5-Werte, um die effektivste Integrationsstrategie zu identifizieren. Wählen Sie die optimale Konfiguration für alle nachfolgenden Experimente aus.
Qualitative Leistungsbewertung
Erkennungsergebnis-Visualisierung: Wählen Sie 12–15 repräsentative Bilder aus dem Testset aus, die herausfordernde reale Szenarien zeigen, darunter starke Hintergrundbeleuchtung, starke Objektverdeckung, komplexe Hintergründe und unkonventionelle Kamerawinkel. Für jedes Vergleichsmodell (Baseline, YOLOv5n, YOLOv8n, YOLOv9s und WTLS-YOLOv11n) wird auf diesen Bildern mit einer Konfidenzschwelle von 0,25 und einer IoU-Schwelle von 0,45 abgeleitet. Erzeugen und rendern Sie die Erkennungsausgaben (Begrenzungsboxen mit Klassenlabels und Konfidenzwerten) auf diese Bilder. Ordne Visualisierungen in einem Rasterformat an, bei dem Zeilen verschiedene Szenarien und Spalten unterschiedliche Modelle darstellen.
Robustheitsanalyse: Vergleichen Sie visuell die gerenderten Erkennungsergebnisse verschiedener Modelle. Bewerten Sie die Robustheit und Überlegenheit des vorgeschlagenen Modells, indem Sie Fälle von falschen Negativen (verpasste Objekte), falsch positiven (falsche Erkennungen) und doppelten Erkennungen (mehrere Boxen für ein einzelnes Objekt mit IoU > 0,7 zwischen den Vorhersagen) zählen und markieren. Verwenden Sie rote Kreise, um problematische Erkennungen in der Visualisierungsfigur hervorzuheben. Berechnen Sie pro Modell Fehlnegativraten und Falsch-Positiv-Zählungen über die ausgewählten Testbilder. Extrahieren und vergleichen Sie die mittleren Konfidenzwerte für echte positive Detektionen zwischen den Modellen.
Interpretierbarkeitsanalyse
Um die inhärente Interpretierbarkeit des C3K2-WTConv-Moduls zu validieren, wurde ein Visualisierungsverfahren an ausgewählten Eingabebildern durchgeführt. Das trainierte WTLS-YOLOv11n-Modell wurde geladen und ein Forward Hook wurde am C3K2-WTConv-Modul an der vierten Schicht des Backbone registriert. Die Vorwärtsausbreitung wurde auf dem Eingabebild durchgeführt, und der Ausgabe-Merkmalstensor wurde erfasst. Aus dem erfassten Merkmalstensor der Form [Batch, Kanäle, Höhe, Breite] wurden Kanäle, die verschiedenen Frequenzkomponenten entsprechen, getrennt. Die ersten 25 % der Kanäle wurden als Niederfrequenzkomponente (LL) bezeichnet, die restlichen 75 % als Hochfrequenzkomponenten (LH, HL, HH). Für jeden Frequenzkomponententyp wurde der Durchschnitt über alle Kanäle innerhalb dieser Komponente berechnet, um Einzelkanal-Aktivierungskarten zu erstellen. Die L2-Norm wurde bei Bedarf über räumliche Dimensionen hinweg angewendet, um starke Aktivierungen zu betonen. Die Aktivierungskarten wurden auf den Bereich [0, 1] normalisiert und eine Farbkarte (z. B. 'Jet'-Farbkarte) angewendet. Die Heatmaps wurden durch bilineare Interpolation auf die ursprüngliche Bildauflösung angepasst. Die Niederfrequenz-Heatmap und die Hochfrequenz-Heatmap wurden mit 40 % Transparenz auf das Originalbild gelegt, um interpretierbare Visualisierungen zu erstellen, die zeigen, wo das Modell sich auf strukturelle versus texturale Merkmale konzentriert.
Um rigoros zu validieren, dass Erkennungsentscheidungen von Frequenzbereichsmerkmalen abhängen, wurde eine quantitative Analyse durchgeführt. Eine Teilmenge von 200 bis 300 Bildern wurde aus dem Testsatz ausgewählt, die erfolgreiche Erkennungen enthielten (Konfidenzwert größer als 0,5). Diese Teilmenge wurde so gesichert, dass sie vielfältige Szenarien darstellt, um Stichprobenverzerrungen zu vermeiden. Für jedes Bild in der Teilmenge wurde eine Vorwärtspropagation durchgeführt, und die Ausgabe des C3K2-WTConv-Moduls auf der vierten Backbone-Schicht wurde extrahiert. Der Merkmalstensor wurde in Niederfrequenzkomponenten (LL, erste 25 % der Kanäle) und Hochfrequenzkomponenten (HF, verbleibende 75 % der Kanäle) unterteilt. Für jedes Bild wurde die mittlere absolute Aktivierungsstärke über die räumlichen Dimensionen (Höhe und Breite) für beide Frequenzkomponenten berechnet. Die LL-Stärke wurde als Mittelwert des Absolutwerts der LL-Merkmale über alle räumlichen Positionen berechnet, und die HF-Stärke als Mittelwert des Absolutwerts der HF-Merkmale über alle räumlichen Positionen. Diese Werte wurden zusammen mit dem maximalen Erkennungskonfidenzwert für dieses Bild erfasst. Mit den gesammelten Daten (LL_strength, HF_strength, confidence_score) über alle Bilder hinweg wurden Pearson-Korrelationskoeffizienten berechnet. Die Korrelation zwischen LL-Stärke und Vertrauensscore wurde berechnet, ebenso wie die Korrelation zwischen HF-Stärke und Vertrauenswert. Statistische Software oder Pythons Funktion scipy.stats.pearsonr wurden verwendet, um sowohl Korrelationskoeffizienten (r) als auch p-Werte zu erhalten. Die statistische Signifikanz wurde anhand eines p-Schwellenwerts von 0,05 bewertet, wobei ein p-Wert unter 0,05 anzeigte, dass die Korrelation statistisch signifikant war. Die Größe der Korrelationskoeffizienten wurde verglichen, um zu bestimmen, welche Frequenzkomponente eine stärkere Verbindung mit der Erkennungskonfidenz hatte. Die folgenden Metriken wurden in strukturiertem Format erfasst: LL-Korrelationskoeffizient (rLL) und p-Wert (pLL), HF-Korrelationskoeffizient (rHF) und p-Wert (pHF), mittlere Aktivierungsstärken (meanLL, meanHF) und Stichprobengröße (Anzahl der analysierten Bilder). Die erwarteten Ergebnisse waren, dass niederfrequente Merkmale eine stärkere positive Korrelation mit Konfidenzwerten (rLL größer als 0,60, p unter 0,001) als hochfrequente Merkmale (rHF etwa 0,40 bis 0,50, p unter 0,01) aufweisen würden, was darauf hindeutet, dass Erkennungsentscheidungen überwiegend von strukturellen Informationen aus der LL-Komponente bestimmt wurden.
Diese quantitative Analyse etablierte eine Interpretierbarkeit jenseits der Beschränkungen qualitativer Visualisierung. Die Kombination aus Korrelationsanalyse (die statistische Assoziation demonstriert) und Frequenzdomänenvisualisierung (Anzeige räumlicher Aufmerksamkeit) lieferte rigorose empirische Belege dafür, dass die Entscheidungsfindung des Modells tatsächlich auf niedrigfrequenten strukturellen Merkmalen beruhte, wie es vom Design beabsichtigt war.
Erwartete Ergebnisse: Niederfrequente Merkmale sollten eine stärkere positive Korrelation mit Konfidenzwerten (rLL > 0,60, S. < 0,001) im Vergleich zu hochfrequenten Merkmalen (rHF≈ 0,40–0,50, S. < 0,01) aufweisen, was darauf hindeutet, dass Erkennungsentscheidungen überwiegend von strukturellen Informationen aus der LL-Komponente getroffen werden.
Diese quantitative Analyse stellt eine Interpretierbarkeit jenseits qualitativer Visualisierung dar. Die Kombination aus Korrelationsanalyse (Nachweis statistischer Assoziation) und Frequenzdomänenvisualisierung (Anzeige räumlicher Aufmerksamkeit) liefert rigorose empirische Belege dafür, dass die Entscheidungsfindung des Modells tatsächlich auf niederfrequenten strukturellen Merkmalen beruht, wie es vom Design beabsichtigt ist.