DRAgent: Discriminative Reasoning Agent for Referring Expression Segmentation
Organizations: School of Computer Science and Technology, Hangzhou Dianzi University, Hangzhou, 310018, China. · Khoury College of Computer Sciences, Northeastern University, Vancouver, BC V6B 1Z3, Canada.
Abstract
Referring Expression Segmentation (RES) aims to generate a pixel-level mask for the object specified by a language expression. Recent methods based on multimodal large language models (MLLMs) often rely on one-pass coordinate prediction for visual localization, which serializes continuous spatial locations as discrete text tokens and may lead to localization bias and alignment errors. To address these issues, we propose DRAgent, an MLLM-driven discriminative reasoning (DR) framework for RES. Instead of requiring the MLLM to generate localization coordinates, DRAgent first constructs a detector-generated candidate space and then uses the MLLM as a visual-semantic target discriminator. Specifically, the MLLM performs reliable target selection among potential distractors through a two-stage DR mechanism, which first screens high-recall candidates and then performs instance-wise verification. The selected target box is subsequently used as a spatial prompt for a foundation segmentation model to produce the final pixel-level mask. Furthermore, we construct a self-consistency-filtered reasoning-chain data pipeline for LoRA-based fine-tuning, providing more reliable supervision for enhancing the MLLM's discriminative reasoning capability. Experiments demonstrate that DRAgent achieves competitive performance on RefCOCO, RefCOCO+, and RefCOCOg.
Figures & tables
| Method | Venue | RefCOCO | RefCOCO+ | RefCOCOg | |||||
| val | testA | testB | val | testA | testB | val(U) | test(U) | ||
| Non-MLLM-based | |||||||||
| LAVT [ 1 ] | CVPR’22 | 72.7 | 75.8 | 68.8 | 62.1 | 68.4 | 55.1 | 61.2 | 62.1 |
| SEEM [ 24 ] | NeurIPS’23 | – | – | – | – | – | – | 65.7 | – |
| ReLA [ 2 ] | CVPR’23 | 73.8 | 76.5 | 70.2 | 66.0 | 71.0 | 57.7 | 65.0 | 66.0 |
| PolyFormer [ 11 ] | CVPR’23 | 75.96 | 78.29 | 73.25 | 69.33 | 74.56 | 61.87 | 69.20 | 70.19 |
Referring Expression Segmentation
Referring expression segmentation maps natural-language descriptions to the pixels or 3D points of the objects they identify.
See all 28 papers