THU-BPM/MarkDiffusion

何を解決するか

MarkDiffusionは、潜在拡散モデル（LDM）によって生成されたメディアにおける真正性と出典の検証の必要性に対応します。AI生成画像および動画に水印を埋め込み、検出および評価する標準化された方法を提供し、不正利用を防ぎ、出典を追跡します。

仕組み

このツールキットは、複数の水印アルゴリズムを統合した単一インターフェースを持つ統一されたPythonフレームワークを提供します。主に2つのカテゴリの手法をサポート：パターンベース（特定のフィンガープリントを埋め込む）と鍵ベース（暗号化鍵を使用）。水印の埋め込みおよび抽出の様子を可視化するサブセットと、さまざまな「攻撃」（切り取り、圧縮、ノイズなど）に対して水印の耐性と最終メディア品質への影響を測定する評価モジュールを含みます。

対象ユーザー

このツールキットは、セキュリティ、著作権保護、真正性検証のために水印を実装する必要がある生成モデルを扱うAI研究者および開発者向けに設計されています。

主な特徴

• 広範なアルゴリズム対応：画像および動画用に11の最先端の水印アルゴリズムを実装。
• 包括的な評価：検出性、耐性、出力品質を測定する33のツールを含む。
• 耐性テスト：JPEG圧縮、ガウスノイズ、Diffusion Purificationなどの高度な再生成攻撃を含む専用攻撃ツール。
• 自動化パイプライン：品質および検出分析をスムーズに行うための8つの自動評価パイプラインを提供。