WACV 2027 · Algorithms Track

ATLAS

Learning to Adapt LVLM Attention for Reasoning Video Segmentation

Attention Transformation for Localization through Adaptive Segmentation

Hamidreza Dastmalchi1 Ali Cheraghian2 Aijun An1 Abdelwahed Khamis3 Lars Petersson3

1York University 2Macquarie University 3CSIRO Data61

ATLAS segmenting a green moving truck across a ReasonVOS video for the expression 'Which vehicle would be most helpful for me to move (e.g. furniture…'
ATLAS on ReasonVOS. The expression never names the object (“Which vehicle would be most helpful for me to move (e.g. furnitur…”). ATLAS adapts the frozen LVLM's attentions into localization maps, and a frozen SAM2 follows the truck through the clip.

Abstract

Reasoning Video Object Segmentation (RVOS) aims to localize and segment objects referred to by natural-language expressions throughout a video. Although large vision-language models (LVLMs) implicitly encode localization cues in their attention maps, these attentions are often noisy, diffuse, and scattered across layers, heads, and modalities, limiting their direct use for segmentation. Existing methods typically rely on handcrafted attention-selection heuristics or computationally expensive attention refinement. We propose ATLAS, an attention adaptation framework that learns to transform raw LVLM attentions into object-centric localization cues. ATLAS treats attentions from all layers, heads, and modalities as structured features, combining high-resolution frame-level attentions that capture spatial information with low-resolution video-level attentions that encode temporal context. A lightweight U-Net equipped with temporal attention and FiLM-based modulation fuses these complementary representations to generate coarse segmentation heatmaps, which are propagated across the video using SAM2. Extensive experiments on challenging referring and reasoning video segmentation benchmarks show that ATLAS outperforms existing frozen-backbone methods.

Why is this hard?

Reasoning expressions call for an LVLM, but getting a precise location out of a frozen LVLM is unreliable.

✕ Fine-tuning

Jointly tuning LVLM + SAM

End-to-end fine-tuning works, but it optimizes several large foundation models together. That costs a lot of compute, needs large annotated datasets, and ties the adaptation to one specific LVLM and segmenter.

✕ Unreliable & scattered

Asking the LVLM for (x, y)

LVLMs are trained for text generation, not coordinate prediction. Querying Qwen2.5-VL 200 times at temperature 0.7 gives coordinate estimates scattered widely across the image.

✕ Unreliable & noisy

Averaging attention maps

Visual-token attentions do encode localization, but averaging over all layers and heads gives diffuse, noisy maps. Prior training-free methods hand-pick a few heads or refine attention with repeated inference.

The gap: much of the localization signal is spread across layers and heads and goes unused. In video, the model also has to work out when the referred object becomes relevant, not only where it is. ATLAS learns to adapt the full multimodal attention stack into object-centric localization cues, keeping the LVLM and SAM2 frozen.

ATLAS Overview

ATLAS architecture. (1) Attention Extraction: the expression with an object-extraction prompt and keyframes 1 to 16 of the video clip go into a frozen LVLM; attention maps from heads 1 to H of layers L0 to L are stacked into tensors A1 to ATs. (2) Attention Adaptation: a U-Net with Conv + Pooling encoders, GAP, a Transformer, a linear layer producing (gamma, beta), FiLM modulation Z(1+gamma)+beta, and Upsample + Conv decoders with skip connections outputs localization maps P1 to PTs. (3) Segmentation and Propagation: a frozen SAM produces tracklets 1 to M, and Tracklet Selection gives the final segmentation.
Overview of ATLAS. A frozen LVLM extracts visual attentions from sampled keyframes. A lightweight attention adaptation network turns them into object-centric localization maps. These maps guide a frozen SAM2 model, which generates and propagates candidate segmentation tracklets, and the most reliable tracklet is selected as the final segmentation. (Click the figure for the full-resolution PDF.)

Given a video 𝒱 = {It}Tt=1 and a reasoning expression 𝓔, ATLAS predicts binary masks 𝓜 = {Mt}Tt=1 in three stages. Only the attention adaptation network is trained. The LVLM and SAM2 stay frozen.

How ATLAS works

  1. 1

    Keyframes & prompt

    Uniformly sample Ts keyframes (16 by default). Wrap the expression in a fixed object extraction prompt and have the LVLM generate a single response token.

  2. 2

    Frame-level attention

    Query each keyframe separately with a high visual-token budget (1024). Stack the response token's attention over all heads of layers L0…L, with L0 the middle layer, to get A(f)k ∈ ℝC×Rf×Wf. This captures where the object is.

  3. 3

    Video-level attention

    Run one joint pass over all keyframes with a small budget (64 tokens) to get A(v)k. This captures when the object is relevant. Resize it and concatenate it with the frame-level stack: Ak ∈ ℝ2C×Rf×Wf.

  4. 4

    U-Net + temporal FiLM

    A two-stage U-Net encodes Ak into Zk. A two-layer Transformer runs over the GAP vectors z1…Ts, and the result produces γk, βk, which modulate the bottleneck: Z′k = Zk ⊙ (1+γk) + βk. The decoder outputs Pk.

  5. 5

    Point prompts → SAM2

    Each map's maximum pk = argmax Pk becomes a positive point that prompts a frozen SAM2 on its keyframe.

  6. 6

    Seed de-duplication

    Starting from the earliest seed, propagate its mask through the video. Drop any later seed whose mask has IoU > τ = 0.8 with the propagated mask. Repeat until no seeds are left, giving M distinct tracklets.

  7. 7

    Tracklet selection

    Pick the tracklet with the highest per-keyframe Pearson correlation to Pk, averaged over keyframes. If the maximum correlation is below 0.1, return an empty mask.

Training & implementation details
  • Loss: 𝓛 = 𝓛BCE + 𝓛Dice between Pk and the ground-truth mask Mtk.
  • Adapter: U-Net with two encoder and two decoder stages of two 3×3 convolutions each, BN + ReLU, and a 512-d bottleneck. Input is 2C = 784 channels (C = 392 per modality) on a 24×42 grid. Temporal Transformer: 2 layers, 4 heads.
  • Training data: Ref-YouTube-VOS training split only.
  • Optimization: AdamW, lr 1×10−4 → 1×10−5 (linear decay), weight decay 10−4, batch size 8 clips, 20 epochs. About 6 GPU hours on a single NVIDIA RTX Pro 6000.
  • Backbones: LLaVA-OV-7B, InternVL3-8B, Qwen2-VL-7B, Qwen2.5-VL-7B. Segmentation model: SAM2-Hiera-Large.
Object extraction prompt
{Expression}

What is the main object (or objects) referred to in the given expression or question?
Focus on the primary subject or agent involved in the described action or behavior.
Respond with a single word (e.g., "cat", "person", "dog") that best describes the target object(s).

What is the main idea?

Treat the whole attention stack as a feature map and learn how to read it, instead of hand-selecting or repeatedly refining attention maps.

Full attention stack as features

Attentions from every head of the middle-to-last layers are stacked as spatial feature channels. A supervised adapter learns to turn them into object-centric localization cues, with no handcrafted head selection and no costly refinement.

Where + when

High-resolution frame-level attentions provide spatial detail. Low-resolution video-level attentions provide temporal context. A temporal Transformer and FiLM inject this context into each keyframe's spatial decoding.

Lightweight & frozen

Only the small adapter is trained (≈6 GPU hours, Ref-YouTube-VOS only). The LVLM and SAM2 stay frozen, and ATLAS achieves state-of-the-art results among frozen-backbone methods.

80.1𝒥&𝓕 on Ref-DAVIS
(Qwen2.5-VL-7B)
67.4𝒥&𝓕 on ReasonVOS
(Qwen2.5-VL-7B)
10.1 → 60.6keyframe localization 𝒥&𝓕
raw attention → ATLAS (Ref-DAVIS)
2.7×end-to-end speedup
over DecAF (matched setting)

Quantitative Results

ATLAS is trained only on Ref-YouTube-VOS and is evaluated across LVLM backbones. It outperforms frozen-backbone methods and is competitive with fully fine-tuned approaches.

Ref-DAVIS, ReasonVOS and ReVOS

Table 1. Comparison of LVLM-based VOS methods. †Loc-Head and DecAF are training-free. ATLAS trains a lightweight adapter with mask supervision. All three keep the LVLM and segmentation model frozen. Best frozen-backbone result per backbone in bold.
MethodLVLMRef-DAVISReasonVOSReVOS (Overall)ReVOS (Referring)ReVOS (Reasoning)
𝒥&𝓕𝒥𝓕𝒥&𝓕𝒥𝓕𝒥&𝓕𝒥𝓕𝒥&𝓕𝒥𝓕𝒥&𝓕𝒥𝓕
Approaches with tuned foundation models
LISALLaVA-7B64.862.267.331.129.133.140.939.142.745.744.347.136.133.838.4
VISAChatUniVi-7B69.466.372.5–––46.944.949.050.949.252.643.040.645.4
GLUSLLaVA-7B–––49.947.552.454.952.457.358.356.060.751.448.853.9
VideoLISALLaVA-Phi-3-V68.864.972.747.545.149.9–––––––––
VRS-HQChatUniVi-7B76.072.679.4–––59.156.661.662.159.864.556.153.558.7
Veason-R1Qwen2.5-VL-7B–––59.956.063.861.358.264.463.660.766.559.055.862.2
Approaches with frozen foundation models
Loc-Head†LLaVA-7B56.352.160.533.629.338.032.528.236.936.932.541.328.123.832.5
DecAF†LLaVA-OV-7B59.454.864.052.849.356.340.035.844.143.439.147.636.632.640.7
ATLASLLaVA-OV-7B68.665.971.353.049.856.248.044.851.353.450.456.443.039.546.6
Loc-Head†InternVL3-8B66.362.470.244.341.047.543.739.947.546.742.950.640.736.944.5
DecAF†InternVL3-8B62.856.968.658.955.162.747.443.751.251.747.955.543.239.546.8
ATLASInternVL3-8B74.471.577.457.254.260.253.550.856.260.057.462.647.444.550.3
Loc-Head†Qwen2-VL-7B61.958.065.834.031.836.244.040.847.252.749.156.235.432.638.2
DecAF†Qwen2-VL-7B64.159.468.952.549.056.045.341.649.052.748.956.437.934.341.5
ATLASQwen2-VL-7B75.572.178.964.461.867.053.850.856.959.856.962.648.344.951.7
Loc-Head†Qwen2.5-VL-7B64.660.268.941.137.944.347.043.350.753.149.356.940.837.244.4
DecAF†Qwen2.5-VL-7B75.270.979.563.960.567.254.250.158.258.754.862.649.745.453.9
ATLASQwen2.5-VL-7B80.176.883.467.464.770.256.253.559.061.558.964.151.048.153.9

MeViS and Ref-YouTube-VOS

Table 2. †ATLAS is trained on the Ref-YouTube-VOS train set and evaluated on its val set. All other results are zero-shot.
MethodLVLMMeViSRef-YouTube-VOS†
𝒥&𝓕𝒥𝓕𝒥&𝓕𝒥𝓕
VISAChat-UniVi-7B44.541.847.161.559.863.2
VideoLISALLaVA-Phi-3-V44.441.347.663.761.765.7
GLUSLLaVA-7B51.348.554.267.365.569.0
VRS-HQChat-UniVi-7B50.647.653.770.468.372.5
Veason-R1Qwen2.5-VL-7B52.248.456.0–––
Loc-HeadQwen2.5-VL-7B39.435.543.651.046.855.2
DecAFQwen2.5-VL-7B48.144.052.159.956.263.5
ATLASQwen2.5-VL-7B54.551.058.069.567.771.3
ATLASQwen2-VL-7B52.148.455.871.069.272.8
ATLASLLaVA-OV-7B46.042.249.959.958.161.7
ATLASInternVL3-8B51.447.954.866.064.267.8

Attention localization quality

Table 3. Adapter output thresholded at 0.5 on sampled keyframes vs. normalized averaged raw LVLM frame-level attention, with no SAM2 and no post-processing.
DatasetMethod𝒥𝓕𝒥&𝓕
Ref-DAVISATLAS47.274.060.6
Raw Attn.2.018.210.1
ReasonVOSATLAS40.357.448.9
Raw Attn.4.327.315.8

Adaptation raises keyframe localization 𝒥&𝓕 from 10.1 to 60.6 on Ref-DAVIS and from 15.8 to 48.9 on ReasonVOS, turning diffuse LVLM attentions into reliable cues for mask propagation.

Runtime (NVIDIA RTX PRO 6000)

Table 8. Seconds per video, Qwen2.5-VL-7B on ReasonVOS.
Methodvid / frmAttn.Adapt.SeedSAM2E2E𝒥&𝓕
ATLAS (ours)64 / 10242.060.0030.00113.215.367.4
ATLAS256 / 10242.460.0030.00113.115.667.5
DecAF256 / 10248.08–0.03033.241.363.9

At the matched 256-video-token setting, ATLAS is 2.7× faster end-to-end (15.6 s vs. 41.3 s) and more accurate (67.5 vs. 63.9).

Ablation Studies

All ablations use Qwen2.5-VL-7B.

Video attention & temporal FiLM

Frame only60.4
Frame + FiLM61.5
Frame + Video64.9
Frame + Video + FiLM67.4

ReasonVOS 𝒥&𝓕 (axis 55–70)

Table 4. Temporal and video ablation.
MethodRef-DAVISReasonVOS
𝒥𝓕𝒥&𝓕𝒥𝓕𝒥&𝓕
Frame only75.080.177.557.863.060.4
Frame + Video76.382.079.162.267.764.9
Frame + FiLM75.581.378.458.864.161.5
Frame + Video + FiLM76.883.480.164.770.267.4

The two components are complementary. On ReasonVOS, combining them adds +7.0 𝒥&𝓕, compared with +2.6 on Ref-DAVIS. This suggests reasoning expressions rely more on temporal cues.

Adapter architecture

Linear48.3
Conv361.3
MLP61.9
U-Net64.9
U-Net + FiLM67.4

ReasonVOS 𝒥&𝓕 (axis 45–70)

Table 7. Adapter architecture ablation.
AdapterRef-DAVISReasonVOS
𝒥𝓕𝒥&𝓕𝒥𝓕𝒥&𝓕
Linear53.661.557.545.351.348.3
MLP70.176.273.259.464.561.9
Conv371.277.374.358.663.961.3
U-Net76.382.079.162.267.764.9
U-Net + FiLM76.883.480.164.770.267.4

A linear probe falls far short. The MLP and single-scale conv head come within 4–5 points of the U-Net, which suggests the U-Net's gains come mainly from multi-scale aggregation.

Tracklet selection criterion

Random30.8
SAM2 score41.1
Mean prob.50.9
Captured mass51.6
Soft IoU62.6
Cosine sim.63.6
Volumetric Pearson64.2
Hard IoU66.1
Per-frame Pearson67.4
Oracle82.6

ReasonVOS 𝒥&𝓕 (axis 25–85). Ref-DAVIS: per-frame Pearson 80.1, oracle 86.4.

Per-frame Pearson correlation is the best practical criterion on both benchmarks. Its scale and offset invariance suits uncalibrated attention maps. The oracle gap shows there is room for better selection.

Number of keyframes

Table 6. Effect of the number of keyframes on ReasonVOS.
# Keyframes4812162024
𝒥59.461.663.064.763.564.2
𝓕64.566.868.970.270.170.4
𝒥&𝓕62.064.265.967.466.867.3
4812162024 67.462.0

𝒥&𝓕 rises from 62.0 with 4 keyframes to a peak of 67.4 at 16, then saturates. 16 keyframes is the default.

Citation

@inproceedings{dastmalchi2027atlas,
  title     = {ATLAS: Learning to Adapt LVLM Attention for Reasoning Video Segmentation},
  author    = {Dastmalchi, Hamidreza and Cheraghian, Ali and An, Aijun and Khamis, Abdelwahed and Petersson, Lars},
  booktitle = {Proceedings of the IEEE/CVF Winter Conference on Applications of Computer Vision (WACV)},
  year      = {2027}
}