Zum Inhalt springen

FlowTracer: Gezieltes Reinforcement Learning durch Nachverfolgung von Informationsfluss in LLMs

Share on:

Auf den Punkt: FlowTracer modelliert Informationspropagation als gerichteten Graphen und leitet Token-Credits aus globaler Flussstruktur ab, um Reinforcement-Learning-Signale präzise auf entscheidende Reasoning-Schritte zu konzentrieren.

Forscher schlagen FlowTracer vor, ein Verfahren, das Aufmerksamkeitsmuster zur Identifikation entscheidender Tokens bei der LLM-Optimierung nutzt. Statt alle Tokens gleichzubehandeln, wird das Reinforcement Learning auf diejenigen konzentriert, die tatsächlich Information zur korrekten Antwort leiten.

Bei der Optimierung von Language Models via Reinforcement Learning besteht ein grundlegendes Problem: RL-Verfahren behandeln typischerweise alle Tokens gleich und können nicht unterscheiden zwischen Tokens, die zentral für den Reasoning-Prozess sind, und solchen, die nur Formatierung oder sprachliche Flüssigkeit leisten. Dies führt zu ineffizienten Lerngradienten und Verlusten bei der Modellsteuerung.

FlowTracer löst dies durch Modellierung der Informationspropagation als gerichteten azyklischen Graphen (DAG): Knoten entsprechen Tokens, Kanten werden mit aggregierten Attention-Gewichten gewichtet. Das System extrahiert aus dieser Graphenstruktur einen „Information-Flow-Backbone“, der die Frage mit der Antwort verbindet, und bewertet jeden Token nach seinem Durchsatz in diesem Flow. Dabei bleiben Zwischentokens in ihrer effektiven Masse erhalten (lokale Flusskonservation), um Verzerrungen durch Pfadlänge oder irrelevante Branches zu vermeiden. Die Kantengewichte werden zudem so angepasst, dass nur Einfluss berücksichtigt wird, der die Antwortregion erreichen kann.

Daraus ergeben sich Token-Level Rewards, die das Modell explizit auf Tokens trainieren, die Information zur richtigen Antwort leiten oder von falschen Antworten weglenken. Besonders bedeutsam sind Hubs und Aggregationspunkte, die langreichweitige Abhängigkeiten vermitteln. Erste Ergebnisse zeigen konsistente Leistungsgewinne über verschiedene Reasoning-Aufgaben hinweg.

Für CTOs und ML-Engineering-Teams bietet FlowTracer eine Methode, Reasoning-Fähigkeiten von LLMs strukturierter und dateneffizienter zu optimieren, indem RL-Signale nicht streuen, sondern gezielt auf kritische Inferenzschritte fallen — ein wesentlicher Fortschritt für produktive Einsätze in Fehlertoleranz und Compliance-kritischen Szenarien.


Quelle: arxiv.org · Erschienen 9. Juni 2026
Lumi AI News — KI-assistierte Kuratierung gemaess Art. 50 EU AI Act. Paraphrase und Klassifikation durch Lumi News Pipeline v1.6.5.

Share on: