開発者や研究者は、画像、テキスト、音声、熱画像など、さまざまなタイプのデータを扱う必要があります。従来の方法では、それぞれのモデルを別々に訓練する必要があり、埋め込み空間が一致しないため、クロスモーダル検索やマッチングが煩雑になります。ImageBindはFacebook Researchによって発表され、この課題を解決します。これは、6つのモダリティ(画像、テキスト、音声、深度、熱感、慣性測定ユニットIMU)を同じ次元にバインドする単一の埋め込み空間を作成し、AI研究者、機械学習エンジニア、マルチモーダルアプリケーション開発者向けに、クロスモーダルタスクを簡単かつ効率的にします。
画像訓練なしで他の5つのモダリティを直接バインド
ImageBindのコアイノベーションは、画像-テキストペアの事前訓練モデルを使用するだけで、音声、深度、熱感、IMUなどのモダリティをゼロショットでバインドできることです。このアプローチは、大量のペアデータの必要性を回避し、音声埋め込みを画像埋め込みに直接整列させることができ、追加の訓練を必要としません。研究によると、このバインディング方式はクロスモーダル検索タスクで優れた性能を示し、音声クリップを使って対応する画像を見つける際には、CLIPベンチマークの80%以上の精度を達成しています。リソースが限られている開発者にとって、これは特に便利で、既存のモデルを新しいモダリティに迅速に拡張できます。
実際の操作では、ユーザーは事前訓練された重みを読み込むだけで、統一された埋め込みベクトルを生成できます。単一モダリティモデルと比較して、ImageBindはモデル管理の複雑さを減少させ、ゼロショット分類などの下流タスクの性能を向上させます。
クロスモーダル検索は音声で画像を探し、多モーダルの組み合わせをサポート
検索アプリケーションにおいて、ImageBindは1つのモダリティを使って別のモダリティをクエリすることを可能にします。たとえば、音楽クリップを再生すると、関連する画像や映像を見つけることができます。この機能は統一された埋め込み空間に基づいて構築されており、複数のベンチマークデータセットで平均リコール率がベースラインモデルを上回ります。従来の方法では、各モダリティの類似度を別々に計算する必要がありますが、ImageBindは同じ空間内でコサイン類似度を直接計算するため、より速く、より正確です。
さらに、熱画像とIMUデータを組み合わせて現実世界のオブジェクトを検索するなど、新興モダリティの組み合わせをサポートしています。このようなアプリケーションはAR/VRやロボットナビゲーションに特に価値があり、開発者はセンサーのデータを視覚モデルに簡単に統合できます。
オープンソースコードで研究者がカスタマイズ拡張可能
ImageBindはGitHubで完全にオープンソースとして提供されており、訓練コード、事前訓練モデル、評価スクリプトが含まれています。ユーザーはリポジトリをフォークして、バインディングプロセスを変更したり、新しいモダリティを追加したりできます。文書はモデルアーキテクチャを詳細に説明しており、対比学習を用いて埋め込みをバインドする方法が説明されているため、研究者は結果を再現したり、自分の改善をベンチマークしたりすることが容易です。最新のコミットはチームが継続的に更新しており、より多くのデータセットやハードウェアの最適化をサポートしています。
学術用途において、このプロジェクトはAudioCapsやVGGSoundデータセットの検索結果など、標準ベンチマークを提供し、比較を容易にします。開発者は提供されたAPIを利用して自社のパイプラインに統合し、マルチモーダルAIプロトタイプの開発を加速できます。
マルチモーダル生成アプリケーションとしてのEmergent能力のデモ
検索に加えて、ImageBindは文字から音声を生成したり、画像と深度データを用いて運動軌跡を予測したりするなどのエマージェント能力を示しています。これらは主な焦点ではありませんが、統一空間の副産物として生じる能力であり、デモでは簡単なプロンプトを用いてクロスモーダル出力を生成します。研究者はこれを基に生成タスクを探求し、IMUシーケンスを視覚化された軌跡に変換することができます。
全体として、ImageBindはマルチモーダル研究のハードルを下げ、より多くの実験を促進します。このプロジェクトには貢献ガイドラインや安全ポリシーもあり、コミュニティの参加を歓迎しています。
製品名:ImageBind
公式ウェブサイト:https://github.com/facebookresearch/ImageBind

