Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

LISA: Reasoning Segmentation via Large Language Model

About

Although perception systems have made remarkable advancements in recent years, they still rely on explicit human instruction or pre-defined categories to identify the target objects before executing visual recognition tasks. Such systems cannot actively reason and comprehend implicit user intention. In this work, we propose a new segmentation task -- reasoning segmentation. The task is designed to output a segmentation mask given a complex and implicit query text. Furthermore, we establish a benchmark comprising over one thousand image-instruction-mask data samples, incorporating intricate reasoning and world knowledge for evaluation purposes. Finally, we present LISA: large Language Instructed Segmentation Assistant, which inherits the language generation capabilities of multimodal Large Language Models (LLMs) while also possessing the ability to produce segmentation masks. We expand the original vocabulary with a <SEG> token and propose the embedding-as-mask paradigm to unlock the segmentation capability. Remarkably, LISA can handle cases involving complex reasoning and world knowledge. Also, it demonstrates robust zero-shot capability when trained exclusively on reasoning-free datasets. In addition, fine-tuning the model with merely 239 reasoning segmentation data samples results in further performance enhancement. Both quantitative and qualitative experiments show our method effectively unlocks new reasoning segmentation capabilities for multimodal LLMs. Code, models, and data are available at https://github.com/dvlab-research/LISA.

Xin Lai, Zhuotao Tian, Yukang Chen, Yanwei Li, Yuhui Yuan, Shu Liu, Jiaya Jia• 2023

Related benchmarks

TaskDatasetResultRank
Referring Expression ComprehensionRefCOCO+ (val)--
354
Referring Expression ComprehensionRefCOCO (val)--
348
Referring Expression ComprehensionRefCOCO (testA)--
346
Reasoning SegmentationReasonSeg (val)
gIoU65
327
Referring Expression SegmentationRefCOCO (testA)
cIoU79.1
315
Referring Expression ComprehensionRefCOCOg (val)--
300
Referring Expression ComprehensionRefCOCOg (test)--
300
Anomaly DetectionVisA--
293
Referring Expression SegmentationRefCOCO+ (testA)
cIoU70.8
288
Referring Image SegmentationRefCOCO (val)
mIoU74.9
274
Showing 10 of 393 rows
...

Other info

Code

Follow for update