long8v / PTIR

Paper Today I Read
19 stars 0 forks source link

[6] Crossing the Format Boundary of Text and Boxes: Towards Unified Vision-Language Modeling #6

Open long8v opened 2 years ago

long8v commented 2 years ago

Crossing the Format Boundary of Text and Boxes: Towards Unified Vision-Language Modeling arxiv image

image problem : pix2seq는 object detection만을 위한 framework, 이를 다양한 Vision-Language 문제로 unified framework로 확장해보자 solution : pix2seq와 매우 유사하게 object box를 bin으로 자른 뒤, generation 방법으로 task를 품. 이미지는 이미지 인코더(CNN)에 넣고, task prefix, task input을 텍스트 인코더에 넣고 concat한 뒤 트랜스포머 인코더에 넣음, 이후 하나의 트랜스포머 디코더가 output을 generation함. Object Detection 외의 task들에 대해 output sequence를 정의함
result : grounded captioning, VQA, image catptioning, object detection 등 다양한 태스크를 하나의 unified model로 풀 수 있었고, 이 중 몇개의 task는 multi-task learning으로 성능에 이득.