Source-linked AI summary
NExT-Chat: An LMM for Chat, Detection and Segmentation
Ao Zhang, Yuan Yao, Wei Ji, Zhiyuan Liu, Tat-Seng Chua
TL;DR
LMMs can describe images and answer visual questions, but holistic understanding and discrete-coordinate localization leave region reasoning and mask outputs limited. NExT-Chat introduces pix2emb, which decodes location embeddings into multiple formats, and reports stronger results than comparison systems across several tasks.
Problem
LMMs have been limited to holistic image understanding, while pix2seq’s discrete coordinate outputs struggle to support fine-grained formats such as segmentation masks.
Source: §1 · p. 1Method
Pix2emb models location information as embeddings, using <trigger> hidden states for localization and corresponding decoders to produce formats such as boxes and masks.
Source: §1 · p. 2Results
NExT-Chat reports 87.7 accuracy vs. Shikra’s 86.9 on POPE-Random, 68.9 average cIoU vs. LISA’s 67.9 on RES, and 79.6 CIDEr vs. Kosmos-2’s 62.3 on RefCOCOg region captioning.
Source: §1 · p. 2Takeaways & Limitations
NExT-Chat handles visual grounding, region captioning, grounded captioning, and complex question reasoning.
Source: §6 · p. 7Training primarily on individual-image inputs limits handling of multiple images, and insufficient diverse-domain training data hinders accurate medical and satellite image analysis.
Source: §7 · p. 7
Abstract
from arXiv · showhide
The development of large language models (LLMs) has greatly advanced the field of multimodal understanding, leading to the emergence of large multimodal models (LMMs). In order to enhance the level of visual comprehension, recent studies have equipped LMMs with region-level understanding capabilities by representing object bounding box coordinates as a series of text sequences (pix2seq). In this paper, we introduce a novel paradigm for object location modeling called pix2emb method, where we ask the LMM to output the location embeddings and then decode them with different decoders. This paradigm allows us to use different location formats (such as bounding boxes and masks) in multimodal conversations. Leveraging the proposed pix2emb method, we train an LMM named NExT-Chat and demonstrate its capability of handling multiple tasks like visual grounding, region captioning, and grounded reasoning. Comprehensive experiments show the effectiveness of our NExT-Chat on various tasks, e.g., NExT-Chat (87.7) vs. Shikra (86.9) on POPE-Random, NExT-Chat (68.9) vs. LISA (67.9) on referring expression segmentation task, and NExT-Chat (79.6) vs. Kosmos-2 (62.3) on region caption task. The code and model are released at https://github.com/NExT-ChatV/NExT-Chat.
1. Introduction
Existing LMMs can describe images and answer visual questions but lack region-level reasoning, while pix2seq-based localization is limited to discrete coordinates and struggles with masks. NExT-Chat introduces pix2emb for flexible location formats and reports results across multiple tasks.
- Earlier LMMs focused on holistic image understanding, and pix2seq’s discrete coordinate outputs struggle to represent fine-grained formats such as segmentation masks.Source: §1 · pp. 1–2
- Pix2emb represents locations as embeddings decoded into different formats, using <trigger> to initiate localization and <loc> as a placeholder for object-location embeddings.The embedding-based formulation treats localization as regression, allowing L1, IoU, and GIoU losses.Source: §1 · p. 2
- 87.7 accuracy vs. Shikra’s 86.9 on POPE-Random, 68.9 average cIoU vs. LISA’s 67.9 on RES, and 79.6 CIDEr vs. Kosmos-2’s 62.3 on RefCOCOg region captioning.Source: §1 · p. 2
- NExT-Chat combines chat, region input, detection, and segmentation in one LMM built on pix2emb.The paper also presents examples and results to demonstrate the method’s effectiveness.Source: §1 · p. 2
2. Related Works
Related work covers LMM architectures and instruction tuning, then traces region understanding from feature-based inputs to coordinate-token localization. NExT-Chat instead uses trigger hidden states to decode boxes and masks.
2.1. LMM
- LMMs use visual features for captions and question answering; Flamingo [1] uses cross-attention, while BLIP-2 [17] and Kosmos [11] feed visual features into LLMs as soft prompts.Source: §2.1 · p. 2
- Instruction-tuning approaches include MiniGPT-4 [46], LLaVA, Otter [15] [16], and LLaVA-1.5 [20], while earlier LMMs [1] [21] lack region understanding.Source: §2.1 · p. 2
2.2. LMM for Region Reasoning
- Region methods include GPT4ROI [44] feature inputs, Pix2seq [5] coordinate tokens, and extensions to vision-language tasks [36] [38] [32] [27]; Kosmos-2 [27] uses 32×32 image bins.Source: §2.2 · p. 2
- Shikra [4] reuses the original vocabulary for box coordinates but requires roughly 26 tokens per box; NExT-Chat instead decodes boxes and masks from <trigger> hidden states.Source: §2.2 · p. 3, Figure 2 · p. 3
3. Method
NExT-Chat combines a LLaVA-like multimodal architecture with pix2emb location modeling to support bounding-box detection and segmentation. Its training proceeds in three stages, extending dialogue and localization capabilities to segmentation.
3.1. LMM Architecture
- The LLaVA-like architecture projects 24×24 CLIP ViT-L/14@336px image patches into the LLM’s word-embedding dimension as visual tokens.The decoder-only LLM is Vicuna-1.5.Source: §3.1 · p. 3
3.2. Pix2Emb Method
- For detection, the <trigger> hidden state is passed to a 2-layer MLP box decoder that regresses coordinates, supervised by L1 and GIoU losses.The predicted box uses [x0, y0, x1, y1] coordinates.Source: §3.2 · p. 3
- For segmentation, the <trigger> hidden state is projected as a SAM prompt, optionally concatenated with a detected-box prompt, while SAM also receives the original image.Mask training uses IoU, Dice, and Focal losses, with β set to 20.Source: §3.2 · p. 3
- Cycle loss binds box encoder and decoder training.Source: Figure 3 · p. 4
3.3. Training Process
- Training proceeds through pre-training, instruction tuning, and segmentation; the final stage freezes most parameters, trains the LMM–SAM projector and SAM decoder, and uses RefCOCO, RefCOCO+, and RefCOCOg segmentation splits.The final-stage training takes 3 hours on 8 A100 (80G) GPUs.Source: §3.3 · p. 4
4. Experiment
The experiments pair qualitative demonstrations across multiple scenarios with quantitative comparisons on image-level hallucination, referring segmentation and detection, and region-level captioning.
4. Experiment
- Quantitative comparisons cover POPE image hallucination, RES segmentation, REC detection, and RefCOCOg region captioning.The reported table scopes identify POPE, RES, REC, and RefCOCOg as the evaluation settings.Source: §4 · p. 4, Table 1 · p. 5, Table 2 · p. 5
4.1. Applications across Different Scenarios
- NExT-Chat localizes queried objects, including each of four bears and the sky, and handles grounding questions about a TV remote and a boat.The boat example uses an object location supplied as input.Source: §4.1 · pp. 4–5
- For region captioning, NExT-Chat describes the specified region rather than the whole image, including small flags and background trees.The examples indicate that descriptions are tailored to the provided bounding boxes.Source: §4.1 · p. 5
- Grounded captions reference two or three major image objects and show a reduced tendency to mention nonexistent objects.The object references are organized into coherent sentences.Source: §4.1 · p. 5
- In detailed reasoning, NExT-Chat infers a man's occupation from contextual cues, localizes relevant regions, and explains possible duties for each hypothesis.The example uses the man's uniform and horse as contextual cues.Source: §4.1 · p. 6
5. Comparison with SOTAs
The section compares NExT-Chat with prior models across hallucination diagnosis, referring segmentation and detection, and region captioning.
5.1. Hallucination
- NExT-Chat ranks best on POPE random and popular splits and second on the adversarial split.The comparison is against state-of-the-art multimodal models.Source: §5.1 · p. 6
5.2. Referring Expression Segmentation
- NExT-Chat improves on LISA-7B in six segmentation dataset splits, including a 4.5-point gain on RefCOCO+ testA, with 127k masks versus over tenfold larger datasets.Against non-LMM methods, it is highest or second-highest across splits except RefCOCO+ val.Source: §5.2 · p. 7
5.3. Referring Expression Comprehension
- NExT-Chat beats VILLA, UNITER, and TranVG on every REC split, but scores slightly below Shikra-7B despite a similar detection-training data recipe.The authors hypothesize that loss balancing and the lack of regression pretraining contribute to the Shikra gap; they argue regression tasks remain important, especially for embodied AI.Source: §5.3 · p. 7
5.4. Region Caption
- NExT-Chat achieves the best CIDEr among the RefCOCOg region-caption baselines and beats the four-shot Kosmos-2 version.The evaluation uses CIDEr and METEOR.Source: §5.4 · p. 7
6. Conclusion
The conclusion presents pix2emb as a way to support multiple location formats and NExT-Chat as a model for varied region-aware tasks.
6. Conclusion
- Pix2emb represents locations as embeddings that support outputs such as bounding boxes and segmentation masks.The conclusion describes it as a location modeling method.Source: §6 · p. 7
- NExT-Chat handles visual grounding, region captioning, grounded captioning, and complex question reasoning.The paper reports exploratory experiments demonstrating the method's effectiveness.Source: §6 · p. 7
- Future directions include improving detection and segmentation and extending NExT-Chat to multimodal agents that handle complex region-understanding tasks.Source: §6 · p. 7
7. Limitation
NExT-Chat is limited in handling multiple images and in making accurate predictions for medical and satellite image analysis because its training data lacks diverse domains.
7. Limitation
- The training data is primarily single-image and lacks diverse domains, limiting multiple-image handling and prediction accuracy on medical and satellite images.Source: §7 · p. 7