google-ai-edge/model-explorer

Model Explorer – AI 모델 그래프를 위한 시각적 도구

그것이 무엇인가

신경망 모델(TensorFlow, TFLite, TensorFlow-JS, MLIR, PyTorch-Exported-Program 등)을 읽고 계산 그래프를 상호작용적인 계층 다이어그램으로 표시하는 오픈소스 시각화 도구입니다.

왜 중요한가

대규모 모델은 많은 논리적 블록에 분산된 수백 개의 연산을 가질 수 있습니다. Model Explorer는 연산을 중첩된 레이어로 그룹화하여 섹션을 확장하거나 축소하여 전체 그림을 보거나 단일 연산으로 드릴다운할 수 있게 합니다.
개발자가 형태 불일치를 디버그하고, 예상치 못한 연산을 찾고, 원시 코드를 읽지 않고 모델 구조를 이해하는 데 도움이 됩니다.

주요 기능(README 설명에 따름)

계층 보기 – 레이어를 중첩할 수 있으며, 클릭하여 확장/축소합니다.
I/O 강조 – 입력 및 출력 노드를 빠르게 찾습니다.
메타데이터 오버레이 – 노드에 사용자 정의 정보(예: 텐서 형태, 데이터 유형)를 첨부합니다.
상호작용 팝업 – 노드를 가리키거나 클릭하여 세부 정보를 봅니다.
검색 – 이름이나 유형으로 연산을 찾습니다.
중복 레이어 감지 – 동일한 하위 그래프를 표시합니다.
GPU 가속 렌더링 – 큰 그래프에서도 부드러운 탐색.
확장 프레임워크 – 개발자가 새 모델 형식에 대한 어댑터를 추가할 수 있습니다.
명령줄 및 API – 터미널에서 도구 사용(model-explorer) 또는 Python 코드에 포함.
웹 구성 요소 – 시각화 코어는 npm 패키지(ai-edge-model-explorer-visualizer)로도 게시되어 웹 앱에 통합할 수 있습니다.

지원되는 모델 형식

시작하기

# Python 패키지 설치
pip install ai-edge-model-explorer
# GUI 실행
model-explorer path/to/your/model.tflite

웹 기반 버전도 Hugging Face Spaces에서 호스팅되어 빠른 업로드가 가능합니다.
자세한 단계별 지침은 프로젝트 Wiki(설치, 사용자 가이드, CLI 가이드, API 가이드, Colab 노트북, 사용자 정의 어댑터 작성 방법)에 있습니다.

확장성

저장소는 사용자 정의 어댑터 기여를 장려합니다. "Develop Adapter Extension" Wiki 페이지에서 필요한 인터페이스를 설명하며, 관리자는 README에 새 어댑터 링크를 추가합니다.
시각화 구성 요소는 npm 패키지를 통해 다른 프런트엔드에 포함할 수 있으며, src/custom_element_demos 폴더에 데모가 제공됩니다.

커뮤니티 및 리소스

사용 대상

결론: Model Explorer는 AI 모델을 위한 실용적이고 크로스 프레임워크 시각적 디버깅 도우미로, 데스크톱 GUI와 재사용 가능한 웹 구성 요소를 모두 제공하며 추가 모델 형식 지원 확장을 위한 명확한 경로를 제공합니다.