OneRef: Unified One-tower Expression Grounding and Segmentation with Mask Referring Modeling

Constrained by the separate encoding of vision and language, existing grounding and referring segmentation works heavily rely on bulky Transformer-based fusion en-/decoders and a variety of early-stage interaction technologies. Simultaneously, the current mask visual language modeling (MVLM) fails to capture the nuanced referential relationship between image-text in referring tasks. In this paper, we propose OneRef, a minimalist referring framework built on the modality-shared one-tower transformer that unifies the visual and linguistic feature spaces. To modeling the referential relationship, we introduce a novel MVLM paradigm called Mask Referring Modeling (MRefM), which encompasses both referring-aware mask image modeling and referring-aware mask language modeling. Both modules not only reconstruct modality-related content but also cross-modal referring content. Within MRefM, we propose a referring-aware dynamic image masking strategy that is aware of the referred region rather than relying on fixed ratios or generic random masking schemes. By leveraging the unified visual language feature space and incorporating MRefM's ability to model the referential relations, our approach enables direct regression of the referring results without resorting to various complex techniques. Our method consistently surpasses existing approaches and achieves SoTA performance on both grounding and segmentation tasks, providing valuable insights for future research. Our code and models are available at https://github.com/linhuixiao/OneRef.

Paper

References (100)

Scroll for more · 38 remaining

Similar papers

Peer review

Reviewer SMKQ5/10 · confidence 3/52024-07-02

Summary

The authors propose a unified one-tower referring framework that introduces the MRefM paradigm to capture the referential relationships between vision and text. They demonstrate the effectiveness and generality of MRefM across three settings on REC, PG, and RES tasks, consistently achieving good results.

Strengths

The proposed method is structurally simpler than previous methods and achieves higher grounding and RES performance. The ablation experiments are thorough, validating the effectiveness of the two Ref mask modeling strategies.

Weaknesses

Adapting BEiT, a robust general-purpose multimodal model, to downstream grounding tasks and achieving performance improvement does not seem to be a particularly interesting finding. Additionally, it is puzzling why introducing two types of relation scores in MIM and MLM would enhance the model's referring performance. Especially regarding the four types of masks in the Visual target-relation score, their effective mechanism is not intuitive and lacks explanation. Why did the authors decide to introduce referring information by predicting these four scores? Are there any reference works? In Table 3, comparisons with some related works, such as UNINEXT [1] and HIPIE [2], are missing. [1] Universal instance perception as object discovery and retrieval. CVPR 2023. [2] Hierarchical Open-vocabulary Universal Image Segmentation. NeurIPS 2023.

Questions

See weaknesses.

Rating

5

Confidence

3

Soundness

3

Presentation

3

Contribution

3

Limitations

Yes.

Authorsrebuttal2024-08-14

We thank the reviewers for their efforts. As the discussion phase is nearing its end, we kindly remind the reviewer SMKQ to respond to our responses if it is convenient. We sincerely thank the reviewer SMKQ for the positive rating of our paper. Currently, no new questions have been raised by the reviewer, so we sincerely hope that our reply has successfully addressed his/her concerns. If not, we would sincerely appreciate any further comments that can improve the quality of our paper. We will incorporate all suggestions during this rebuttal to comprehensively revise our paper to achieve a higher standard! Lastly, we thank the reviewer SMKQ once again for the time and effort spent on reviewing our work! Thank you!

Reviewer vP755/10 · confidence 4/52024-07-04

Summary

This paper proposes a Mask Referring Modeling (MRefM) paragram and a unified and extremely concise grounding and referring segmentation framework named UniRef that no longer requires the fusion or interaction of the Transformer structure and the special grounding tokens. A masked referring modeling (MRefM) is proposed to model the referential relationship, which encompasses referring-aware mask image modeling and referring-aware mask language modeling.

Strengths

A mask-referring modeling paragram is proposed to model the referential relation between visual and language effectively. They also propose a one-tower framework for grounding and referring segmentation in a unified modality-shared feature space. Experiments demonstrate the effectiveness of this method and its components.

Weaknesses

1. Lack of discussion of related work [a]. I suggest discussing the difference between this paper's referring-aware mask language modeling and masked contrastive learning in [a]. 2. Lack of computation costs analysis with other methods, including parameters/FLOPs/speed. [a] VLT: Vision-Language Transformer and Query Generation for Referring Segmentation. In TPAMI 2023.

Questions

See weakness.

Rating

5

Confidence

4

Soundness

3

Presentation

2

Contribution

3

Limitations

Yes.

Authorsrebuttal2024-08-14

We thank the reviewers for their efforts. As the discussion phase is nearing its end, we kindly remind the reviewer vP75 to respond to our responses if it is convenient. We sincerely thank the reviewer vP75 for the positive rating of our paper. Currently, no new questions have been raised by the reviewer, so we sincerely hope that our reply has successfully addressed his/her concerns. If not, we would sincerely appreciate any further comments that can improve the quality of our paper. We will incorporate all suggestions during this rebuttal to comprehensively revise our paper to achieve a higher standard! Lastly, we thank the reviewer vP75 once again for the time and effort spent on reviewing our work! Thank you!

Reviewer a3fR6/10 · confidence 4/52024-07-14

Summary

This manuscript proposes UniRef, a framework aimed at unifying visual and linguistic feature spaces for referring expression comprehension and segmentation. The key presented is the Masked Referring Modeling (MRefM) paradigm, which includes referring-aware MIM and MLM. This approach seeks to streamline the architecture by eliminating the need for separate modality-specific encoders and complex interaction modules, achieving sota performance on several datasets.

Strengths

1. The introduction of the MRefM paradigm effectively captures the referential relationship between visual and linguistic features, contributing to the robustness and accuracy of the model. 2. The integration of visual and linguistic feature spaces into a unified framework simplifies the model architecture and potentially improves computational efficiency. 3. The authors provide extensive experimental results across five datasets, demonstrating the effectiveness of the proposed approach in outperforming existing methods.

Weaknesses

1. The technical contribution of the proposed method appears to be insufficient, as it primarily adapts traditional Masked Autoencoders into a 'Referring MAE', despite the rich experimentation presented. 2. Section 3.2 is intricate and challenging to follow. I recommend a thorough proofreading and revision of this section to enhance its clarity and readability. 3. Could you explain the rationale behind designing the system to encompass four masks: x-, y-, w-, and h-masks? 4. The manuscript asserts that the approach is lightweight, raising questions about its real-time applicability. It would be helpful if the authors detailed the computational demands and processing speed of the method in practical scenarios. 5. (Minor) While the manuscript has discussed the limitations of the proposed approach, I am interested in understanding how MrefM performs in more general contexts. For instance, evaluating the backbone linear probing (or its integration with LLMs) could demonstratebroader applicability. This should certainly be considered for future work.

Questions

See weakness.

Rating

6

Confidence

4

Soundness

3

Presentation

2

Contribution

2

Limitations

None.

Reviewer a3fR2024-08-13

Thank you for your response. After reviewing other reviews and responses, I have decided to increase my rating to 6. I hope you can carefully integrate the rebuttals into the revised version and thoroughly proofread the entire manuscript.

Authorsrebuttal2024-08-13

We sincerely thank the reviewer a3fR for the positive feedback and the increased rating of our paper! We would like to express our deep gratitude for the reviewer's valuable comments provided in this review, and we will incorporate all suggestions during this rebuttal to comprehensively revise our paper to achieve a higher standard and quality. Lastly, we thank the reviewer once again for the time and effort spent on reviewing our work! Thank you!

Program Chairsdecision2024-09-25

Decision

Accept (poster)

© 2026 NYSGPT2525 LLC