KI-Hardware und Software-Stack
Welche Aussage zum beim Training verwendeten Gradient Checkpointing (Neuberechnung von Aktivierungen) ist richtig?
- AEs speichert nur einen Teil der Zwischenergebnisse des Forward Pass und berechnet den Rest im Backward Pass bei Bedarf neu, sodass Speicher auf Kosten zusätzlicher Berechnung gespart wird
- BEs speichert die Modellgewichte während des Trainings auf der Festplatte
- CEs trainiert, ohne Gradienten zu berechnen
Antwort und Erläuterung anzeigen
Antwort: A. Es speichert nur einen Teil der Zwischenergebnisse des Forward Pass und berechnet den Rest im Backward Pass bei Bedarf neu, sodass Speicher auf Kosten zusätzlicher Berechnung gespart wird
Der Backward Pass benötigt die Zwischenergebnisse des Forward Pass jeder Schicht; werden alle gespeichert, wächst der Speicher proportional zu Schichtzahl und Batchgröße. Beim Gradient Checkpointing werden nur die Ergebnisse an jeweils einigen Schichten gespeichert, und wenn der Backward Pass diesen Abschnitt erreicht, wird der Forward Pass ab dem gespeicherten Punkt noch einmal berechnet, um die benötigten Werte zu rekonstruieren. Der Forward Pass wird dadurch etwa einmal zusätzlich berechnet, dafür sinkt der Aktivierungsspeicher stark.