월~금 오전 7시, 일요일 오전 8시 — AI 뉴스와 용어를 보내드립니다메일로 받아보기

METAL LAB

이미지-텍스트 매칭을 '텍스트에서 찾을 부분'까지 스스로 판단하게 만든 모델 ConCor-1

arXiv:2608.078862026-08-07

Vision-Language Grounding as Bidirectional Concept Correspondence

이미지-텍스트 매칭을 '텍스트에서 찾을 부분'까지 스스로 판단하게 만든 모델 ConCor-1

기존 비전-언어 그라운딩은 '이 단어가 어디 있냐'를 묻는 문제였지만, 이 논문은 '텍스트의 어느 부분이 이미지의 무엇과 대응하는가'를 양방향으로 동시에 찾는 문제로 재정의했다. 저자들은 학습 가능한 브릿지 토큰으로 이미지 영역과 텍스트 구간을 동시에 예측하는 ConCor-1을 만들어, 긴 캡션 데이터셋과 대규모 카테고리 목록(LVIS) 모두에서 기존 방법보다 큰 차이로 앞섰다. 저자는 Jieyu Zhang이며 논문은 arXiv 2608.07886로 공개됐다.

METAL LAB 해설 도표

ConCor-1이 텍스트-이미지 대응을 찾는 흐름

증거 상태측정 결과가 보고됨

  1. 입력이미지와 짝지어진 텍스트(캡션, 참조 표현, 또는 카테고리 목록)를 함께 받는다.
  2. 공유 백본 처리사전학습된 Qwen3.5가 이미지 토큰과 텍스트 토큰을 하나의 시퀀스로 처리한다.
  3. 브릿지 토큰 삽입학습된 브릿지 토큰들이 이미지·텍스트 토큰을 함께 어텐션으로 참조해 후보 대응관계를 형성한다.
  4. 세 갈래 예측각 브릿지 토큰에서 텍스트 마스크, 이미지 마스크, 대응 존재 점수(presence score)를 각각 예측한다.
  5. 결과 매칭Hungarian 매칭으로 예측된 대응들을 정답과 비교해 JointF1, mJS 등으로 평가한다.
METAL LAB이 원문을 바탕으로 재구성한 해설 도표이며, 논문 저자의 원문 figure가 아닙니다.

무엇을 했나

  1. 문제 재정의: 기존 방법은 미리 정해진 문구나 카테고리명을 주고 이미지에서 위치를 찾는 '한쪽 방향' 문제였지만, 이 논문은 텍스트의 어느 부분이 시각적으로 지칭 대상인지조차 모른 채로, 텍스트 구간과 이미지 인스턴스 마스크 사이의 모든 대응 관계를 복원하는 문제로 바꿨다.
  2. 모델 구조: 사전학습된 비전-언어 모델(Qwen3.5, 0.8B) 위에 학습 가능한 '브릿지 토큰'을 붙여 각 토큰이 이미지 마스크, 텍스트 마스크, 대응 존재 여부(presence score)를 동시에 예측하게 했다. 브릿지 토큰은 이미지 토큰과 텍스트 토큰을 함께 어텐션으로 참조해 후보 대응관계를 표현한다.
  3. 학습 데이터 구성: 기존 그라운딩·세그멘테이션 데이터셋(GoldG, COCONut-PanCap, COCO, EntitySeg 등)을 LLM으로 재작성해 공지시(coreference) 표현을 하나의 텍스트 마스크로 묶고, 이미지 마스크가 없는 언급은 제거해 통일된 대응 관계 형식으로 변환했다.
  4. 실험 결과: 긴 캡션 데이터셋인 COCONut-PanCap에서 가장 강한 베이스라인 대비 JointF1을 48%, mJS를 41% 개선했고, 제로샷 LVIS(대규모 카테고리 목록) 설정에서는 JointF1을 23.1에서 29.9로 29% 개선했다. 특히 ConCor-1은 LVIS의 전체 카테고리 목록을 한 번의 순전파로 처리할 수 있었다.
  5. 내부 분석: 브릿지 토큰과 텍스트/이미지 토큰 사이의 어텐션을 층별로 뽑아 마스크로 변환해보면, 언어모델의 깊은 층으로 갈수록 그라운딩 품질이 점차 좋아졌고, 최종 예측 헤드가 이를 다시 크게 끌어올렸다.
Figure 2: Model overview. ConCor-1 builds on a pretrained vision-language backbone and introduces learnable bridge tokens to represent candidate text–image correspondences. Given an image and paired text, vision tokens, language tokens, and bridge tokens are jointly processed by the language model. For each bridge token, the model predicts an image mask with a vision segmentation head, a text mask with a text segmentation head, and a presence score indicating whether the token corresponds to a valid grounded concept.
Figure 2: Model overview. ConCor-1 builds on a pretrained vision-language backbone and introduces learnable bridge tokens to represent candidate text–image correspondences. Given an image and paired text, vision tokens, language tokens, and bridge tokens are jointly processed by the language model. For each bridge token, the model predicts an image mask with a vision segmentation head, a text mask with a text segmentation head, and a presence score indicating whether the token corresponds to a valid grounded concept.
Table 1: Image-caption results on COCONut-PanCap, GroundedRef, and Flickr30k validation sets.
COCONut-PanCapGroundedRefFlickr30k
MethodJointF1MaskF1TextF1mJSmMaskIoUmSpanIoUJointF1MaskF1TextF1mJSmMaskIoUmSpanIoUJointF1MaskF1TextF1mJSmMaskIoUmSpanIoU
GDINO+SAM34.164.839.848.067.944.245.056.450.937.844.239.781.685.286.781.986.981.5
MM-GDINO+SAM32.259.537.451.170.847.439.858.144.331.944.233.384.687.789.682.687.382.3
LLMDet+SAM31.552.837.959.175.455.754.669.062.868.177.970.176.779.685.475.378.777.1
GPT-5.4 (medium)+SAM43.847.488.948.643.582.725.537.568.233.934.370.837.343.486.644.741.487.0
Florence-236.753.453.647.955.856.752.661.279.256.455.277.768.770.192.869.461.688.9
GLaMM2.539.13.011.927.48.06.147.310.018.432.816.953.169.159.555.460.358.2
Qwen3.5-FT59.963.493.163.355.687.951.556.382.149.745.877.570.272.695.269.160.395.1
ConCor-1 (random init.)51.859.467.663.263.677.932.443.456.944.648.170.555.061.172.764.462.383.7
ConCor-188.891.993.489.587.095.770.376.478.769.668.878.791.492.495.191.888.697.4
Figure 3: Grounding emerges progressively across language model layers. We probe bridge-to-text and bridge-to-vision attention across language-model layers and evaluate the induced masks with mSpanIoU, mMaskIoU, and mJS. Grounding quality improves progressively across layers, and the final prediction heads provide a further substantial boost.
Figure 3: Grounding emerges progressively across language model layers. We probe bridge-to-text and bridge-to-vision attention across language-model layers and evaluate the induced masks with mSpanIoU, mMaskIoU, and mJS. Grounding quality improves progressively across layers, and the final prediction heads provide a further substantial boost.
Table 2: Image-category results on COCO validation set, LVIS-minival, and EntitySeg validation set.
COCOLVIS-minivalEntitySeg
MethodJointF1MaskF1TextF1mJSmMaskIoUmSpanIoUJointF1MaskF1TextF1mJSmMaskIoUmSpanIoUJointF1MaskF1TextF1mJSmMaskIoUmSpanIoU
GDINO+SAM63.066.573.376.071.390.715.322.417.733.147.835.721.530.627.036.945.543.3
MM-GDINO+SAM65.768.775.677.172.091.69.418.412.453.675.756.829.436.937.045.551.755.7
LLMDet+SAM62.465.972.177.272.291.94.38.75.853.276.056.522.231.729.244.556.255.2
GPT-5.4 (medium)+SAM34.235.181.739.732.580.017.220.943.025.826.353.720.230.142.430.734.451.9
Florence-29.814.811.97.610.19.16.217.78.217.038.019.611.820.923.521.228.832.1
GLaMM38.540.845.733.131.538.923.124.029.719.418.424.122.223.126.018.317.421.2
Qwen3.5-FT39.441.251.437.337.448.33.926.95.24.428.05.231.138.647.633.138.043.6
ConCor-1 (random init.)15.132.822.925.149.535.60.118.50.20.232.50.32.820.45.44.326.66.5
ConCor-172.474.381.079.874.292.729.941.433.839.851.445.049.859.959.148.953.757.4
Figure 4: Qualitative comparison on COCONut-PanCap. We visualize ground truth, GDINO+SAM, and ConCor-1 predictions, using matching colors for grounded text spans and image masks. ConCor-1 better disambiguates referred instances and recovers more complete text–image correspondences, while GDINO+SAM can confuse different people or objects mentioned in the caption and may miss some grounded entities. Additional examples are provided in Appendix 8.
Figure 4: Qualitative comparison on COCONut-PanCap. We visualize ground truth, GDINO+SAM, and ConCor-1 predictions, using matching colors for grounded text spans and image masks. ConCor-1 better disambiguates referred instances and recovers more complete text–image correspondences, while GDINO+SAM can confuse different people or objects mentioned in the caption and may miss some grounded entities. Additional examples are provided in Appendix 8.
Table 3: Ablation study. We train each variant for 15K steps and report the average score for each metric across tasks. Ablations are performed sequentially; the vision encoder is frozen for the scoring-function, segmentation-head, and attention ablations, and fine-tuned for the subsequent vision-feature and bridge-assignment ablations. Bidirectional attention is used for the vision-feature and bridge-assignment ablations.
ComponentVariationJointF1MaskF1TextF1mJSmMaskIoUmSpanIoU
Scoring functionDot product48.755.460.359.959.773.3
Bilinear49.356.061.259.459.072.9
Image segmentation headBilinear-up (4x4)47.854.360.359.158.073.3
Two Conv-up (4x4)49.356.061.259.459.072.9
Three Conv-up (2x2)48.856.360.159.059.571.6
Four Conv-up (1x1)48.856.460.658.559.370.4
AttentionCausal49.356.061.259.459.072.9
Bidirectional51.357.262.661.460.474.7
Vision featureVision encoder feature + VLM vision tokens52.459.462.562.262.773.9
Vision encoder feature only51.859.161.961.762.773.4
VLM vision tokens only52.559.562.861.061.472.8
Bridge token assignmentRaster order (64 tokens)43.453.358.448.351.660.6
Spatial assignment (8x8 tokens)51.759.262.560.861.772.4
Spatial assignment (11x11 tokens)51.359.161.461.362.572.6
Multi-scale spatial assignment52.459.462.562.262.773.9
Figure 5: Text-to-image attention visualization. We visualize layer-20 attention from selected text tokens (highlighted in red) to image tokens for the original Qwen3.5, Qwen3.5 fine-tuned on our dataset, and our model. Compared with the diffuse or noisy baselines, ConCor-1 produces more localized and semantically aligned attention. Additional visualizations are provided in Appendix 9.
Figure 5: Text-to-image attention visualization. We visualize layer-20 attention from selected text tokens (highlighted in red) to image tokens for the original Qwen3.5, Qwen3.5 fine-tuned on our dataset, and our model. Compared with the diffuse or noisy baselines, ConCor-1 produces more localized and semantically aligned attention. Additional visualizations are provided in Appendix 9.
Table 4: Quantitative text-to-image attention analysis across full-attention layers, averaged over the image-caption evaluation datasets. Attn-IoU measures the overlap between the highest-attention visual tokens and the phrase-specific ground-truth mask, using the ground-truth mask area to determine the number of selected tokens. Top-1 Hit measures whether the visual token receiving the maximum attention lies inside the ground-truth mask. Entropy measures the normalized spatial dispersion of attention, where lower values indicate more concentrated attention.
MetricModelL4L8L12L16L20
Attn-IoU ↑Qwen3.50.160.200.220.220.21
Qwen3.5-FT0.160.270.230.190.15
ConCor-10.290.420.480.330.57
Top-1 Hit ↑Qwen3.50.320.430.430.430.40
Qwen3.5-FT0.410.560.470.360.27
ConCor-10.560.700.700.550.75
Entropy ↓Qwen3.50.820.690.730.740.73
Qwen3.5-FT0.780.730.710.700.63
ConCor-10.730.660.650.680.54
Figure 6: Bridge-token attention and prediction. We visualize two bridge tokens at layer 20 for one test sample, including bridge-to-image attention, bridge-to-text attention, spatial prior (red bounding box), and final prediction. Different bridge tokens specialize to distinct object-level correspondence hypotheses, such as the dog and the seated person. Additional examples are provided in Appendix 10.
Figure 6: Bridge-token attention and prediction. We visualize two bridge tokens at layer 20 for one test sample, including bridge-to-image attention, bridge-to-text attention, spatial prior (red bounding box), and final prediction. Different bridge tokens specialize to distinct object-level correspondence hypotheses, such as the dog and the seated person. Additional examples are provided in Appendix 10.
Table 5: Effect of the loss weights (λtext,λbce,λdice,λpres).
λtext:λbce:λdice:λpresJointF1MaskF1TextF1mJSmMaskIoUmSpanIoU
1 : 0.5 : 0.5 : 160.566.870.766.866.578.4
1 : 2 : 0.5 : 161.367.271.367.366.878.9
1 : 5 : 5 : 159.966.169.966.066.077.4
1 : 2.5 : 0 : 259.765.570.866.165.679.3
1 : 2 : 0.5 : 262.067.772.168.067.379.5
Figure 7: Spatial specialization of bridge tokens across token groups. We visualize the spatial activation patterns of bridge tokens from different scale groups under our multi-scale bridge-token design. Each group corresponds to a spatial partition of the image, ranging from a global 1×1 group to increasingly fine grids. Bridge tokens in coarse groups respond to broad image regions, while tokens in finer groups activate on increasingly localized areas aligned with their assigned spatial cells. This shows that bridge tokens learn structured spatial priors and progressively cover the image at multiple resolutions, supporting correspondence prediction for objects of different sizes and locations.
Figure 7: Spatial specialization of bridge tokens across token groups. We visualize the spatial activation patterns of bridge tokens from different scale groups under our multi-scale bridge-token design. Each group corresponds to a spatial partition of the image, ranging from a global 1×1 group to increasingly fine grids. Bridge tokens in coarse groups respond to broad image regions, while tokens in finer groups activate on increasingly localized areas aligned with their assigned spatial cells. This shows that bridge tokens learn structured spatial priors and progressively cover the image at multiple resolutions, supporting correspondence prediction for objects of different sizes and locations.
Table 6: Model size and computational profile. FLOPs, peak GPU memory, and latency are measured for one forward pass with a 512×512 image, 512 text tokens, batch size 1, bf16 precision, and an A100-80GB GPU.
ModelSeq. Len.FLOPsPeak Mem.LatencyParams
Qwen3.5-0.8B backbone7701.061 T1.687 GiB54.0 ms852.99 M
ConCor-11,1551.549 T1.846 GiB55.4 ms866.79 M
Δ+385+46.0%+9.5%+2.6%+1.62%
Figure 8: Qualitative visualization examples.
Figure 8: Qualitative visualization examples.
Table 7: Component-level parameter and FLOPs breakdown. The computation induced by the bridge tokens is included in the language-model row.
ComponentAdded ParamsGFLOPs
(M)BackboneConCor-1𝚫
Vision tower221.5221.5
Language model839.31,279.2+439.9
New bridge embeddings0.145
Visual segmentation head12.0146.3+46.3
Text segmentation head1.111.1+1.1
Presence head0.520.4+0.4
Total13.801,060.81,548.5+487.7
Figure 10: Qualitative visualization examples.
Figure 10: Qualitative visualization examples.
Table 8: Training data mixture used for ConCor-1. We report the source-level mixture weight in the final training setup. The image augmentation column indicates the fraction of each source group sampled with collage or crop augmentation; the remaining fraction is sampled without image augmentation.
Source datasetMix weightImage aug.
Instance Segmentation Data
COCO 2415%80% collage
EntitySeg 3410%10% crop
PixMo Points 610%10% crop
COCONut 75%70% collage, 10% crop
SA-1B 185%10% crop
ADE20K 575%50% collage, 20% crop
Roboflow-VL-100 375%
Caption Grounding Data
GoldG 2518%
COCONut-PanCap 715%
GroundedRef12%
Total100%
Table 9: Statistics of training data. Image counts are unique images per source and the total is their sum, so COCO images shared by COCO, COCONut-PanCap and GroundedRef are counted once per source.
Source datasetNum. of imagesNum. of instancesAnnotation TypeRole
Instance Segmentation Data
COCO 24117.3K860.0KHuman annotatedStandard benchmark
COCONut (Objects365 subset) 8242.6K3.6MHuman annotatedStandard benchmark
EntitySeg 348.1K79.1KHuman annotatedStandard benchmark
ADE20K 5725.6K639.5KHuman annotatedScene-centric objects
Roboflow-VL-100 (81 datasets) 3792.9K812.3KConverted benchmark dataDiverse domains
PixMo Points 656.8K1.7MAuto-constructedDense scenes + diverse text queries
SA-1B 1871.1K249.5KAuto-constructedDense scenes + diverse text queries
Caption Grounding Data
GoldG 2575.2K2.6MRewritten + pseudo masksHuman-caption grounding
COCONut-PanCap 7100.1K1.3MRewritten + GT masksHuman-caption grounding
GroundedRef60.2K230.5KSynthetic captions + GT masksReferring and compositional grounding
Total849.8K12.0M
Table 10: Statistics of the three image-caption benchmarks we curated or repurposed. Caption length is measured by whitespace-tokenized words per caption. Span multiplicity denotes the number of distinct character spans associated with each mask instance.
DatasetImagesCaptionsMasksCaption lengthSpan multiplicity
words/captionspans/mask
Flickr30k7792,0026,71711.01 ± 4.361.06 ± 0.26
GroundedRef1,3182,0256,19310.34 ± 4.201.04 ± 0.19
COCONut-PanCap2,2132,21310,82150.84 ± 16.891.68 ± 0.90
Table 11: Effects of annotation standardization and model architecture. Image-caption JointF1 is averaged over the three caption benchmarks, while image-category JointF1 is averaged over COCO and LVIS-minival.
Training annotationsMethodImage-captionImage-category
JointF1JointF1
Original sourceMM-GDINO-FT + SAM57.033.3
Original sourceConCor-160.844.9
StandardizedMM-GDINO-FT + SAM66.433.6
StandardizedConCor-170.645.2

실제로 확인된 결과

  • COCONut-PanCap(긴 캡션 데이터셋)에서 ConCor-1은 가장 강한 베이스라인 대비 JointF1을 59.9에서 88.8로, mJS를 63.3에서 89.5로 개선했다.
  • 제로샷 LVIS-minival에서 ConCor-1은 JointF1을 23.1에서 29.9로 개선했으며(29% 상대 개선), 카테고리 목록을 나눠 여러 번 추론해야 하는 다른 베이스라인과 달리 전체 목록을 한 번에 처리했다.
  • 자기회귀 생성 방식으로 텍스트 구간과 폴리곤 마스크를 직접 출력하는 Qwen3.5-FT는 텍스트 지표에서는 강했지만 마스크·joint 지표에서 ConCor-1보다 뒤졌다.
  • 어텐션 층별 분석에서 언어모델의 얕은 층은 그라운딩 정보가 거의 없었고 깊은 층으로 갈수록 mMaskIoU, mSpanIoU, mJS가 꾸준히 향상됐으며, 학습된 예측 헤드가 이를 최종적으로 크게 끌어올렸다.
  • 어노테이션 표준화(원문 재작성 파이프라인) 효과 분석에서, 표준화된 데이터로 학습하면 MM-GDINO-FT는 image-caption JointF1이 9.4점, ConCor-1은 9.8점 향상됐다.

어디에 쓸 수 있나

  • 이미지 캡션이 실제로 이미지 내용을 얼마나 정확히 지칭하는지(충실도) 자동 평가
  • 텍스트-이미지 생성 모델이 프롬프트의 각 표현을 실제로 이미지에 반영했는지 검증
  • 긴 서술문이나 복잡한 참조 표현이 포함된 이미지 설명에서 물체 단위로 텍스트-이미지 정렬 자동 생성
  • 대규모 카테고리 목록을 다루는 개방형 탐지·세그멘테이션 파이프라인에서 다회 추론 없이 한 번에 처리

한계와 남은 검증

  • 실험은 COCONut-PanCap, GroundedRef, Flickr30k(이미지-캡션)와 COCO, LVIS-minival, EntitySeg(이미지-카테고리) 등 특정 벤치마크에 한정돼 있어 다른 도메인 일반화는 검증되지 않았다.
  • 백본으로 0.8B 크기의 Qwen3.5를 사용했다는 점에서 훨씬 큰 모델이나 다른 백본에서의 결과는 논문에 제시되지 않았다.
  • 브릿지 토큰 수는 고정되어 있어, 한 이미지-텍스트 쌍에 포함된 실제 대응 개수가 이보다 많은 극단적 상황에서의 동작은 별도로 검증되지 않았다.
  • 어노테이션 표준화 효과는 15K 스텝만 학습한 축소된 실험 조건에서 측정된 것으로, 전체 학습 스케일에서의 효과는 별도로 보고되지 않았다.
  • Winograd 스타일의 지시대상 모호성 해소나 시각적 공지시(coreference) 해결 같은 확장 응용은 논문에서 개념적으로만 연결됐고 별도의 정량 실험은 제시되지 않았다.

왜 중요한가

이 방식은 '어떤 문구를 찾아라'고 미리 알려주지 않아도 이미지와 텍스트 사이의 대응을 통째로 복원할 수 있어, 캡션의 충실도 평가나 텍스트-이미지 생성 결과의 정합성 검증 같은 실무에 바로 쓸 수 있는 도구가 될 수 있다. 또한 대규모 카테고리 목록을 여러 번 나눠 돌리지 않고 한 번에 처리할 수 있다는 점은 실제 서비스에서 속도와 비용 이점으로 이어질 수 있다.

이 논문의 용어

  • 그라운딩(grounding) · 텍스트의 표현이 이미지의 어떤 부분을 가리키는지 연결하는 작업
  • 브릿지 토큰 · 이미지와 텍스트 사이의 후보 대응관계 하나를 표현하도록 학습되는 특수 토큰
  • JointF1 / mJS · 텍스트 구간과 이미지 마스크가 모두 맞았는지를 함께 평가하는 지표(임계값 기반 F1과 부드러운 점수)
  • 오픈 보캐뷸러리 탐지 · 미리 정해지지 않은 임의의 카테고리 이름으로 물체를 찾는 작업
  • presence score · 브릿지 토큰이 실제로 유효한 텍스트-이미지 대응을 나타내는지 판단하는 점수

본문에 싣지 못한 그림

  • Figure 1: Bidirectional concept correspondence. Given an image paired with different forms of text, including a full caption, a referring expression, or a list of category names, the concept correspondence model predicts a complete set of correspondences between text segments and image segments. For each correspondence, the model identifies the visually grounded text span, including repeated or coreferential mentions, and localizes the corresponding instance-level mask in the image.
  • Figure 9: Qualitative visualization examples.
원문에서 그림 보기 →

저자 · Jieyu Zhang

arXiv에서 원문 보기

최신 논문

논문 전체 보기 →

METAL LAB 최신 기사

그림 출처: Jieyu Zhang et al., arXiv:2608.07886, CC BY-SA 4.0