google-ai-edge/model-explorer

A modern model graph visualizer and debugger

Model Explorer – AI モデルグラフのビジュアルツール

概要

  • ニューラルネットワークモデル(TensorFlow、TFLite、TensorFlow-JS、MLIR、PyTorch-Exported-Program など)を読み取り、その計算グラフをインタラクティブな階層図として表示するオープンソースのビジュアライザーです。

重要性

  • 大規模なモデルには、多くの論理ブロックに分散した数百の操作が含まれることがあります。Model Explorer は操作をネストされたレイヤーにグループ化し、セクションを展開または折りたたんで全体像を把握したり、単一の操作にドリルダウンしたりできます。
  • 開発者が形状の不一致をデバッグし、予期しない操作を特定し、生のコードを読まずにモデルの構造を理解するのに役立ちます。

主な機能(README の説明による)

  • 階層ビュー – レイヤーをネスト可能。クリックで展開/折りたたみ。
  • I/O の強調表示 – 入力ノードと出力ノードをすばやく特定。
  • メタデータオーバーレイ – ノードにカスタム情報(テンソル形状、データ型など)を添付。
  • インタラクティブなポップアップ – ノードにホバーまたはクリックして詳細を表示。
  • 検索 – 名前またはタイプで操作を検索。
  • 重複レイヤー検出 – 同一のサブグラフを表示。
  • GPU 高速化レンダリング – 大きなグラフでもスムーズなナビゲーション。
  • 拡張フレームワーク – 開発者は新しいモデル形式のアダプタを追加できます。
  • コマンドラインと API – ターミナルからツールを使用(model-explorer)または Python コードに埋め込む。
  • Web コンポーネント – ビジュアライザーコアは npm パッケージ(ai-edge-model-explorer-visualizer)としても公開され、Web アプリに統合できます。

サポートされるモデル形式

  • TensorFlow Lite (.tflite)
  • TensorFlow (SavedModel/graphdef)
  • TensorFlow.js (.json)
  • MLIR(Google のコンパイラスタックで使用される中間表現)
  • PyTorch Exported Program(TorchScript に似た新しい形式)
  • ONNX、VGF、TOSA、PTE 用のコミュニティアダプタがあります。

はじめに

# Python パッケージをインストール
pip install ai-edge-model-explorer
# GUI を起動
model-explorer path/to/your/model.tflite
  • Web ベースのバージョンも Hugging Face Spaces でホストされており、すばやくアップロードできます。
  • 詳細な手順はプロジェクト Wiki(インストール、ユーザーガイド、CLI ガイド、API ガイド、Colab ノートブック、独自アダプタの作成方法)にあります。

拡張性

  • リポジトリはカスタムアダプタの貢献を奨励しています。「Develop Adapter Extension」Wiki ページで必要なインターフェースが説明されており、メンテナは README に新しいアダプタへのリンクを追加します。
  • ビジュアライザーコンポーネントは npm パッケージを介して他のフロントエンドに埋め込むことができ、src/custom_element_demos フォルダにデモがあります。

コミュニティとリソース

  • YouTube のビデオ紹介と Google Research Blog のブログ記事で概要を説明しています。
  • ユーザーは Google フォームを介してツールに関する研究に参加できます。

対象ユーザー

  • モデルアーキテクチャの検査やデバッグが必要な ML エンジニア。
  • 新しいモデル形式を構築し、すばやく視覚化したい研究者。
  • AI を活用した Web アプリを構築し、既製のグラフビジュアライザーが必要なフロントエンド開発者。

結論: Model Explorer は、AI モデル向けの実用的でクロスフレームワークな視覚デバッグ支援ツールであり、デスクトップ GUI と再利用可能な Web コンポーネントの両方を提供し、追加のモデル形式への拡張を明確にサポートします。

関連

  • Dispatch
  • プロジェクト
  • プロジェクト
  • Dispatch
  • プロジェクト