基盤モデルの学習に向けて、テキスト・画像・音声・動画データをクリーニング、フィルタリング、重複排除、合成するための、200種類以上の組み合わせ可能なオペレーターを備えたデータ処理システム。
NVIDIA-NeMo/Curator
AI学習用の大規模なテキスト、画像、動画、音声データセットを、フィルタリング、重複排除、変換する再現性の高いパイプラインを構築するGPU高速化ツールキット。
- スター
- 1,818
- フォーク
- 351
- 未解決の Issue
- 297
- 最終コミット
- 6 時間前
これは何?
NeMo Curatorは、AIモデルの学習に使う大規模なデータセットを準備するためのPythonツールキットです。MLエンジニアやデータチームは、データの読み込み、フィルタリング、重複排除、分類、書き出しを行うステージを組み合わせてパイプラインを構築できます。テキスト、画像、動画、音声には、それぞれ専用の構成要素が用意されています。
各ステージで必要なCPU、GPU、レプリカ数を指定でき、プラグイン可能なexecutorがリソースに応じて処理をスケールします。同じパイプライン定義を、単一マシンでも複数ノードのRayクラスターでも実行でき、Slurmで管理されたHPC環境にも対応します。プロジェクトのドキュメントではXenna executorを本番環境でのデフォルトとしており、Rayベースのバックエンドは実験的な選択肢として提供されています。
このツールキットはNVIDIA NeMoスイートの一部で、NemotronおよびNemotron-CCデータセットの説明にあるデータパイプラインで使用されています。uvまたはDockerでインストールでき、Linux上でGPUを使ったテキスト処理にはCUDA 12の構成が用意されています。
主な機能
- 01データの読み込み、フィルタリング、重複排除、分類、書き出しを行うモジュール式のステージ。
- 02RAPIDSを利用した、GPU高速化対応の重複排除と分類。
- 03同じパイプラインをノートPCでも複数ノードのRayクラスターでも実行可能。
- 04テキスト、画像、動画、音声に対応するモダリティ別プラグイン。
- 05合成データのワークフロー向けに、OpenAI互換の推論サーバーを搭載。
ユースケース
- LLMの事前学習前に、Webテキストの重複を排除し、品質に基づいてフィルタリングする。
- 学習データセットに使う大規模な画像コレクションをフィルタリングし、埋め込みを生成する。
- シーン検出と動きのフィルタリングを使って動画クリップを準備する。
- 音声モデルの学習に向けて、音声データを文字起こしし、フィルタリングする。
クイックスタート
- 1
uv パッケージマネージャーをインストールします(README のインストールスクリプトは sh にパイプされているため、ここではコマンドを記載していません)
- 2
仮想環境を作成し、CPU テキスト用の追加パッケージをインストールします
uv venv && source .venv/bin/activate uv pip install "nemo-curator[text_cpu]" - 3
バージョンを表示して、インストールを確認します
python -c "import nemo_curator; print(nemo_curator.__version__)"
使う前に知っておきたいこと
- GPU テキスト処理には、Linux x86_64 上の CUDA 12 と約 16 GB の GPU メモリが必要です。
- CUDA 12 テキスト用の追加パッケージは通常の pip install ではサポートされておらず、専用の uv オーバーライドファイルが必要です。
- 音声および動画のワークフローには FFmpeg が必要です。カスタムコンテナのビルドで対応するのが最適です。
- NVIDIA は 26.09 以降、NGC で新しい NeMo Curator コンテナイメージを公開しなくなります。
トピック
よくある質問
- Curator とは?
- AI学習用の大規模なテキスト、画像、動画、音声データセットを、フィルタリング、重複排除、変換する再現性の高いパイプラインを構築するGPU高速化ツールキット。 NeMo Curatorは、AIモデルの学習に使う大規模なデータセットを準備するためのPythonツールキットです。MLエンジニアやデータチームは、データの読み込み、フィルタリング、重複排除、分類、書き出しを行うステージを組み合わせてパイプラインを構築できます。テキスト、画像、動画、音声には、それぞれ専用の構成要素が用意されています。
- Curator は無料で使えますか?ライセンスは?
- Curator は Apache-2.0 ライセンスのオープンソースです。商用製品で使う前にライセンス条件を確認してください。
- Curator は今もメンテナンスされていますか?
- 活発にメンテナンス中。Curator の最終コミットは 2026年10月21日 です。
- Curator は何に使えますか?
- LLMの事前学習前に、Webテキストの重複を排除し、品質に基づいてフィルタリングする。 学習データセットに使う大規模な画像コレクションをフィルタリングし、埋め込みを生成する。 シーン検出と動きのフィルタリングを使って動画クリップを準備する。 音声モデルの学習に向けて、音声データを文字起こしし、フィルタリングする。
- Curator のインストール方法は?
- uv パッケージマネージャーをインストールします(README のインストールスクリプトは sh にパイプされているため、ここではコマンドを記載していません) 仮想環境を作成し、CPU テキスト用の追加パッケージをインストールします: uv venv && source .venv/bin/activate uv pip install "nemo-curator[text_cpu]" バージョンを表示して、インストールを確認します: python -c "import nemo_curator; print(nemo_curator.__version__)"
- Curator はどこで入手できますか?
- ソースコードとインストール手順は GitHub にあります:https://github.com/NVIDIA-NeMo/Curator
概要はプロジェクトの README をもとに AI が生成しています。GitHub の数値は毎日更新されます。
関連プロジェクト
独自データとLLMをつなぎ、RAG、エージェント、ドキュメント処理アプリケーションを構築するためのPythonフレームワーク。
自然言語で記述した大規模な構造化・非構造化データ向けの、LLMを活用するPython製map-reduce/ETLパイプラインシステム。
OpenAI互換APIを備え、GPU、ブラウザー、モバイル端末で大規模言語モデルをネイティブに実行する機械学習コンパイラー兼デプロイエンジン。
複数のLLMプロバイダーで共通のOpenAI形式APIを使える軽量なPythonライブラリ。ツールとMCPに対応したAgents APIも備えています。
MNNはAlibabaが開発する軽量な深層学習の推論・学習エンジンで、端末上のLLMを含む、モバイル端末や組み込みデバイスでのモデル実行を想定しています。