本論文では、深層学習モデルをフレームワーク間で変換する際に発生する障害を自動的に特定し、修復する手法「Fix-Con」を提案している。
モデル変換プロセスには様々な問題が発生する可能性があり、変換後のモデルが展開できなかったり、予測精度が大幅に低下したりする。Fix-Conは、モデル入力、パラメータ、ハイパーパラメータ、モデルグラフなどの変換時の障害を検出し、適切に修復する。
Fix-Conは、コードリポジトリやフォーラムで報告された変換の問題から抽出した6種類の障害タイプを使用して、変換後のターゲットモデルの潜在的な変換障害を特定し、適切に修復する。これは、ソースモデルとの出力ラベルの差異を解消するまで、データセット内の各画像について繰り返し行われる。
実験では、3つの一般的な画像認識モデルを4つの異なる深層学習フレームワーク間で変換し、Fix-Conの有効性を評価した。全体として、Fix-Conは755件の検出された変換障害のうち462件を修復し、15件の誤った変換ケースのうち14件を完全に修復または大幅に改善することができた。
לשפה אחרת
מתוכן המקור
arxiv.org
שאלות מעמיקות