AISBench/benchmark
AISBench Benchmark is a model evaluation tool built on OpenCompass, compatible with OpenCompass’s configuration system, dataset structure, and model backend implementation, while extending support for service-based models.
AISBench Benchmark は、OpenCompass に基づいて構築された AI モデル向けの評価ツールであり、その構成体系、データセット構造、モデルバックエンドと互換性があります。さらに、サービス化されたモデル(vLLM、Triton など)のサポートが拡張されています。
2 つの主要な評価カテゴリをサポートしています。
- 精度評価: サービス化されたモデルまたはローカルモデルの、QA、推論、マルチモーダルなどのベンチマークデータセット(GSM8K、MMMU、SWE-Bench など)における正確性を評価します。
- パフォーマンス評価: サービス化されたモデルの遅延、スループットを評価し、高並行性、安定したパフォーマンス、実際のトラフィックシミュレーション、prefix cache ヒット率テストなどをサポートします。
主な機能:
- マルチタスク並列処理: 複数の精度またはパフォーマンス評価タスクを同時に実行でき、効率が向上します。
- マルチモーダルサポート: 画像、ビデオ生成品質の評価(VBench、OneIG-Benchmark など)をサポートします。
- エージェント評価: SWE-Bench、τ²-Bench などのエージェントベンチマークに対応し、Docker コンテナ内での実行をサポートします。
- 豊富なデータセット: 100 以上のオープンソースデータセットが組み込まれており、カスタムデータセットもサポートしています。
- 柔軟な構成: Python 構成ファイル(推奨)またはコマンドライン引数の 2 つの方法を提供します。
インストールと使用:
- Python 3.10/3.11/3.12 をサポートし、pip インストール(
pip install ais_bench_benchmark)またはソースからのインストールが可能です。 - まず、OpenAI 互換の推論サービス(vLLM など)を起動し、モデルサービスのアドレスとデータセットを指定して、
ais_benchコマンドを実行すると評価を開始できます。
プロジェクトドキュメントは充実しており(ReadTheDocs)、中国語と英語のバージョンが提供されています。全体として、本番環境向けの AI モデル評価ツールとして位置付けられており、標準化された評価プロセスを必要とするチームに適しています。
関連
- プロジェクト
- プロジェクト
- プロジェクト
- プロジェクト
- プロジェクト