hijohnnylin/neuronpedia

open source interpretability platform 🧠

해결하는 문제

Neuronpedia는 연구자와 개발자가 신경망의 내부 동작을 이해하는 데 도움을 주는 오픈소스 해석성 플랫폼입니다. 스파스 오토인코더(SAE) 및 기타 모델 활성화의 특징을 탐색, 분석, 해석할 수 있는 중앙 집중형 허브를 제공합니다.

작동 방식

이 플랫폼은 로컬 또는 클라우드에서 독립적으로 실행할 수 있는 여러 서비스로 구성되어 있습니다. 기계적 해석성에 특화된 여러 도구를 통합하고 있습니다:

  • 추론 서버: vLLM 및 PyTorch와 같은 백엔드를 지원하여 모델 실행, 활성화 테스트 및 방향 조절을 처리합니다.
  • Autointerp 서버: EleutherAI의 Delphi를 사용해 신경망 특징에 대해 자동으로 설명을 생성하고 점수를 매깁니다.
  • 그래프 서버: circuit-tracer 기반으로 정보가 모델의 회로를 통해 어떻게 흐르는지 시각화하는 할당 그래프를 생성합니다.
  • 웹앱: 대시보드, 검색 기능(설명의 의미 검색에 pgvector 사용), 데이터 내보내기/업로드 도구를 제공하는 프론트엔드 및 API입니다.

대상 사용자

AI 안전 연구자, 기계적 해석성 전문가, 그리고 LLM의 내부 표현을 분석하고자 하는 개발자에게 주로 적합합니다.

주요 특징

  • 포괄적인 도구 세트: 방향 조절, 활성화, 회로/그래프, 자이아코비안 렌즈 지원 포함.
  • 의미 기반 검색: 자연어 설명의 의미에 기반해 특징을 검색할 수 있습니다.
  • 자동 해석: 특징의 자동 해석과 점수 매기기를 통합하여 분석 속도를 높입니다.
  • 확장 가능한 아키텍처: Hugging Face 및 SAELens 통합을 통해 사용자 정의 SAE 및 모델 추가를 지원합니다.

관련

  • Dispatch
  • 프로젝트
  • Dispatch
  • 프로젝트
  • 프로젝트