google-ai-edge/model-explorer

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 로 UI 를 시작하거나 프로그래밍 방식으로 사용할 수 있습니다.

지원되는 모델 형식 (기본 제공)

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 위키 페이지를 따라 새로운 형식 지원을 제공할 수 있습니다.

시작하기

pip install ai-edge-model-explorer   # Python 패키지 설치
model-explorer                       # UI 실행

Hugging Face Spaces 에서 웹 기반 버전도 사용 가능하여 빠르게 업로드할 수 있습니다.

더 알아보기

위키 – 설치, 사용자 가이드, CLI 가이드, API 가이드, Colab 노트북, 어댑터 개발 문서 등 상세 정보 제공.
YouTube 소개 영상 및 Google Research 블로그 포스트 는 고수준 개요를 제공합니다.
NPM 패키지 – 시각화 컴포넌트(ai-edge-model-explorer-visualizer)는 웹 앱에 임베드할 수 있습니다.

결론: Model Explorer 는 AI/ML 모델 아키텍처를 이해하고 디버깅하거나 발표할 필요가 있는 엔지니어를 위한 진정한 생산 환경 대응 가능한 시각화 디버깅 도구이며, 여러 주요 형식을 지원합니다.