Tiny-Dashboard: A Lightweight Feature Activation Analysis Tool

Published in Open Source Project, 2024

Tiny-Dashboard is a lightweight Python package designed to make feature activation analysis in neural language models more accessible and customizable. While other tools like SAEDashboard and sae_vis offer comprehensive features, Tiny-Dashboard prioritizes simplicity and hackability.

Key Features

The package supports both offline and online analysis modes:

Offline Analysis

  • Explore pre-computed feature activations
  • Visualize max activation examples
  • Generate shareable HTML reports
  • Support for both in-memory and database-backed storage

Online Analysis

  • Real-time feature activation analysis
  • Support for chat template formatting
  • Custom prompt testing
  • Interactive visualization

Implementation

The tool is built with flexibility in mind, offering both class-based and function-based approaches for customization. It includes specialized implementations like the CrosscoderOnlineFeatureDashboard, which was developed during the CrossCoders project to analyze combined base and instruct model activations.

The package is available on PyPI:

pip install tiny-dashboard

For detailed usage examples and documentation, visit the GitHub repository.