google-ai-edge/model-explorer
A modern model graph visualizer and debugger
Model Explorer – これは何ですか
Model Explorer は、Google‑AI‑Edge のオープンソースツールで、ニューラルネットワークモデルを階層的なグラフとして可視化できるものです。モデルファイル(TensorFlow、TensorFlow‑Lite、TensorFlow‑JS、MLIR、または PyTorch ExportedProgram)を解析し、操作を層にグループ化されたノードとして描画します。ユーザーは層を展開または折りたたむことができます。UI はローカルで実行可能(または Hugging Face Space で)され、GPU による加速レンダリングにより、スムーズな操作が可能になります。
コア機能(README に記載)
- 階層ビュー – 操作が論理的な層にネストされ、巨大なグラフのナビゲーションを容易にします。
- 動的な展開/折りたたみ – 層を詳細に掘り下げたり、不要な情報を一時的に非表示にできます。
- I/O のハイライト – モデルの入力および出力ノードをすばやく特定できます。
- メタデータオーバーレイ – 任意のノードにカスタム情報を添付して表示できます。
- 検索 & 重複層検出 – ノード名で検索でき、同一のサブグラフがどこに出現しているかを確認できます。
- GPU による加速レンダリング – 大規模モデルでもスムーズなパン/ズームが可能。
- 拡張フレームワーク – 開発者は新しいモデル形式用のアダプタを追加できます。
- コマンドライン & API –
model-explorerで起動可能、またはプログラム的に利用可能。
サポートされるモデル形式(標準搭載)
- TensorFlow Lite(
.tflite) - TensorFlow(
SavedModel/ GraphDef) - TensorFlow.js(
.json+ 重み) - MLIR(Google の中間表現)
- PyTorch ExportedProgram(エッジデプロイ用に使用される Torch-Script に似た表現)
拡張性
このプロジェクトはコミュニティによるアダプタの開発を奨励しています。既存のコミュニティアダプタには以下があります:
- ONNX
- VGF(Arm のビジョングラフ形式)
- TOSA(Tensor Operator Set Architecture)
開発者は Develop Adapter Extension の Wiki ページに従って、新しい形式のサポートを提供できます。
使い始め
pip install ai-edge-model-explorer # Python パッケージをインストール
model-explorer # UI を起動
Hugging Face Spaces でも、ウェブベースのバージョンが利用可能で、簡単にアップロードできます。
詳細情報
- Wiki – 詳細なインストール手順、ユーザーガイド、CLIガイド、APIガイド、Colab ノートブック、アダプタ開発ドキュメント。
- YouTube イントロ動画 および Google Research ブログ記事 で、概要を紹介しています。
- NPM パッケージ – ビジュアライザー コンポーネント(
ai-edge-model-explorer-visualizer)は、Web アプリに埋め込むことができます。
結論:Model Explorer は、AI/ML モデルのアーキテクチャを理解・デバッグ・プレゼンテーションするための本格的で生産環境対応の可視化ツールであり、複数の主要な形式に対応するエンジニアに向けたものです。
関連
- プロジェクト
- Dispatch
- プロジェクト
- Dispatch
- プロジェクト