VL-KE-T5은 KE-T5와 ViT의 임베딩 벡터를 Vision-Language parallel Corpus를 이용하여 정렬한 모델입니다. 영어와 한국어 모두 지원하며, Vision-Language Parallel 데이터 셋들을 Google 번역 API를 이용하여 한국어로 번역한 데이터를 추가적으로 이용하였습니다.
학습에 사용된 Vision-Language Parallel 데이터셋은 다음과 같습니다.
English | Korean (Translated) | Korean | ||||||||
CC 3M | COCO | SBU | Visual Genome | WIT | CC 3M | COCO | SBU | Visual Genome | WIT | WIT |
2,862,265 | 414,113 | 772,438 | 4,322,358 | 3,265,279 | 2,862,264 | 414,113 | 772,438 | 4,322,358 | 3,265,273 | 54,956 |
pip install torch
conda install -c pytorch faiss-gpu # or faiss-cpu
pip install transformers sentencepiece
faiss의 자세한 설차 방법은 FAISS를 참고하시길 바랍니다.
import os
import csv
import logging
import tqdm
import torch
from transformers import AutoTokenizer
from index_scorer import FaissScorerExhaustive
# # GPU ------------------------------------------------
# from index_scorer import FaissScorerExhaustiveMultiGPU
# # ----------------------------------------------------
from modeling_encoder import T5EncoderMean
INDEX_PATH="cc12m_filtered.index"
URL_PATH="cc12m_filtered.tsv"
# # GPU ------------------------------------------------
# FVECS_ROOT="fvecs"
# GPU4MODEL=0
# GPUS4FAISS=[1, 2, 3, 4]
# # ----------------------------------------------------
LANGUAGE_MODEL_PATH="KETI-AIR/ke-t5-base"
ENCODER_PATH="hf_model"
LANGUAGE_ENCODER_PATH=os.path.join(ENCODER_PATH, "language")
TOPK=8
text_tokenizer = AutoTokenizer.from_pretrained(LANGUAGE_MODEL_PATH)
model = T5EncoderMean.from_pretrained(LANGUAGE_ENCODER_PATH)
# # GPU ------------------------------------------------
# model_device = torch.device('cuda:{}'.format(GPU4MODEL))
# model = model.to(model_device)
# # ----------------------------------------------------
model.eval()
text_query = [
"바닷가에서 달리는 사람들",
"화려한 원피스를 입은 젊은 여성",
]
text_feature = text_tokenizer(
text_query,
return_tensors="pt",
truncation='longest_first',
padding=True)
faiss_scorer = FaissScorerExhaustive(
index_path=INDEX_PATH
)
# # GPU ------------------------------------------------
# faiss_scorer = FaissScorerExhaustiveMultiGPU(
# fvec_root=FVECS_ROOT,
# gpu_list=GPUS4FAISS
# )
# # ----------------------------------------------------
url_data = [
item for item in tqdm.tqdm(csv.DictReader(
open(URL_PATH, "r"),
delimiter="\t",
quoting=csv.QUOTE_MINIMAL,
fieldnames=['path', 'image_url']
), desc="loading item...")
]
with torch.no_grad():
outputs = model(
input_ids=text_feature["input_ids"],
attention_mask=text_feature["attention_mask"],
)
q_vecs = outputs[1]
q_vecs = q_vecs.numpy()
# # GPU ------------------------------------------------
# outputs = model(
# input_ids=text_feature["input_ids"].to(model_device),
# attention_mask=text_feature["attention_mask"].to(model_device),
# )
# q_vecs = outputs[1]
# q_vecs = q_vecs.cpu().numpy()
# # ----------------------------------------------------
scores, indice = faiss_scorer.get_topk(q_vecs, TOPK)
result_list = []
for t, score, index in zip( range(len(text_query)),
scores,
indice):
result = [ {
"k": k+1,
"score": s,
"image_url": url_data[i]["image_url"]
} for k, s, i in zip(range(TOPK), score, index)]
result_list.append(result)
for query, result in zip(text_query, result_list):
print(f"query: {query}\nresults\n"+'-'*40)
print(result)
print('-'*40+'\n\n')
Query | 바닷가에서 달리는 사람들 | ||
Top 1 | Top 2 | Top 3 | Top 4 |
|
|
|
|
Top 5 | Top 6 | Top 7 | Top 8 |
|
|
|
|
Query | 화려한 원피스를 입은 젊은 여성 | ||
Top 1 | Top 2 | Top 3 | Top 4 |
|
|
|
|
Top 5 | Top 6 | Top 7 | Top 8 |
|
|
|
|
T5EncoderMean
와 forward
대신에 VisionT5MeanBiEncoder
와 encode_text
를 사용하실 수 있습니다.
다만 VisionT5MeanBiEncoder
를 사용하실 경우 ViT 모델도 로드되기 때문에 메모리 사용에 있어 약간 비효율적입니다.
from modeling_encoder import VisionT5MeanBiEncoder
ENCODER_PATH="hf_model"
# ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
model = VisionT5MeanBiEncoder.from_pretrained(ENCODER_PATH)
model = model.to(model_device)
model.eval()
# ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
q_vecs = model.encode_text({
"input_ids":text_feature["input_ids"].to(model_device),
"attention_mask":text_feature["attention_mask"].to(model_device),})
q_vecs = q_vecs.cpu().numpy()
cc12m 데이터의 인덱스는 약 33Gb 정도가 됩니다.
from index_scorer import FaissScorerExhaustive
INDEX_PATH="cc12m_filtered.index"
faiss_scorer = FaissScorerExhaustive(
index_path=INDEX_PATH
)
from index_scorer import FaissScorer
INDEX_PATH="cc12m_filtered_OPQ64_256-IVF262144_HNSW32-PQ64.index"
# INDEX_PATH="cc12m_filtered_OPQ192_768-IVF262144_HNSW32-PQ192.index"
faiss_scorer = FaissScorer(
index_path=INDEX_PATH,
)
from index_scorer import FaissScorerExhaustiveGPU
FVECS_ROOT="fvecs"
GPU_ID=0
faiss_scorer = FaissScorerExhaustiveGPU(
fvec_root=FVECS_ROOT,
gpu=GPU_ID
)
from index_scorer import FaissScorerExhaustiveMultiGPU
FVECS_ROOT="fvecs"
GPU_ID_LIST=[0, 1, 2, 3]
faiss_scorer = FaissScorerExhaustiveMultiGPU(
fvec_root=FVECS_ROOT,
gpu_list=GPU_ID_LIST
)
VisionT5MeanBiEncoder (Language Lock)
VisionT5MeanBiEncoder + CC 12M 3 epochs
VisionT5MeanBiEncoder (norm) + CC 12M 3 epochs
VisionEncoderLanguageDecoder(VELD) + CC 12M 1 epoch
Model | Data | # of images | url data | fvecs | exhaustive index | OPQ64-256 | OPQ192_768 |
---|---|---|---|---|---|---|---|
VisionT5MeanBiEncoder | CC 12M | 10,793,580 | (1.67GB) | (30.96GB) | (30.88GB) | (1.04GB) | (2.83GB) |
VisionT5MeanBiEncoder | mmcommons | 99,144,306 | (14.31GB) | (284GB) | - | (6.97GB) | (19.29GB) |
VisionT5MeanBiEncoder (Language Lock) | CC 12M | 10,793,580 | (1.67GB) | (33.2GB) | - | (1.04GB) | (2.83GB) |
개인적으로 준비한 이미지를 이용하고 싶으신 경우 아래 블럭과 같이 tsv 포맷으로 이미지와 이미지의 URL을 만들어 줍니다.
(꼭 image URL일 필요는 없습니다. 2번째 칼럼 값이 return 되기 때문에 해당 이미지에 대하여 return하고 싶은 값이면 됩니다.)
test.tsv
{path_to_image_0}\t{image_url_0}
{path_to_image_1}\t{image_url_1}
{path_to_image_2}\t{image_url_2}
...
create_index4retriever.py
를 이용하여 index를 만들어 줍니다.
DATA_PATH=test.tsv
FVECS_OUT_DIR=test_fvecs
ENCODER_PATH=hf_model
CUDA_VISIBLE_DEVICES="0,1,2,3" python -m torch.distributed.launch --nproc_per_node=4 create_index4retriever.py \
--batch_size 64 \
--batch_write \
--model_cls VisionT5MeanBiEncoder \
--data_path $DATA_PATH \
--fvecs_dir $FVECS_OUT_DIR \
--hf_path $ENCODER_PATH
# # GPU가 하나만 있는 경우
# python create_index4retriever.py \
# --batch_size 64 \
# --batch_write \
# --model_cls VisionT5MeanBiEncoder \
# --data_path $DATA_PATH \
# --fvecs_dir $FVECS_OUT_DIR \
# --hf_path $ENCODER_PATH
2번째 칼럼이 image url인 경우 아래와 같이 테스트할 query json 파일을 만듭니다.
test_query.json
[
"따뜻한 분위기의 카페",
"화려한 원피스를 입은 젊은 여성",
"심플한 원피스를 입은 젊은 여성",
"축구경기를 응원하는 사람들",
"강아지와 해변을 산책하는 남자",
"강아지와 해변을 산책하는 여자",
]
retrieve_images.py
를 이용하여 검색된 이미지를 확인합니다.
DATA_PATH=test.tsv
FVECS_OUT_DIR=test_fvecs
ENCODER_PATH=hf_model
QUERY_PATH=test_query.json
MD_OUT_DIR=md_out
python retrieve_images.py \
--data_path $DATA_PATH \
--fvecs_dir $FVECS_OUT_DIR \
--hf_path $ENCODER_PATH \
--query_path $QUERY_PATH \
--markdown_out $MD_OUT_DIR \
--model_cls VisionT5MeanBiEncoder
--markdown_out
으로 출력된 markdown 파일들을 확인합니다. VS code와 같은 편집기로 확인하면 Preview를 통해 한눈에 볼 수 있습니다.
cc12m에서 검색된 이미지 샘플들을 참조하려면 CC 12M 샘플을 참조하세요.
mmcommons에서 검색된 이미지 샘플들을 참조하려면 exhaustive 샘플, OPQ192-768 샘플, OPQ64-256 샘플을 참조하세요.
cc12m에서 언어모델을 학습되지 않도록 gradient를 freeze시키고 학습한 모델로 검색된 이미지 샘플들을 참조하려면 exhaustive 샘플, OPQ192-768 샘플, OPQ64-256 샘플을 참조하세요.
구글의 Contrastive Captioners와 같은 방식으로 학습된 Vision Encoder + Language Decoder 구조의 모델입니다.
from modeling_veldt5 import VELDT5Model
from transformers import AutoTokenizer, ViTFeatureExtractor
from PIL import Image
MODEL_PATH = "veld_e1_linear"
tokenizer = AutoTokenizer.from_pretrained(MODEL_PATH)
feature_extractor = ViTFeatureExtractor.from_pretrained(MODEL_PATH)
images = [Image.open("images/sample.jpg"), Image.open("images/sample2.jpg")]
pixel_values = feature_extractor(images=images, return_tensors="pt").pixel_values
model = VELDT5Model.from_pretrained(MODEL_PATH)
outputs = model.generate(
pixel_values=pixel_values,
num_beams=4,
# do_sample=True,
max_new_tokens=40,
num_return_sequences=4,
)
for img_idx, output in enumerate(outputs):
print("image only {}: ".format(img_idx), tokenizer.decode(output, skip_special_tokens=True))
''' stdout
image only 0: A baseball player is standing in front of a baseball field.
image only 1: A baseball player is standing in front of a baseball stadium.
image only 2: A baseball player is standing in front of a field.
image only 3: A baseball player is standing in front of a stadium.
image only 4: 해변에 서 있는 사람.
image only 5: 해변에 서 있는 여자
image only 6: 해변에 서 있는 여자.
image only 7: 해변에 서 있는 사람
'''
본 연구는 정부(과학기술정보통신부)의 재원으로 지원을 받아 수행된 연구입니다. (정보통신기획평가원, 2022-0-00320), (정보통신기획평가원, 2022-0-00608), (정보통신기획평가원, 2021-0-00537), (정보통신산업진흥원, S0316-21-1002)
- language model gradient freeze 하고 학습하기
- hard negative sample 만들기
- 언어 모델 바꾸기