本論文では、ステアード混合専門家(SMoE)ゲーティングネットワークとラジアル基底関数(RBF)ネットワークの最適化を高速化するための新しい適応的セグメンテーションに基づく初期化手法を提案している。
提案手法の主な特徴は以下の通りである:
適応的初期化による高周波成分の高精度キャプチャ:
最適化時間の大幅な短縮:
広範な適用性:
提案手法では、セグメンテーション再構築とパラメータ出力の2つの新しい中間ステップを導入している。セグメンテーション再構築では、各セグメントのカーネル数、位置、ステアリングパラメータを最適化する。パラメータ出力では、各セグメントのローカルパラメータを集約、スケーリング、グローバル最適化のための初期化に準備する。
実験結果から、提案手法はグリッド初期化やK-Means初期化、先行研究のセグメンテーションベース初期化に比べ、大幅な品質向上と高い疎sparse性を実現できることが示された。同じ品質レベルで、提案手法は約50%のカーネル数削減を達成できる。また、収束時間も最大50%短縮できる。さらに、セグメンテーションベースの初期化自体が高い並列性を持つため、4つのGPUを使えば初期化時間を50%短縮できる。
他の言語に翻訳
原文コンテンツから
arxiv.org
深掘り質問