Boosting Weakly-Supervised Referring Image Segmentation via Progressive Comprehension

This paper explores the weakly-supervised referring image segmentation (WRIS) problem, and focuses on a challenging setup where target localization is learned directly from image-text pairs. We note that the input text description typically already contains detailed information on how to localize the target object, and we also observe that humans often follow a step-by-step comprehension process (\ie, progressively utilizing target-related attributes and relations as cues) to identify the target object. Hence, we propose a novel Progressive Comprehension Network (PCNet) to leverage target-related textual cues from the input description for progressively localizing the target object. Specifically, we first use a Large Language Model (LLM) to decompose the input text description into short phrases. These short phrases are taken as target-related cues and fed into a Conditional Referring Module (CRM) in multiple stages, to allow updating the referring text embedding and enhance the response map for target localization in a multi-stage manner. Based on the CRM, we then propose a Region-aware Shrinking (RaS) loss to constrain the visual localization to be conducted progressively in a coarse-to-fine manner across different stages. Finally, we introduce an Instance-aware Disambiguation (IaD) loss to suppress instance localization ambiguity by differentiating overlapping response maps generated by different referring texts on the same image. Extensive experiments show that our method outperforms SOTA methods on three common benchmarks.

Paper

Similar papers

Peer review

Reviewer K7SG7/10 · confidence 4/52024-07-02

Summary

Aiming for the WRIS task, the work proposes a novel framework to leverage target-related textual cues from the description for progressively localizing the target object. The authors use a Large Language Model (LLM) to decompose the input text description into short phrases which are taken as target-related cues and fed into a Conditional Referring module. Besides, Region-aware Shrinking (RaS) loss and Instance-aware Disambiguation (IaD) loss are proposed to facilitate fine-grained cross-modal alignment.

Strengths

1. The motivation is clear and easy to understand. The idea of leveraging target-related textual cues decomposed by LLM for progressive localization makes sense. 2. This work develops multiple consecutive Conditional Referring Modules. RaS and IaD two loss objectives are seamlessly integrated into the framework. 3. It is novel to implement RaS and IaD two loss objectives by harnessing the capabilities of segmentation foundation models. The SAM[1] does not include strong semantic prior knowledge like GroundingDINO and Grounded-SAM[2], and it is suitable for the weakly-supervised setting. 4. The work shows outstanding object localization ability and outperforms counterparts on three common benchmarks. Especially after refined by the SAM, the results are promising. [1] Segment anything ICCV2023 [2] Grounding dino: Marrying dino with grounded pre-training for open-set object detection

Weaknesses

1. In table 1, only the results refined by SAM[1] are reported. Considering that FreeSOLO [2] is an unsupervised segmentation model, I am curious about the results refined by FreeSOLO. 2. This work introduces multiple stages for progressive localization. It may increase the training and inference time of the model. The time comparison of model training and inference may be needed. [1] Segment anything ICCV2023 [2] Freesolo: Learning to segment objects without annotations CVPR2022

Questions

1. The mask proposals by SAM are often part of the whole instance. It will affect the refined results. Does this work do anything special to alleviate the problem? The author can give more details about this issue. 2. Table 1 shows the results of the pseudo labels obtained at the wealy-supervised training stage. Can the authors give the results after pseudo labels based on supervised training?

Rating

7

Confidence

4

Soundness

4

Presentation

4

Contribution

4

Limitations

NaN

Reviewer Tfcw5/10 · confidence 5/52024-07-02

Summary

This paper proposes the Progressive Comprehension Network (PCNet) for the WRIS task. this model achieves visual localization by progressively incorporating target-related textual cues for visual-linguistic alignment. Although experimental results have demonstrated the effectiveness of this paper, several aspects of the paper lack clarity.

Strengths

1. The corresponding experiments validate the effectiveness of the proposed method on three popular benchmarks. the proposed method outperforms existing methods. 2. The two loss functions introduced in this paper effectively enhance the accuracy of the response maps.

Weaknesses

1. The Conditional Referring Module (CRM) is implemented through multiple cross-attentions, which is a relatively common approach, lacking novelty. 2. What is the general idea of classification loss mentioned in the paper? Please explain the fundamental research insights of this work. 3. In the Region-aware Shrinking (RaS) loss, what does the ⊙ symbol represent in Equation 5? Please clarify its meaning. 4. This article only conducted ablation experiments on the RefCOCOg dataset, so the ablation experiments are insufficient. It is recommended to validate on multiple datasets (such as RefCOCO+, RefCOCO) to comprehensively evaluate the effectiveness of our method. 5. The order of the tables is somewhat confusing, with Table 9 preceding Table 6. It is recommended to reorder the tables to make them more in line with logical sequence or reading habits. 6. Which datasets were the ablation experiments in Tables 7-9 conducted on? Please specify.

Questions

None

Rating

5

Confidence

5

Soundness

3

Presentation

2

Contribution

2

Limitations

See the weakness

Reviewer kTSb5/10 · confidence 3/52024-07-12

Summary

Inspired by human's step-by-step cognitive process for localizing a target object in an image, the paper proposes Progressive Comprehension Network (PCNet) for the task of weakly-supervised referring image segmentation (WRIS) where text is the only supervision signal. PCNet first decomposes a long, complex text description into multiple target-related cues using a large language model (LLM), and the specialized designed module, Conditional Referring Module (CRM), progressively refines the text-to-image response map using the generated text cues in multiple stages. Region-aware Shrinking (RaS) is introduced for constraining the latter stage to have a more compact, target-related response map, and Instance-aware Disambiguation (IaD) loss is proposed for differentiating between the target object and the other objects in the input image. The experiment results show the superiority of the proposed method.

Strengths

- The paper provides extensive experiment results proving the validity of the proposed method, PCNet, and also ablation studies showing the contributions of main components of the model. - Figures help readers understand the proposed method.

Weaknesses

- The proposed method uses the contrastive loss of TRIS, but not generates independent response maps from a positive cue and L negative cues. In Equation (3) and (4), it integrates all the text cue features and generates a single response map R_n that contains information from both positive and negative cues. This does not make sense. How did the authors differentiate between positive and negative cues when computing the classification loss L_Cls? (This is why I rated "poor" for presentation). - IaD, the loss to differentiate between the target object and other objects in the same image is similar to the calibration loss used in TRIS; the idea is the same while the calibration loss uses CLIP image-text similarity for differentiation. The authors should have discussed about this loss in the paper, and also done an ablation study comparing the two losses. - The PCNet uses the mask proposals generated by a pre-defined segmentation mask generator such as FreeSOLO and SAM. We cannot ignore the possibility of knowledge being distilled from the mask generator into PCNet, especially through RaS and IaD losses, main contributions of the paper. More specifically, we cannot be sure that the performance gains from using RaS and IaD do not come from the mask generator's knowledge.

Questions

Please rebut the above weaknesses. I also have minor questions: - In Figure 2, the arrow comes from Q_{n+1} to Cls. How is Q_{n+t} used for computing the classification loss? - How did the authors sample L negative textual cues? one cue from each of the examples in the same mini-batch? - I am curious about the rationale behind Equations (2) and (3). Why did the authors use residual connection only after the mlp layer, not also after the self-attention layer like in Transformer? - Does the order in which text cues are used affect the performance of the model? E.g., is there a performance difference between feeding into the CRM module in the order of q_0, q_1, q_2 and in the order of q_0, q_2, q_1?

Rating

5

Confidence

3

Soundness

2

Presentation

1

Contribution

2

Limitations

The paper addressed the limitations of the work.

Reviewer kTSb2024-08-10

Thank the authors for the detailed response. The original formulation of the classification loss was quite confusing, but now it makes sense. Also, the comparison with the baseline + IoU loss convinces me of the contribution of the RaS loss. The authors also addressed my other concerns. It depends on further discussion with other reviewers, but I am leaning towards raising my rating. Please add the conducted comparison experiments with the calibration loss in the final draft.

Authorsrebuttal2024-08-13

Thanks for your response and your willingness to consider raising the score. We genuinely appreciate your time and the valuable feedback. We are pleased that our rebuttal has addressed your concerns. However, we notice that your current rating still suggests that our work is still not qualified to be accepted. Considering that the discussion period is ending soon, if there are any remaining issues or concerns, we would greatly appreciate the opportunity to discuss them further with you.

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

Summary

This paper proposes a Progressive Comprehension Network for weakly-supervised referring image segmentation, which mimics the human process of progressive understanding by breaking down sentences into segments and gradually narrowing down the target range. The main contributions include: A multi-stage Conditional Referring Module to progressively comprehend text cues. A Region-aware Shrinking Loss to constrain the target region to gradually shrink. An Instance-aware Disambiguation Loss to eliminate overlap between different instances.

Strengths

1. Presentation: The majority of the paper is well-written and easy to follow, providing clear explanations of the key concepts. 2. Motivation: The motivation is reasonable and convincing. By mimicking the human process of understanding concepts from coarse to fine, the method progressively refines the final mask. Directly requiring the mask to continuously shrink could lead to trivial solutions, and two losses are introduced to further support this pipeline. The proposed method aligns well with the motivation. 3. Experiments: Comprehensive ablation experiments demonstrate the effectiveness of the proposed modules

Weaknesses

1. The concept of mimicking the human process of understanding concepts from coarse to fine-grained, along with multi-stage refinement for different parts of sentences, has been explored in prior REC research [1]. This approach involves parsing sentences into multiple parts and conducting multi-stage refinement. Applying REC's ideas [1] to WRIS can also be considered a contribution, but it should be thoroughly discussed in comparison to [1], highlighting any differences, and cited appropriately. 2. There is a lack of discussion and citation of recent WRIS work, such as [2], in related work and the main text. It is necessary to discuss and cite the latest WRIS work in both the related work section and the main table. 3. Section 3.4 is not entirely understandable. I suggest the authors re-describe this section to make it clearer and easier to understand. 4. The top of Fig.2 is not clear enough, especially the subscripts a and d, I suggest the author re-design this part. [1] Yang S, Li G, Yu Y. Dynamic graph attention for referring expression comprehension[C]//Proceedings of the IEEE/CVF International Conference on Computer Vision. 2019: 4644-4653. [2] Dai Q, Yang S. Curriculum Point Prompting for Weakly-Supervised Referring Image Segmentation[C]//Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition. 2024: 13711-13722.

Questions

1. Why is the range of Equation 6 set to [0,1]? Can the combination of other proposals (denoted as 𝑚_𝑏 ) be simply approximated by the complement of the foreground, since the proposals segmented by SAM nearly cover the entire image? In this case, if IoU(𝑅_𝑛,𝑚_𝑓) < IoU(𝑅_𝑛,𝑚_𝑏), wouldn't Equation 6 be greater than 1? 2. Does 𝑚_𝑓 change at different stages in Equation 7, for instance, when different proposals are selected at different stages? What issues could this cause? In Equation 7, are 𝑛 and 𝑛+1 reversed? According to Section 3.3, the ambiguity score should be lower at later stages. 3. Section 3.1 mentions that there are five phrases for each sentence, and Section 3.2 states that each stage uses one text cue, so there should be five stages. Why do the implementation details mention only three stages? 4. In Section 3.4, how do you sample extra 𝑁_𝑑 texts? Are they randomly sampled from multiple text descriptions corresponding to the same image? How do you ensure that the sampled text descriptions correspond to different target referents? Because in refcoco series dataset, one referent may have multiple text expressions. 5. In the bottom part of Tab.1, are the numbers all obtained by using a single peak point as the prompt for SAM? Did you try using the response map as the prompt for SAM?

Rating

7

Confidence

4

Soundness

2

Presentation

2

Contribution

2

Limitations

Yes.

Authorsrebuttal2024-08-13

Thanks very much for your positive feedback that we have addressed your most concerns and your willingness to raise the score. Here we give further explanations about IaD loss and Q4. - IaD loss. Thanks for your careful comments! In GroupViT, the Grouping Block groups similar semantic regions using the hard assignment strategy during the group tokens training and inferring process. In our IaD loss, we also adopt a similar hard assignment for deriving the loss function. The reason is that we aims to get the pseudo mask prediction by the accurate peak value point (i.e., the hard assignment results) instead of relying on whole score distribution $S(\cdot)$ (e.g., $S_{a}$, $S_{d}$ in Section 3.4). Thus utilizing the hard assignment to derive the IaD loss well matches our purpose, which helps rectify the ambiguous localization results. If we use the soft assignment (e.g., measuring KL divergence between $S_{a}$ and $S_{d}$), though the equivalent may be simpler, it not only does not match our purpose but also introduces more tricky components for optimization (e.g., extra distribution regularization is required). In order to verify the argument, we conduct a comparison on RefCOCOg(G) val dataset as follows. The KL_Loss even causes a slight decline, while the IaD loss brings a clear improvement on localization. | Methods | PointM | mIoU | | :-------------------------------------- | ------ | ---- | | $\mathcal{L}_{\texttt{CLs}}$ | 51.7 | 25.3 | | $\mathcal{L}_{\texttt{CLs}}$ + KL_loss | 51.2 | 24.8 | | $\mathcal{L}_{\texttt{CLs}}$ + IaD_loss | 53.1 | 26.6 | - Q4. Yes, you are right. `refer_id` is important information for the data_loader sampler and adopted in TRIS and SAG. Here, we further consider and utilize the prior (i.e., different texts corresponding to different instances within the same image should activate different regions) for localization optimization. If there are any additional clarifications needed, please do not hesitate to reach out.

Reviewer oLx42024-08-13

Thank you for your detailed response, which addressed all my concerns. I decide to raise the score. Please include the previously mentioned discussions in the final revision.

Authorsrebuttal2024-08-13

Official Comment by Authors

We thank Reviewer oLx4 for reviewing our work and for raising the review score. We really appreciate it. We will include the discussions in our revision.

Reviewer K7SG2024-08-08

I would like to thank the authors for their detailed responses. All my concerns were thoroughly addressed. Overall, this paper presents a reasonable and well-explained approach to mimicking the progressive process of understanding language instructions. By leveraging visual localization and segmentation tools (e.g., SAM), the method achieves superior language comprehension and target localization, supported by solid experiments and extensive visualization analysis. This work provides clear insights and significant contributions to the field. Therefore, I will maintain my original rating. I also recommend that the authors include suitable explanations and new results in the revision.

Reviewer Tfcw2024-08-12

Thanks for the rebuttal. My concerns are addressed, therefore I increase my score.

Authorsrebuttal2024-08-12

Thanks very much for your positive feedback! We deeply appreciate that you decided to increase the score. We'll include the explanations and more convincing results in our revision according to your careful comments.

Reviewer oLx42024-08-13

Thank you for your detailed response, which addressed most of my concerns. I would like to further discuss the IaD loss and Q4. In GroupViT, a similar operation is used to backpropagate gradients, but they do this to achieve hard assignment during inference, while the gradients are computed in a soft form. However, in this paper, there is no need for hard assignments during inference. Is there a simpler equivalent form to replace the current IaD loss? Regarding the answer to Q4, can I understand that you utilize the prior of different texts corresponding to different instances within the same image using **refer_id** ? Is it used in TRIS? I will raise my score if you can provide an explanation. Thank you very much!

Program Chairsdecision2024-09-25

Decision

Accept (poster)

© 2026 NYSGPT2525 LLC