Stable Diffusion - Image to Promptsに参加しました

作成日: 2023年06月27日(火) 00:00
最終更新日: 2024年03月05日(火) 17:14
カテゴリ: Kaggle
タグ:  Kaggle AI 画像キャプション 生成AI コンピュータビジョン

こんにちは.高山です.
参加から大分時間が経ってしまったのですが,KaggleのStable Diffusion-Image to Promptsというコンテストに参加していました.
ふと流行りのStable Diffusionについて勉強してみたくなってコンテスト終了2週間前に参加を決めました.
流石にこの期間だと私の実力ではベースラインを少しいじる程度が限界でしたが,タイトな分集中できて有意義な時間が過ごせたと思います.

このコンテストは画像を入力してその説明文を生成する,画像キャプショニングと呼ばれるタスクなのですが問題設定が少し変わっています.
図1に示すように,入力画像はStable Diffusion 2.0 [1,2]という文字列から画像を自動生成する技術で作られた画像になります.
参加者は各自の画像キャプショニングモデルを用いて生成画像からキャプションを生成します.
入力画像を生成するのに使われたキャプション (以降,参照キャプション) と,入力画像から生成したキャプション (以降,生成キャプション) の意味の類似度を評価して競い合います.

図1: AI生成画像からの画像キャプショニング
AI生成画像からの画像キャプショニング

なお,意味の類似度は下記のような手順で評価します.

このタスクは一般的な写真画像で学習された画像キャプショニングモデルでは性能に限界があるため,生成画像用のデータセットを用いてモデルを学習する必要があります.
コンテスト期間中は簡単なファインチューニング程度しかできませんでしたが,時間があればまた色々と試してみたいと思います.

  • [1]: "Stable Diffusion 2.0," available here, 2023.
  • [2]: R. Rombach et al., "High-Resolution Image Synthesis with Latent Diffusion Models," CVPR 2022, 2022.
  • [3]: "Sentence-Transformers," available here, 2023.
  • [4]: N. Reimers and I. Gurevych, "Sentence-BERT: Sentence Embeddings using Siamese BERT-Networks," EMNLP 2019, 2019.