フェーズ勾配フローによる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の安定性は重要な利点である。