フェーズ勾配フローによるSSMのメモリ壁の突破

Research Paper#Deep Learning, State Space Models, Memory Optimization🔬 Research|分析: 2026年1月3日 19:16
公開: 2025年12月28日 20:27
1分で読める
ArXiv

分析

本論文は、Selective State Space Models (SSM) のバックプロパゲーションにおける重要なメモリボトルネックに対処し、大規模なゲノムデータやその他の長いシーケンスデータへの応用を制限している。提案されたPhase Gradient Flow (PGF) フレームワークは、中間計算グラフを保存する必要をなくし、状態空間多様体で直接正確な解析的導関数を計算することにより、解決策を提供する。これにより、大幅なメモリ節約(O(1)のメモリ複雑度)とスループットの向上が実現し、以前は実現不可能だった非常に長いシーケンスの分析が可能になる。スティッフODE体制下でもPGFの安定性は重要な利点である。
引用・出典
原文を見る
"PGF delivers O(1) memory complexity relative to sequence length, yielding a 94% reduction in peak VRAM and a 23x increase in throughput compared to standard Autograd."
A
ArXiv2025年12月28日 20:27
* 著作権法第32条に基づく適法な引用です。