HyperAIHyperAI

Command Palette

Search for a command to run...

Hierarchische Textklassifizierung mit verstärktem Label-Zuordnung

Yuning Mao Jingjing Tian Jiawei Han Xiang Ren

Zusammenfassung

Während bestehende Methoden der hierarchischen Textklassifizierung (HTC) versuchen, Labelhierarchien für das Modelltraining zu erfassen, treffen sie entweder lokale Entscheidungen bezüglich jedes Labels oder ignorieren die Hierarchieinformationen vollständig während der Inferenz. Um die Diskrepanz zwischen Training und Inferenz zu beheben und Labelabhängigkeiten auf principiellere Weise zu modellieren, formulieren wir HTC als einen Markov-Entscheidungsprozess und schlagen vor, eine Labelzuweisungspolitik durch tiefes Reinforcement Learning zu erlernen, um zu bestimmen, wo ein Objekt platziert werden soll und wann der Zuweisungsprozess beendet werden sollte. Die vorgeschlagene Methode, HiLAP, erkundet die Hierarchie sowohl während des Trainings als auch während der Inferenz auf konsistente Weise und trifft interdependente Entscheidungen. Als allgemeines Framework kann HiLAP verschiedene neuronale Encoder als Basismodelle für das end-to-end-Training integrieren. Experimente mit fünf öffentlichen Datensätzen und vier Basismodellen zeigen, dass HiLAP im Vergleich zu flachen Klassifizierern durchschnittlich eine Verbesserung von 33,4 % im Macro-F1-Wert erreicht und deutlich besser als state-of-the-art-Methoden der hierarchischen Textklassifizierung abschneidet. Daten und Code sind unter https://github.com/morningmoni/HiLAP abrufbar.


KI mit KI entwickeln

Von der Idee bis zum Launch – beschleunigen Sie Ihre KI-Entwicklung mit kostenlosem KI-Co-Coding, sofort einsatzbereiter Umgebung und bestem GPU-Preis.

KI-gestütztes kollaboratives Programmieren
Sofort einsatzbereite GPUs
Die besten Preise

HyperAI Newsletters

Abonnieren Sie unsere neuesten Updates
Wir werden die neuesten Updates der Woche in Ihren Posteingang liefern um neun Uhr jeden Montagmorgen
Unterstützt von MailChimp