Concepts de base
Slipstream, ein Softwarerahmenwerk, identifiziert und überspringt dynamisch veraltete Einbettungen, um die Trainingsleistung zu verbessern und den CPU-GPU-Bandbreithenverbrauch sowie den Speicherzugriff zu optimieren.
Résumé
Der Artikel präsentiert Slipstream, ein Softwarerahmenwerk, das darauf abzielt, die Trainingseffizienz von Empfehlungsmodellen zu optimieren, indem es veraltete Einbettungen dynamisch identifiziert und deren Updates überspringt.
Der Artikel beginnt mit einer Einführung in Empfehlungssysteme und deren Trainingsaufbau. Es werden drei Beobachtungen hervorgehoben, die die Motivation für Slipstream liefern:
- Der hohe Anteil der Einbettungsoperationen an der Gesamttrainingszeit
- Das stark verzerrte Zugriffsmuster auf Einbettungen, das zu "heißen" und "kalten" Einbettungen führt
- Die Sättigung bestimmter "heißer" Einbettungen im Laufe des Trainings, was auf unnötige Aktualisierungen hindeutet
Slipstream besteht aus drei Komponenten:
- Snapshot-Block: Erfasst regelmäßig Momentaufnahmen der "heißen" Einbettungen, um deren Trainingsdynamik zu verfolgen.
- Sampling-Block: Verwendet Stichprobennahme, um effizient einen Schwellenwert zu finden, um stabile Einbettungen zu identifizieren.
- Eingabe-Klassifizierungs-Block: Klassifiziert Eingaben basierend auf den identifizierten stabilen Einbettungen und überspringt Aktualisierungen für stabile Einbettungen.
Slipstream zeigt signifikante Beschleunigungen von 2x, 2,4x, 20% und 18% gegenüber den Baselines XDL, Intel-optimiertem DLRM, FAE und Hotline, während es die Genauigkeit beibehält oder sogar verbessert.
Stats
Die Größe der Einbettungstabellen in realen Datensätzen kann bis zu mehrere Hundert Gigabyte betragen.
Nur ein kleiner Bruchteil der Einbettungen (ca. 500 MB) macht etwa 75% der Gesamtzugriffe aus.
Bestimmte "heiße" Einbettungen zeigen nach etwa 25% der Trainingsiterationen eine Sättigung und benötigen kaum noch Aktualisierungen.
Citations
"Training von Empfehlungsmodellen stellt erhebliche Herausforderungen in Bezug auf Ressourcennutzung und Leistung dar."
"Slipstream optimiert die Trainingseffizienz, indem es selektiv Einbettungswerte aktualisiert, basierend auf Datenbewusstsein."
"Slipstream erzielt Trainingszeitreduzierungen von 2x, 2,4x, 20% und 18% gegenüber den Baselines XDL, Intel-optimiertem DLRM, FAE und Hotline."