alphaFeedback

WorksComputer Science

OmniGround: Strengthening Omni-Modal Temporal Grounding via Attention-Consistency Training

AI contributions
Not recorded for this paper
View details

Abstract

Omni-modal large language models can jointly understand video, audio, and text, yet they remain weak at temporal grounding, a capability essential for real-world audio-visual applications. To understand this limitation, we look beyond final predictions and examine the model's internal temporal evidence. We identify a temporal generation gap: the internal attention of the model localizes the target moment more accurately than the timestamps it generates. To close this gap, we propose TACG (Temporal Attention-Consistency Grounding), a post-training method that treats internal attention as a reliable temporal pivot and aligns it with both the model prediction and the ground-truth annotation through attention-consistency training. We further build a 580K-sample training dataset by reconstructing temporal QA from dense audio-visual annotations and organizing general omni-modal instruction data. Using TACG and this training dataset, we train OmniGround, a general-purpose omni-modal model with stronger temporal grounding capability, without injecting any additional temporal cues. OmniGround reaches state-of-the-art omni-modal temporal grounding, including 37.5 mIoU on LongVALE and 35.7 mIoU on UnAV-100, while staying competitive on both general audio-visual understanding and vision-only grounding.

Abstract

Omni-modal large language models can jointly understand video, audio, and text, yet they remain weak at temporal grounding, a capability essential for real-world audio-visual applications. To understand this limitation, we look beyond final predictions and examine the model’s internal temporal evidence. We identify a temporal generation gap: the internal attention of the model localizes the target moment more accurately than the timestamps it generates. To close this gap, we propose TACG (Temporal Attention-Consistency Grounding), a post-training method that treats internal attention as a reliable temporal pivot and aligns it with both the model prediction and the ground-truth annotation through attention-consistency training. We further build a 580K-sample training dataset by reconstructing temporal QA from dense audio-visual annotations and organizing general omni-modal instruction data. Using TACG and this training dataset, we train OmniGround, a general-purpose omni-modal model with stronger temporal grounding capability, without injecting any additional temporal cues. OmniGround reaches state-of-the-art omni-modal temporal grounding, including 37.5 mIoU on LongVALE and 35.7 mIoU on UnAV-100, while staying competitive on both general audio-visual understanding and vision-only grounding.

Introduction

Omni-modal large language models ([19] (Xu et al. 2025a; Wu et al. 2024)), which jointly perceive video, audio, and text within a single backbone, have advanced rapidly and now excel at a wide range of multimodal understanding tasks. As these models move toward real-world audio-visual applications such as video editing, content moderation, and multimodal retrieval, recognizing what happens no longer suffices: a model must also pinpoint when it happens. This temporal capability is critical for practical audio-visual applications, with temporal video grounding (TVG) serving as a central benchmark. Yet current omni-modal models still perform poorly on TVG despite their strong general understanding.

Existing temporal grounding methods mainly follow two directions. One family modifies the model or its inputs by injecting explicit temporal information, such as special temporal tokens, frame markers, or task-specific heads ([24] (Guo et al. 2025a; Wu et al. 2025)). Although effective for localization, these designs make temporal grounding depend on customized input formats and specialized output parsing, which limits their compatibility with a general-purpose omni-modal interface. A second family improves grounding purely through post-training, such as reinforcement learning ([30]) or distillation from a stronger teacher (Li et al. 2026a). This training-only direction preserves the inference interface and better maintains general ability, but does not directly exploit the model’s own temporal evidence. We adopt this philosophy, but push it further.

We begin with a diagnostic question: when an omni-modal model grounds imprecisely, does it fail to localize the event, or merely to express the time? For each query, we compare the interval read from the model’s internal attention with its generated timestamp, and measure which lands closer to the ground truth. Across multiple omni-modal models and grounding datasets, the attention-derived interval is more accurate than the prediction on a clear majority of samples, up to 78.3%. We call this the temporal generation gap: the model knows where to look but cannot accurately say when. This pattern holds for both base models and models with stronger grounding ability, indicating that the gap is not merely a weak-model artifact, as illustrated in Figure 1. If left unaddressed, this gap limits further improvements in temporal localization precision.

Overview of OmniGround

Figure 1. Overview of OmniGround. Left: both the base model and the stronger SFT model exhibit a temporal generation gap: internal attention localizes the target event more accurately than the generated timestamp. TACG closes it to achieve stronger grounding. Right: unlike methods that rely on additional temporal inputs or specialized output formats, OmniGround strengthens temporal grounding while preserving a general-purpose omni-modal interface.

Building on this diagnosis, we propose Temporal Attention-Consistency Grounding (TACG). Its core term aligns the generated timestamp with the model’s own internal attention, using the more reliable internal signal to pull the prediction onto the target and directly close the temporal generation gap. Two auxiliary terms anchor this correction to the ground truth: one supervises the attention directly against the annotation, making the attention itself localize the target more accurately; the other supervises the prediction by its temporal distance to the annotation, supplying the distance-aware gradient that cross-entropy lacks. TACG achieves stronger temporal grounding than standard supervised fine-tuning on the same data, without introducing additional temporal inputs or model parameters.

We also build a 580k-sample omni-modal training dataset that supports both temporal grounding and general understanding. Starting from the dense audio-visual annotations in AVID ([3]) and LongVALE ([7]), we extract high-quality temporal segments and reformulate them into grounding, dense captioning, and temporal question-answering tasks. We further incorporate general omni-modal instruction data from several sources [39, 19, 35] (Ma et al. 2025) to preserve broad audio-visual understanding. Using this dataset and TACG, we train OmniGround from Qwen2.5-Omni (Xu et al. 2025a). As shown in Figure 1, OmniGround improves temporal grounding without the extra temporal inputs or specialized output formats used by prior baselines. We evaluate OmniGround on seven benchmarks, including three omni-modal grounding benchmarks, two vision-only grounding benchmarks, and two general omni-modal understanding benchmarks. OmniGround sets a new state of the art on omni-modal temporal grounding while staying competitive on the other two categories. Our contributions are summarized as follows:

  • We identify the temporal generation gap in omni-modal language models and propose TACG, a post-training objective that aligns the model’s prediction with its internal attention, consistently improving over standard supervised fine-tuning under identical training data.

  • We build a 580k omni-modal temporal-awareness dataset by reformulating dense audio-visual annotations into temporal grounding and question-answering tasks, applying quality filtering, and balancing the resulting data with curated general data. We train OmniGround on this dataset.

  • OmniGround reaches state-of-the-art omni-modal temporal grounding and stays competitive on general omni-modal understanding and vision-only grounding. We will release the dataset, model, and training framework.

Omni-Modal Language Models

Recent work extends large language models beyond text to perceive multiple modalities. Early video LLMs such as VideoChat (Li et al. 2023), Video-ChatGPT [21], and VideoLLaMA [37] couple a visual encoder with an LLM, while PandaGPT (Su et al. 2023) and NExT-GPT (Wu et al. 2024) add audio. More recent omni-modal models further unify audio, vision, and text within a shared modeling framework, enabling general audio-visual-text understanding [19] (Xu et al. 2025a). While these models have achieved strong general audio-visual understanding, they are not explicitly optimized for precise temporal grounding. As a result, they often recognize the relevant event but fail to generate accurate temporal boundaries. Our work strengthens temporal grounding in a general-purpose omni-modal model.

Model Mechanism Analysis

Prior studies show that model behavior can be analyzed through internal signals [40, 23]. Inspired by this perspective, we inspect internal attention in omni-modal temporal grounding.

Temporal Video Grounding

Temporal video grounding (TVG) localizes the moment described by a natural language query, and existing methods differ in how they acquire this ability. One family injects explicit temporal machinery—regression heads in specialist architectures [38, 18, 22] (Jiang et al. 2023), or, in video LLMs, time-aware encoders [26], special temporal tokens [28, 2] (Guo et al. 2025a), frame-index markers (Wu et al. 2025), and localization heads [24, 13, 29, 36, 11] (Guo et al. 2025b; Li et al. 2026b; Qu et al. 2024). These components depart from the standard backbone, add inference cost, and often weaken general question answering. A second family leaves the architecture and inference untouched and strengthens temporal grounding through additional post-training stages, such as reinforcement learning with IoU rewards [30] or distillation from a stronger teacher (Li et al. 2026a). Our method instead strengthens fine-tuning itself, turning the model’s internal attention into a self-supervision signal.

Preliminary: Temporal Generation Gap

We validate the temporal generation gap across multiple omni-modal models on AVID and LongVALE, including both base models and models with stronger grounding ability. For each benchmark, we construct the diagnostic evaluation split by randomly sampling 500 examples from its official test split and use this fixed subset for all preliminary-analysis statistics. For each sample we obtain three temporal intervals and compare them against the ground truth by temporal IoU. The ground-truth interval IgtI_{\mathrm{gt}} and the generated interval IpredI_{\mathrm{pred}} are taken directly. We extract the attention interval IattI_{\mathrm{att}} from the attention of the last prefill token to the audio-visual tokens at a mid-to-late layer of the model. This attention is aggregated over spatial positions into a temporal distribution, from which we select the shortest contiguous window covering 50% of the cumulative mass. This gives three intervals for each case, IgtI_{\mathrm{gt}}, IpredI_{\mathrm{pred}}, and IattI_{\mathrm{att}}; we compare IoU(Iatt,IgtI_{\mathrm{att}}, I_{\mathrm{gt}}) with IoU(Ipred,IgtI_{\mathrm{pred}}, I_{\mathrm{gt}}) to determine whether the attention interval or the generated prediction is more accurate, and use the same intervals for visualization. For Qwen2.5-Omni-3B, we use layer 27; additional details, including layer choices for other models, layer sensitivity, cumulative-mass threshold sensitivity, and the SFT setup, are provided in the Appendix.

Figure 2 visualizes the internal attention concentrating on the target region while the generated interval is displaced, and Table 1 shows that IattI_{\mathrm{att}} overlaps the ground truth better than IpredI_{\mathrm{pred}} on most valid samples across five omni-modal models and two datasets, despite large differences in grounding mIoU. Together, these results establish that the temporal generation gap is consistent across models and datasets. After SFT, both the attention interval and the generated prediction become more accurate, yet the instance-level gap remains, suggesting that unresolved attention–prediction misalignment limits further improvements in temporal localization precision. This motivates directly aligning attention, prediction, and ground truth in the training objective introduced next.

Visualization of the temporal generation gap

Figure 2. (a) Qwen2.5-Omni-3B (base) (b) Qwen2.5-Omni-3B (SFT)

Figure 2: Visualization of the temporal generation gap. Top: the attention averaged over samples and aligned to the ground-truth event (green), overlaid with the averaged prediction. Bottom: the corresponding averaged attention and predicted intervals. In both base and SFT models, the attention localizes the event more accurately than the prediction.

ModelAVIDLongVALE
Attn. wins†mIoUAttn. wins†mIoU
Qwen2.5-Omni-3B73.9%7.478.3%6.3
Qwen2.5-Omni-3B (SFT)57.7%17.960.0%20.9
Qwen2.5-Omni-7B75.7%6.272.1%12.1
Qwen3-Omni65.0%14.169.7%10.6
OLA73.0%10.370.3%7.7

Table 1. Table 1: The temporal generation gap across models and datasets. For each benchmark, all statistics are computed on a fixed evaluation subset of 500 examples randomly sampled from its official test split. Attn. wins† reports the fraction of non-tied samples where IoU(Iatt,IgtI_{\mathrm{att}}, I_{\mathrm{gt}}) > IoU(Ipred,IgtI_{\mathrm{pred}}, I_{\mathrm{gt}}). mIoU is the grounding accuracy of the generated interval. This gap is consistent across models and datasets.

Method: OmniGround

We present OmniGround, an omni-modal temporal grounding model built on Qwen2.5-Omni. We train it with TACG on our 580k-sample omni-modal training dataset. The framework is illustrated in Figure 3.

Overall architecture and TACG training of OmniGround

Figure 3. Overall architecture and TACG training of OmniGround.

Temporal Attention-Consistency Grounding

The preliminary analysis motivates supervising the generated timestamp with the model’s internal attention while anchoring both signals to the ground truth. Although temporal IoU measures their agreement at evaluation time, it is unsuitable as a training signal: it is non-differentiable and least informative exactly when most needed. Once two intervals are disjoint, its gradient vanishes everywhere, giving no indication of which direction, or how far, to move the prediction—precisely the regime that dominates early training. We therefore recast attention, prediction, and ground truth as continuous temporal distributions over the TT time steps and enforce their mutual consistency with divergences that stay informative under non-overlap (Jensen–Shannon and Wasserstein-1, chosen per edge below).

Attention–prediction consistency: the core objective. The central constraint of TACG forces the generated timestamp to agree with the localization already present in the internal attention. We implement this constraint by expressing both quantities as temporal distributions over the TT time steps.

We first extract the implicit localization signal from the attention at a selected intermediate layer ll. We compute the attention from the last prefill token to all media tokens (both video and audio) and aggregate the mass that falls on each of the TT interleaved time steps:

patt[t]=∑i∈Ctexp⁡(qTki/d)∑jexp⁡(qTkj/d),(1)p_{\mathrm{att}}[t] = \sum_{i \in\mathcal{C}_{t}} \frac{\exp(q^{\mathsf{T}} k_i / \sqrt{d})}{\sum_j \exp(q^{\mathsf{T}} k_j / \sqrt{d})}, \tag*{(1)}

where qq is the query of the last token, kik_i ranges over all video and audio tokens, and Ct\mathcal{C}_t is the set of media tokens at time step tt. As the attention is normalized over all media tokens, pattp_{\mathrm{att}} is automatically a valid distribution over the TT steps, and gradients flow through the layer-ll query and key projections, allowing the consistency loss to refine the attention itself.

We then convert the predicted timestamp into a continuous temporal distribution. Since standard supervised fine-tuning never decodes the logits, we operate on them directly. At each digit position of the timestamp tokens we form the soft expectation over digit values, round it, apply a straight-through estimator so that gradients still reach the logits, and assemble the start and end times by place value:

d^i=∑d∈Dd⋅softmax⁡(logits⁡i[D])d,dˉi=⌊d^i⌉,\hat{d}_i = \sum_{d \in D} d \cdot\operatorname{softmax}(\operatorname{logits}_i[D])_d,\quad\bar{d}_i = \lfloor\hat{d}_i \rceil,
d~i=dˉi−sg⁡(d^i)+d^i,\tilde{d}_i = \bar{d}_i - \operatorname{sg}(\hat{d}_i) + \hat{d}_i,
t^s=∑id~is10pi,t^e=∑id~ie10pi,(2)\hat{t}_s = \sum_i \tilde{d}_i^s 10^{p_i},\quad\hat{t}_e = \sum_i \tilde{d}_i^e 10^{p_i}, \tag*{(2)}

where D={0,…,9}D = \{0,\ldots,9\} are the digit token indices, ⌊⋅⌉\lfloor\cdot\rceil rounds to the nearest integer, sg⁡(⋅)\operatorname{sg}(\cdot) is the stop-gradient (detach) operator, pip_i is the (signed) place value of digit ii, and d~is,d~ie\tilde{d}_i^s,\tilde{d}_i^e are the digits of the start and end timestamps. The second line is the straight-through estimator: the forward pass uses the rounded digit dˉi\bar{d}_i (the two soft terms cancel), while the backward pass routes gradients through the soft expectation d^i\hat{d}_i to the logits. We map the reconstructed interval [t^s,t^e][\hat{t}_s,\hat{t}_e], normalized by the video duration, to a continuous distribution ppredp_{\mathrm{pred}} through a Gaussian projection: a Gaussian centered at the interval center μ^t\hat{\mu}_t with standard deviation σ=0.15⋅(t^e−t^s)\sigma= 0.15 \cdot(\hat{t}_e - \hat{t}_s) proportional to the predicted width. This projection is a smoothing operator, it places the prediction in the same distributional space as the attention so the two can be compared directly, and gives the interval smooth support so that its divergence from other distributions varies continuously with the temporal offset rather than vanishing once they no longer overlap. With both distributions in place, the core objective is their Wasserstein-1 distance, Ltp=W1(patt,ppred)\mathcal{L}_{\mathrm{tp}} = W_1(p_{\mathrm{att}}, p_{\mathrm{pred}}). This attention–prediction edge transfers the reliable localization already present in the internal attention onto the generated timestamps, directly closing the temporal generation gap. The divergence on each edge (Jensen–Shannon or Wasserstein-1) is chosen by ablation (Appendix).

Ground-truth anchoring: two auxiliary constraints. The self-consistency edge alone does not tie either quantity to the true interval, and could in principle drive the two into mutual agreement away from it. Two auxiliary edges therefore anchor the transfer to the ground truth. We apply the same Gaussian projection, with identical parameters, to the ground-truth interval [ts∗,te∗][t_s^*,t_e^*], obtaining a distribution pgtp_{\mathrm{gt}} that is detached from the computation graph and serves as the fixed supervision target. The attention–ground-truth edge keeps the internal signal aligned with the true segment, so that it remains a trustworthy target for the core edge. The prediction–ground-truth edge supplies the distance-aware supervision that cross-entropy lacks:

Ltg=JS⁡(patt,pgt),Lpg=W1(ppred,pgt).(3)\mathcal{L}_{\mathrm{tg}} = \operatorname{JS}(p_{\mathrm{att}},p_{\mathrm{gt}}),\quad\mathcal{L}_{\mathrm{pg}} = W_1(p_{\mathrm{pred}},p_{\mathrm{gt}}). \tag*{(3)}

Full objective. The three edges form a triangle of consistency constraints. With edge weights λtp\lambda_{\mathrm{tp}}, λtg\lambda_{\mathrm{tg}}, and λpg\lambda_{\mathrm{pg}}, the full training objective is

Ltotal=LCE+λtpLtp+λtgLtg+λpgLpg,(4)\mathcal{L}_{\mathrm{total}} = \mathcal{L}_{\mathrm{CE}} + \lambda_{\mathrm{tp}}\mathcal{L}_{\mathrm{tp}} + \lambda_{\mathrm{tg}}\mathcal{L}_{\mathrm{tg}} + \lambda_{\mathrm{pg}}\mathcal{L}_{\mathrm{pg}}, \tag*{(4)}

Data Construction and Training Strategy

After Quality Filtering removes low-resolution videos and overly short clips (grounding videos under 30 s and general clips under 2 s), we assemble the general understanding data from four balanced sources: 121k LLaVA-Video [39] samples re-annotated by OLA [19] from speech and visual perspectives, 37k AVQA [35] samples for scene sound, 32k FortisAVQA (Ma et al. 2025) samples for music, and 200k original LLaVA-Video samples paired with audio for general video proficiency. For grounding data, we use only the training-split videos and dense temporal captions of AVID [3] and LongVALE [7]. We reformat the training captions into three complementary tasks—Temporal Video Grounding (event → time), Reverse Grounding (time → event), and Dense Video Captioning (video→timed events). We then apply Audio-Visual Relevance Filtering, using Gemini 3.1 Pro to remove QA pairs whose text does not require audio-visual evidence. A final Balanced Mixing step combines the two streams, yielding 190k grounding samples (126k TVG, 50k Reverse Grounding, and 14k DVC) and 390k general samples, for a 580k-sample dataset at a grounding-to-general ratio of roughly 1:2; the composition and pipeline are summarized in Figure 4 and detailed in the Appendix.

Dataset composition and data construction pipeline

Figure 4. Left: dataset composition. Right: four-stage construction pipeline and representative QA templates for the three grounding formats.

We apply TACG only to TVG samples because it needs an explicit predicted time interval to construct ppredp_{\mathrm{pred}}, which is available only when the target output is a start–end timestamp. All samples are trained jointly: TVG samples use the full objective Ltotal\mathcal{L}_{\mathrm{total}}, while Reverse Grounding, DVC, and general QA samples use cross-entropy LCE\mathcal{L}_{\mathrm{CE}} only. Implementation details are provided in the Appendix.

Experiments

Experimental Setup

All experiments are conducted on 32 NVIDIA A100 GPUs. Detailed configurations are provided in the Appendix.

Evaluation Benchmarks. We evaluate on seven benchmarks: AVID [3], LongVALE [7], and UnAV-100 [6] for omni-modal grounding, Charades-STA [5] and ActivityNet-Captions [1] for vision-only grounding, and WorldSense [12] and Daily-Omni [41] for general omni-modal understanding. For UnAV-100, we convert its temporal annotations into QA pairs for large-model evaluation, with details provided in the Appendix.

Baselines. We compare against 17 baselines in two groups: vision-only models, including VideoChat (Li et al. 2023), VTimeLLM [13], TimeChat [26], HawkEye [29], VTG-LLM (Guo et al. 2025a), Grounded-VideoLLM [28], TimeSuite [36], NumPro (Wu et al. 2025), ED-VTG [24], and Time-R1 [30]; and omni-modal models, including VideoLLaMA2 [4], OLA [19], Qwen2.5-Omni (Xu et al. 2025a), Qwen3-Omni (Xu et al. 2025b), LongVALE [7], ChronusOmni [2], and the closed-source Gemini 3.1 Pro [8], which we report for reference only.

Temporal Grounding Performance

Table 3 and Table 2 report temporal grounding results across the five grounding benchmarks. We highlight three observations.

Method[email protected][email protected][email protected]mIoU
VideoLLaMA213.37.02.311.5
OLA13.66.62.013.1
Qwen2.5-Omni (3B)20.711.85.515.2
Qwen2.5-Omni (7B)25.916.26.720.1
Qwen3-Omni26.517.58.021.8
LongVALE11.36.13.215.0
ChronusOmni47.333.820.734.1
Gemini 3.1 Pro*61.647.628.344.1
OmniGround(3B)†36.021.610.925.9
OmniGround(7B)†42.332.719.132.2
OmniGround(3B)40.124.013.328.9
OmniGround(7B)46.134.222.235.7

Table 2. Evaluation on UnAV-100. *Closed-source, for reference only. †Trained on the same data but with standard SFT (cross-entropy).

MethodOmni-modal Grounding: AVID-TVG [email protected][email protected][email protected]mIoULongVALE-TVG [email protected][email protected][email protected]mIoUVision-only Grounding: Charades-STA [email protected][email protected][email protected]mIoUActivityNet-Captions [email protected][email protected][email protected]mIoU
Vision-only Models
VideoChat7.65.22.55.22.20.90.43.032.88.60.025.923.512.66.017.4
VTimeLLM9.24.11.07.49.14.31.97.952.027.711.131.544.027.814.330.4
TimeChat9.86.93.07.87.23.41.66.447.831.614.631.630.216.98.221.8
HawkEye7.54.11.75.215.37.82.810.950.631.414.533.749.129.310.732.7
VTG-LLM9.06.42.95.96.22.81.04.254.040.022.038.111.36.83.08.0
Grounded-VideoLLM12.08.55.69.620.212.87.315.854.236.419.736.846.230.319.036.1
TimeSuite1.90.30.12.617.28.74.111.869.948.724.045.531.016.99.823.6
NumPro5.22.80.84.810.44.72.68.163.842.020.641.455.637.520.638.8
ED-VTG––––––––59.539.319.840.252.133.116.035.2
Time-R15.71.90.85.528.415.47.820.778.361.234.953.758.139.021.440.5
Omni-modal Models
VideoLLaMA25.11.60.93.33.31.50.02.419.411.55.012.913.07.12.810.2
OLA14.99.95.310.19.64.21.97.532.210.93.220.027.915.67.020.6
Qwen2.5-Omni (3B)11.57.02.98.48.64.62.06.147.430.914.432.227.515.78.519.9
Qwen2.5-Omni (7B)8.43.91.26.017.99.23.812.365.545.923.343.741.526.014.629.7
Qwen3-Omni18.212.35.214.414.47.73.410.435.423.410.523.337.623.912.626.9
LongVALE4.82.41.14.515.78.63.911.047.919.89.732.646.328.514.231.6
ChronusOmni42.333.721.931.249.732.517.634.572.247.020.846.056.035.520.139.4
Gemini 3.1 Pro⋆55.050.544.747.066.249.031.046.674.644.415.644.264.146.628.746.9
OmniGround(3B)†67.361.248.155.340.227.315.127.750.220.58.432.631.019.211.223.5
OmniGround(7B)†70.367.655.760.446.234.520.232.767.238.519.744.650.436.519.238.3
OmniGround(3B)68.362.250.556.941.428.616.128.660.531.215.138.831.719.611.123.7
OmniGround(7B)75.272.164.965.951.939.525.137.574.552.627.749.356.439.121.141.2

Table 3. Comparison with state-of-the-art methods on temporal video grounding benchmarks. OmniGround sets a new state of the art on the omni-modal temporal grounding; moreover, despite seeing no vision-only grounding data, it still essentially matches specialized methods on the vision-only Charades-STA and ActivityNet-Captions, reflecting strong generalization. Bold: best; underline: second best. ⋆Closed-source, for reference only. †Trained on the same data but with standard SFT (cross-entropy).

Omni-modal temporal grounding. As shown in Table 3, OmniGround achieves the best open-model performance on the audio-visual AVID and LongVALE benchmarks. The 7B model reaches 65.9 mIoU on AVID and 37.5 mIoU on LongVALE, while the 3B model already surpasses all open baselines on AVID. Since our grounding data is reconstructed from the training-split source videos of AVID and LongVALE, we further evaluate on UnAV-100 to isolate out-of-domain audio-visual grounding performance. As shown in Table 2, OmniGround reaches 35.7 mIoU with the 7B model and outperforms evaluated open baselines at the stricter [email protected] and [email protected] thresholds. Together, these results show that OmniGround sets a new open-model state of the art in omni-modal temporal grounding.

Vision-only temporal grounding. The benefit of TACG is not limited to the audio-visual setting. OmniGround never sees any vision-only grounding data during training, and Charades-STA and ActivityNet-Captions are outside the audio-visual grounding domains used for our data construction. Nevertheless, OmniGround remains highly competitive on these benchmarks, essentially matching specialized vision-only methods, trailing only the vision-only specialist Time-R1, and leading all other omni-modal models. This indicates that aligning predictions with internal attention strengthens temporal grounding in general, rather than only fitting the omni-modal training distribution.

TACG versus standard SFT. At full scale, we train otherwise identical 3B and 7B models with standard cross-entropy on the same 580k-sample dataset (OmniGround†); TACG improves grounding on nearly every benchmark at both scales. These full-scale margins are smaller than the clean gain measured in the controlled ablation (Table 5), and by design: the TACG objective activates only on the forward-TVG samples—about one-fifth of the 580k-sample mixture—so its effect is diluted at the full-model level. Even under this dilution, the improvement is consistent across the evaluated benchmarks, and on the never-seen vision-only Charades-STA it is pronounced at both scales (+6.2 mIoU at 3B), showing that aligning predictions with the internal attention instills a transferable localization skill rather than overfitting a particular distribution. Figure 5 makes this alignment visible on AVID: for both OmniGround scales, the predicted intervals become better aligned with attention and move closer to the ground-truth intervals, explaining the stronger localization performance and confirming that TACG closes the temporal generation gap it was designed to target.

Comparison of ground-truth, attention, and generated intervals for OmniGround-3B and OmniGround-7B

Figure 5. TACG closes the temporal generation gap. Average ground-truth, attention, and generated intervals for OmniGround. The persistent displacement observed in the preliminary analysis has been largely absorbed.

ConfigurationAVID [email protected]AVID mIoULongVALE [email protected]LongVALE mIoUΔ%
Baseline (CE only)0.35020.31530.14080.16040.00
Ltg\mathcal{L}_{\mathrm{tg}} only0.35430.32230.15630.1722+4.79
Lpg\mathcal{L}_{\mathrm{pg}} only0.36490.32820.14790.1645+3.32
Ltp\mathcal{L}_{\mathrm{tp}} only0.36710.33150.15140.1663+4.41
Ltg+Lpg\mathcal{L}_{\mathrm{tg}}+\mathcal{L}_{\mathrm{pg}}0.36710.33010.15080.1672+4.47
Full TACG0.38140.34080.15460.1759+8.87

Table 5. Component ablation of the TACG consistency loss. Δ% is the average relative improvement over the SFT baseline across both benchmarks. Bold: best; <u>underline</u>: second best. Full results are provided in the appendix.

General Understanding Performance

Table 4 shows that OmniGround preserves strong general audio-visual understanding while improving temporal grounding. The 7B model remains competitive on World-Sense and achieves the best open-model result on Daily-Omni, indicating that grounding-oriented training does not come at the cost of general comprehension. Relative to the same-scale cross-entropy baselines, TACG keeps World-Sense stable and further improves Daily-Omni, suggesting that sharper temporal alignment can also benefit tasks involving event order and audio-visual correspondence. By contrast, the grounding-specialized LongVALE performs poorly on these general benchmarks, highlighting the advantage of OmniGround’s balanced training.

MethodWorldSense AccDaily-Omni Acc
VideoLLaMA227.835.8
OLA34.054.3
Qwen2.5-Omni (3B)46.062.3
Qwen2.5-Omni (7B)46.162.7
Qwen3-Omni52.068.8
LongVALE9.924.4
ChronusOmni49.570.2
Gemini 3.1 Pro⋆63.181.0
OmniGround (3B)†48.965.4
OmniGround (7B)†48.368.7
OmniGround (3B)48.668.1
OmniGround (7B)50.271.9

Table 4. Evaluation on general audio-visual understanding benchmarks. ⋆Closed-source, for reference only. †Trained on the same data but with standard SFT.

Ablation Study

We conduct ablation experiments to analyze the key design choices in TACG. All ablations are trained on the same set of

10k TVG samples, with an SFT model trained on identical data as the baseline, so the comparison isolates the gains brought by TACG rather than data differences.

Component Ablation

We evaluate the contribution of each consistency edge in Eq. (4) (Ltg\mathcal{L}_{\mathrm{tg}}, Lpg\mathcal{L}_{\mathrm{pg}}, and Ltp\mathcal{L}_{\mathrm{tp}}). Each configuration is reported under its individually optimized weight, so the comparison reflects each edge’s best achievable effect.

As shown in Table 5, each single edge improves over the SFT baseline under its own optimal weight (+3.32% to +4.79%), indicating that all three consistency signals are useful on their own. Stacking the two ground-truth-anchored edges (Ltg+Lpg\mathcal{L}_{\mathrm{tg}}+\mathcal{L}_{\mathrm{pg}}) yields only +4.47%—essentially no gain over either edge alone. In sharp contrast, completing the triangle with the attention-to-prediction edge Ltp\mathcal{L}_{\mathrm{tp}} lifts the gain to +8.87%, nearly doubling the best two-edge result. This identifies attention–prediction consistency as the core component of TACG: the largest gain appears precisely when the objective addresses the temporal generation gap diagnosed in our preliminary analysis.

Discretization of the Predicted Distribution

Constructing the predicted distribution requires mapping the per-position digit logits to a digit value did_i, which is then assembled into a timestamp by place value. Two conventional choices both fail for timestamp regression: Argmax keeps only the most probable digit, so it cannot distinguish a nearly-correct distribution from a confidently-wrong one and its selected digit jumps discontinuously between steps, yielding unstable gradients; Soft uses the continuous expectation for smooth gradients, but the expectation of a discrete distribution rarely lands on an integer, leaving a loss floor that, amplified by place value, biases the gradient at high-order positions.

We therefore adopt Round: we first take the soft expectation and then round it to the nearest digit. This combines smooth gradients with discrete semantics, avoiding both Argmax’s instability and Soft’s vanishing-gradient floor. The metrics confirm these failure modes: both Argmax and Soft fall below the CE baseline at their best settings, while Round is the only choice that yields a stable improvement (Appendix Figure 10). The training dynamics tell the same story (Figure 6): Argmax shows a sharp mid-training spike from mode collapse, whereas Round decreases smoothly and monotonically. The per-λ\lambda sensitivity curves are reported in the appendix.

Training loss curves for Argmax and Round timestamp discretization

Figure 6. Training dynamics for timestamp discretization. Argmax spikes mid-training from mode collapse, whereas Round decreases smoothly and monotonically.

Conclusion

We identified the temporal generation gap in omni-modal language models, where internal attention localizes the target moment more accurately than the generated timestamps, and proposed Temporal Attention-Consistency Grounding (TACG), which couples attention, prediction, and ground truth into a triangle of consistency constraints that strengthens temporal localization with no added parameters, special tokens, or inference overhead. We further constructed a 580k-sample training dataset covering both grounding and general omni-modal understanding. Built on TACG and this dataset, OmniGround reaches state-of-the-art omni-modal temporal grounding while preserving strong general comprehension. More broadly, our results show that a model’s internal attention is a valuable and underexploited supervision signal for temporal precision; we will release the dataset, model, and training framework to support future work.

Technical Appendix

This appendix provides additional details omitted from the main text due to space constraints. It is organized into three parts: (i) supporting material for the preliminary analysis of the temporal generation gap (divergence definitions and per-layer attention analysis); (ii) model architecture and training details; and (iii) additional ablations and full experimental results.

Divergence Definitions

For two distributions pp, qq over the TT time steps, the Jensen–Shannon divergence and the Wasserstein-1 distance used in the consistency loss (Eq. (4)) are

JS⁡(p,q)=12KL⁡(p∥m)+12KL⁡(q∥m),m=12(p+q),\operatorname{JS}(p,q)=\frac{1}{2}\operatorname{KL}(p\mathbin\Vert m)+\frac{1}{2}\operatorname{KL}(q\mathbin\Vert m),\qquad m=\frac{1}{2}(p+q),
W1(p,q)=1T∑t=1T∣∑τ≤t(pτ−qτ)∣(5)\mathcal{W}_1(p,q)=\frac{1}{T}\sum_{t=1}^{T}\left|\sum_{\tau\leq t}(p_\tau-q_\tau)\right| \tag*{(5)}

where KL is the Kullback–Leibler divergence and the second expression is the ℓ1\ell_1 distance between the cumulative distributions, the closed form of the Wasserstein-1 distance in one dimension.

Per-Layer Attention Analysis

We obtain each layer’s audio-visual temporal attention distribution from the last prefill token. We take its attention weights normalized over all video and audio tokens, assign both modalities to their corresponding interleaved temporal grids, sum the attention mass within each grid, and average over heads, yielding the distribution pattp_{\mathrm{att}} over the TT media grids. To make distributions comparable across clips of different length and segment position, we re-align every distribution so that the ground-truth segment is centered at the origin and rescaled to unit width, then average within each layer.

For each benchmark, the diagnostic evaluation split consists of 500 examples randomly sampled from the official test split. All preliminary-analysis statistics in Table 1, including Attn. wins and prediction mIoU, are computed on these fixed benchmark-specific subsets; ties are excluded only from the denominator of Attn. wins. The layer-wise visualization and sensitivity analysis use the same sampling protocol.

Lead rate versus aggregate mIoU. These two statistics characterize different aspects of localization quality. Lead rate measures how frequently attention is closer to the ground truth than the generated prediction at the instance level, whereas aggregate mIoU is also sensitive to the magnitude of a smaller number of large wins. Consequently, attention can lead on more non-tied samples while having a similar or slightly lower aggregate mIoU, as observed for the CE baseline. Our claim is therefore not that attention is uniformly superior on every sample, but that it provides complementary temporal evidence on a substantial fraction of instances. TACG exploits this evidence while anchoring both attention and prediction to the ground truth.

We visualize all 36 transformer layers to examine how the audio-visual temporal-localization signal evolves with model depth. As Figure 7 shows, layers 27–28 exhibit both sharp attention concentration inside the ground-truth segment and substantial sharpening from supervised fine-tuning, whereas earlier layers (e.g. layer 21) are less discriminative and the final layer 35 is diffuse. Figure 8 corroborates this across all 36 layers at once: in both the base and SFT models the attention mass concentrates inside the ground-truth band at essentially every layer, showing that the localization signal is broad rather than confined to a single layer. Throughout our experiments, we use layer 27 of Qwen2.5-Omni-3B (36 layers), layer 22 of Qwen2.5-Omni-7B, layer 32 of the Qwen3-Omni thinker (48 layers), and layer 20 of OLA (28 layers). For TACG training, the fixed attention layer is layer 27 for the 3B model and layer 22 for the 7B model, as reported in Table 8. The absolute index differs across models only because their depths do, not the underlying phenomenon.

GT-aligned attention at representative layers for the SFT and base models

Figure 7. GT-aligned attention at representative layers for the SFT and base models (N=500N=500); the green band marks the ground-truth segment.

GT-aligned temporal attention across all 36 layers for the SFT model, base model, and their difference

Figure 8. GT-aligned temporal attention across all 36 layers (N=500N=500), for the SFT model (left), the base model (middle), and their difference (right). Each row is a layer; the horizontal axis is the offset from the ground-truth center (GT width normalized to 1; green dashed lines mark the segment). Attention concentrates inside the GT band at nearly every layer and is sharpest around layers 26–28, and the SFT−base difference peaks inside the band at the same layers—fine-tuning sharpens these mid-to-late layers the most.

HyperparameterValue
Base modelQwen2.5-Omni-3B / 7B
OptimizerAdamW
Learning rate2.5e-5
LR schedulerCosine
Warmup ratio0.05
Batch size (per device)1
Gradient accumulation2
Number of epochs1
Max sequence length16,384
DeepSpeedZeRO-2
Gradient checkpointingEnabled
Attention implementationFlashAttention
PrecisionBF16
Frozen modulesViT, audio aligner
TACG Configuration
λtp\lambda_{\mathrm{tp}} (attn–pred, W1W_1)0.0500
λtg\lambda_{\mathrm{tg}} (attn–GT, JS)0.0125
λpg\lambda_{\mathrm{pg}} (pred–GT, W1W_1)0.1500
Attention layer ll27 (3B) / 22 (7B)
Gaussian σ\sigma0.15 ×\times segment width

Table 8. Training hyperparameters and TACG configuration.

We further test whether the temporal generation gap depends on the attention-extraction hyperparameters. Figure 9 varies the attention layer and the cumulative-mass cutoff used to convert attention into an interval. Across both sweeps, attention still overlaps the ground truth better than the generated interval on most non-tie samples, showing that the gap is not an artifact of a single layer or the default 50% cutoff.

Sensitivity plots for layer sweep and cumulative-mass cutoff sweep

Figure 9. Sensitivity of the temporal generation gap to attention extraction choices on Qwen2.5-Omni-3B (N=500N=500). Left: layer sweep. Right: cumulative-mass cutoff sweep for extracting IattI_{\mathrm{att}}.

VariantGroupAttn. mIoUPred. mIoUΔ\DeltaIoU
CE BaseAttn. wins40.05.9+34.1
Pred. wins37.968.5−30.6
Ties0.00.00.0
Overall29.831.1−1.3
Full TACGAttn. wins50.08.0+42.0
Pred. wins41.174.0−32.9
Ties0.00.00.0
Overall34.634.1+0.5

Table 6. Attention–prediction alignment of the CE baseline and Full TACG on AVID. All values are reported on the percentage scale. Full TACG improves both localization signals and reduces the aggregate gap between attention and prediction.

Architecture Details

The vision encoder processes input frames at native resolution. Each frame is divided into 14 × 14 pixel patches. A temporal patch size of 2 groups two consecutive frames into a single temporal unit, reducing the temporal token count by half. After ViT encoding, a spatial merge with factor 2 further compresses spatial tokens by merging 2 × 2 adjacent spatial patches, resulting in a 4× reduction in spatial token count.

Audio Encoder. Raw audio waveforms are resampled to 16 kHz and converted to 128-bin mel-spectrograms. The Whisper-style encoder uses 32 transformer layers with 20 attention heads and a hidden dimension of 1280. A linear projection maps the output to 2048 dimensions, matching the language model hidden size. Audio longer than 30 seconds is segmented and encoded in parallel, with outputs concatenated along the sequence dimension.

Chunk-Interleaved Input Construction. For a video of duration DD seconds:

  • If D≤60D \le60s: the video is encoded with ⌊D/2⌋\lfloor D/2 \rfloor temporal grids, each covering exactly 2 seconds. Audio tokens are split into chunks of fixed size matching the 2-second interval. Video and audio chunks are interleaved in chronological order.

  • If D>60D > 60s: video frames are uniformly sampled and encoded into at most 30 temporal grids. Each grid covers D/30D/30 seconds. Audio tokens are uniformly redistributed into 30 chunks of equal size, ensuring each audio chunk aligns temporally with its corresponding video grid.

The interleaved sequence is wrapped with special audio start and end tokens. Within each chunk, visual tokens appear first, followed by audio tokens. The 3D M-RoPE assigns temporal, height, and width position indices to visual tokens, while audio tokens receive sequential position indices synchronized with the temporal position of their chunk.

Training Data Details

Quality Filtering. Before reformatting, we remove low-resolution videos and overly short clips: grounding videos shorter than 30 s and general-understanding clips shorter than 2 s are discarded. General-understanding data is then sampled from four sources so that no single audio type dominates: OLA-reannotated LLaVA-Video for paired speech and visual reasoning, original LLaVA-Video for general video proficiency with audio, AVQA for scene sound, and FortisAVQA for music.

QA Construction. AVID and LongVALE provide source videos with dense caption-style temporal annotations rather than the final QA pairs used for our training. We use only their training-split source videos and temporal captions, then reconstruct new plain-text QA instances in three complementary grounding formats. We preserve the official benchmark separation: videos, temporal annotations, and captions from the held-out evaluation splits are excluded from all construction, filtering, and training stages. Forward TVG maps an event description to its interval—“Locate the event where {description}.” → “[start_sec, end_sec]”; Reverse Grounding inverts this, mapping a queried time interval to the event that occurs there; and Dense Video Captioning asks the model to enumerate all timed events in a clip—“[start, end]: {event}” lines in chronological order. This trains a single model to move in both directions between language and time rather than only emitting timestamps.

Audio-Visual Relevance Filtering and Mixing. After reformatting we apply an audio-visual relevance filter at the individual caption/query level, retaining targets that describe audio-visual events and removing text targets that are audio-only, visual-only, or ambiguous. For TVG and Reverse Grounding, the target is the queried event caption; for DVC, each timed event caption in the same QA instance is judged independently. The filter is text-only: Gemini 3.1 Pro receives only the event caption or QA text as input, labels the target as audio-visual, audio-only, visual-only, or ambiguous, and retains only audio-visual cases. Rejection rates are therefore computed over individual caption/query targets rather than whole videos. The filter is applied only to training-split samples and never sees held-out evaluation videos, annotations, or captions. A final balanced-mixing step combines the two streams into a 580,676-sample dataset at a grounding-to-general ratio of roughly 1:2, which we found preserves general audio-visual understanding while providing enough grounding supervision for the TACG objective to take effect.

Training Hyperparameters

We fine-tune Qwen2.5-Omni-3B and Qwen2.5-Omni-7B in a single stage with AdamW and a cosine schedule, freezing the vision encoder and audio aligner so that training adapts the language model and cross-modal fusion while preserving the pretrained perceptual front-ends. Both scales use the same optimization and TACG configurations except for the attention layer specified in Table 8. The TACG objective is applied selectively—all forward-TVG samples, which emit an explicit interval and thus admit a predicted distribution, use the full consistency loss, while all other samples use cross-entropy alone. The three edge weights (λtp,λtg,λpg\lambda_{\mathrm{tp}}, \lambda_{\mathrm{tg}}, \lambda_{\mathrm{pg}})

ComponentSpecification
Vision Encoder
ArchitectureVision Transformer (ViT)
Hidden size1280
Patch size14×1414 \times14
Temporal patch size2
Spatial merge size2
Input channels3
Audio Encoder
ArchitectureWhisper-style Transformer
Hidden size1280
Output dimension2048
Number of layers32
Attention heads20
Mel bins128
Language Model
ArchitectureTransformer (autoregressive)
Hidden size2048
Number of layers36
Attention heads16
Key-value heads2 (grouped-query attention)
Intermediate size11008
Vocabulary size151936
Position encoding3D M-RoPE (T, H, W)
Chunk Configuration
Seconds per chunk2
Max temporal grids30
Long video threshold60 seconds

Table 7. Detailed architecture specifications of OmniGround.

are the effective values used by the final models.

Computing Infrastructure and Software. The full computing environment is listed in Table 11. Main-text results use a fixed random seed (42); multi-seed mean±std for the key results is reported in the Multi-Seed Stability section (Tables 13 and 14). Code, training data, and model checkpoints will be released publicly upon publication.

Evaluation Protocol and Baseline Reporting

All grounding results are evaluated under the official protocol of each benchmark. For AVID and LongVALE, we use

Rejected typeVisual-onlyAudio-onlyTotal
Rate17%4%21%

Table 9. Text-only audio-visual relevance filtering statistics. Rejection is computed over individual caption/query targets rather than whole videos or QA instances.

TaskSourceSamples
Temporal Grounding Data (from AVID & LongVALE)
Event → Time (TVG)AVID34,041
Event → Time (TVG)LongVALE91,863
Time → Event (Reverse)AVID13,526
Time → Event (Reverse)LongVALE36,474
Dense Captioning (DVC)AVID7,239
Dense Captioning (DVC)LongVALE7,240
Subtotal190,383
General Understanding Data
General video QALLaVA-Video200,000
Speech + visual QALLaVA-Video (OLA)120,925
Scene-sound QAAVQA37,441
Music QAFortisAVQA31,927
Subtotal390,293
Total580,676

Table 10. Training data composition of our final model (580,676 samples).

Reverse Grounding is produced by inverting a 50,000-sample subset of the forward-TVG pool, so it inherits that pool’s AVID/LongVALE split. The SFT model probed in our preliminary analysis is trained only on the forward-TVG data (125,904 samples), without the Reverse Grounding, DVC, or general-understanding data.

the released test splits and compute [email protected], [email protected], [email protected], and mIoU against the annotated temporal intervals; for the constructed UnAV-100 TVG subset, we use the same metrics on the filtered single-interval test instances described below. For Charades-STA and ActivityNet-Captions, we follow the standard temporal grounding evaluation protocol used by prior video-LLM grounding work. General audio-visual understanding results on WorldSense and Daily-Omni are reported with the official accuracy metric.

UnAV-100 TVG Test Subset. We additionally construct a temporal video grounding test subset from the UnAV-100 [6] test annotations. The original test.parquet contains 5,900 audio-visual event annotations over 2,167 videos, where each annotation is a tuple of video_id, event label, and timestamp interval. We first group annotations by (video_id, event_label`). Among the 3,455 resulting video-event combinations, 1,309 correspond to multiple disjoint temporal intervals. Because our evaluation protocol expects one predicted interval per query, these multi-interval cases are ambiguous: a single model output cannot be matched to a unique ground-truth segment without an arbitrary choice. We therefore keep only video-event pairs associated with a single interval, yielding 2,146 QA instances over 1,610 videos.

We then filter the remaining instances by relative event duration. A large fraction of the single-interval cases nearly spans the whole video: 42.8% have ground-truth duration ratio above 0.9, and 53.7% are above 0.7. Such near-full-length events substantially inflate localization metrics, since a trivial predictor that outputs the whole video can already obtain about 0.62 mIoU over the full single-interval pool. To focus the benchmark on genuine temporal localization, we retain only short events whose ground-truth duration divided by video duration is below 0.3. The final UnAV-100 TVG subset contains 653 QA instances over 490 videos. We will release the corresponding JSON file for reproducible evaluation.

ItemSpecification
Hardware
GPUs32 × NVIDIA A100-80GB
ParallelismDeepSpeed ZeRO-2
NVIDIA driver470.161.03
Software
Python3.10.15
PyTorch2.8.0 (CUDA 12.8)
transformers4.57.0
ms-swift4.2.0
accelerate1.13.0
deepspeed0.17.2
peft0.15.0
flash-attn2.8.1
xformers0.0.32
triton3.4.0
datasets4.8.4
tokenizers0.22.1
torchaudio / torchvision2.8.0 / 0.23.0
vllm (inference/eval)0.10.2

Table 11. Computing infrastructure and software environment.

For our models and the re-evaluated open-source baselines, we use the same benchmark input format and parse only the final predicted temporal interval. Outputs in seconds are evaluated directly. Outputs in frame indices or discrete temporal slots, when produced by methods whose native protocol uses such representations, are converted back to seconds using the benchmark video duration or frame rate before metric computation, following the corresponding benchmark protocol. If a model produces multiple intervals, we evaluate the first complete interval unless the model’s official inference protocol specifies otherwise. Invalid or unparsable outputs, including missing start/end times or intervals outside the video duration after clipping, are counted as failed predictions with zero IoU. For start-end inversions, we reorder the two endpoints before clipping to the valid video range, matching common temporal grounding evaluation practice.

The numbers in Table 3 and Table 2 are reported in two ways. For baselines with official benchmark results under the same evaluation setting, we cite the published numbers. For open-source omni-modal models and baselines without directly comparable reported results on AVID, LongVALE, or UnAV-100 TVG, we directly evaluate the released checkpoints; we do not fine-tune these baselines on our reconstructed QA data. The re-evaluated baselines use the same prompt template, video preprocessing budget, timestamp parser, and metric script used for OmniGround. Closed-source Gemini 3.1 Pro is included only as a reference result and is not used for training or model selection. Omni-Ground and OmniGround† are the only models in Table 3 trained on our 580k mixture, whose grounding portion consists of newly reconstructed QA instances derived from the training-split source videos and temporal captions of AVID and LongVALE. Therefore, Table 3 and Table 2 should be read as benchmark comparisons under each method’s available training protocol; the controlled comparison isolating TACG from data effects is the OmniGround† row, which uses the same 580k mixture as OmniGround but replaces TACG with standard cross-entropy.

Timestamp Parsing for TACG

All forward-TVG targets follow a standardized template containing a single well-formed start–end timestamp that can be deterministically parsed into digit positions. During training, we locate the timestamp span from the ground-truth answer first complete interval unless the model’s official inference protocol specifies otherwise. Invalid or unparsable outputs, including missing start/end times or intervals outside the video duration after clipping, are counted as failed predictions with zero IoU. For start-end inversions, we reorder the two endpoints before clipping to the valid video range, matching common temporal grounding evaluation practice. The numbers in Table 3 and Table 2 are reported in two ways. For baselines with official benchmark results under the same evaluation setting, we cite the published numbers. For open-source omni-modal models and baselines without directly comparable reported results on AVID, LongVALE, or UnAV-100 TVG, we directly evaluate the released checkpoints; we do not fine-tune these baselines on our reconstructed QA data. The re-evaluated baselines use the same prompt template, video preprocessing budget, timestamp parser, and metric script used for OmniGround. Closed-source Gemini 3.1 Pro is included only as a reference result and is not used for training or model selection. Omni-Ground and OmniGround† are the only models in Table 3 trained on our 580k mixture, whose grounding portion consists of newly reconstructed QA instances derived from the training-split source videos and temporal captions of AVID and LongVALE. Therefore, Table 3 and Table 2 should be read as benchmark comparisons under each method’s available training protocol; the controlled comparison isolating TACG from data effects is the OmniGround† row, which uses the same 580k mixture as OmniGround but replaces TACG with standard cross-entropy. template and identify the digit-token positions corresponding to the start and end times. The consistency losses are then computed from the logits at these positions, as described in Eq. (2).

The reconstructed timestamps are converted to seconds and normalized by the video duration before Gaussian projection. We use the timestamp precision provided by the training template and keep the same precision across the start and end fields. The ground-truth distribution is constructed from the annotated interval with the same normalization and Gaussian projection as the prediction, but is detached from the computation graph.

Discretization Weight Sensitivity

This section reports the full sweep behind the discretization choice made in the main text. For each of the three ways of turning the per-position digit logits into a predicted timestamp—Argmax, Soft, and our Round (straight-through rounding)—we vary the loss weight λ\lambda on the predicted-distribution edge and measure mIoU on AVID and LongVALE. Figure 10 summarizes each method at its best λ\lambda against the cross-entropy (CE) baseline, and Figure 11 gives the complete mIoU-versus-λ\lambda curves.

Best mIoU per discretization against the cross-entropy baseline

Figure 10. Best mIoU per discretization against the cross-entropy baseline. Only Round yields a positive gain over the CE baseline; Argmax and Soft fall below it on both benchmarks.

mIoU vs. loss weight for each discretization

Figure 11. mIoU vs. loss weight λ\lambda for each discretization.

The two views agree. Round is the only discretization that stays above the CE baseline on both benchmarks and remains stable across the entire λ\lambda range, whereas Argmax and Soft dip below the baseline at their best settings and degrade quickly as λ\lambda grows. This matches the failure modes analyzed in the main text: Argmax’s discontinuous digit selection triggers mode-collapse instability, while Soft’s non-integer expectation leaves a loss floor that its gradient cannot cross. The robustness of Round across λ\lambda is what makes it usable without per-benchmark tuning.

Weight Ablation

We ablate the loss weight λ\lambda of each edge by sweeping across a range of values with both JS divergence and Wasserstein-1 (W1) as distance metrics.

As shown in Figure 12, all edges share the same fundamental trade-off: when λ\lambda is too small, the supervision signal is too weak to meaningfully guide learning; when too large, the distribution-matching gradient overwhelms the cross-entropy loss and degrades generation quality. The optimal weight and distance metric, however, differ across edges. The ground-truth-anchored edges admit stronger supervision: Lpg\mathcal{L}_{\mathrm{pg}} peaks with W1 at λpg=0.2\lambda_{\mathrm{pg}} = 0.2, whose distance-proportional gradients give a smoother landscape that tolerates large weights, whereas Ltg\mathcal{L}_{\mathrm{tg}} favors JS at λtg=0.1\lambda_{\mathrm{tg}} = 0.1 with a broad effective range—for noisy attention supervision, JS measures distributional overlap without sensitivity to the absolute magnitude of misalignment and is thus more stable. The self-supervised edge Ltp\mathcal{L}_{\mathrm{tp}} exhibits a narrower effective range, as it enforces consistency between model-internal signals without external supervision and excessively large weights cause the distributions to reinforce each other’s errors; W1 with λtp=0.1\lambda_{\mathrm{tp}} = 0.1 achieves the best balance. Beyond the per-edge sweeps, Figure 13 maps the joint Ltg+Lpg\mathcal{L}_{\mathrm{tg}}+\mathcal{L}_{\mathrm{pg}} response surface over a 2D weight grid, confirming that the two ground-truth-anchored edges produce a broad region of improvement over the SFT baseline rather than a brittle optimum.

Weight ablation for all three edges on both benchmarks

Figure 12. Weight ablation for all three edges (Ltg\mathcal{L}_{\mathrm{tg}}, Lpg\mathcal{L}_{\mathrm{pg}}, Ltp\mathcal{L}_{\mathrm{tp}}) on both benchmarks.

TACG TG+PG Weight Grid — Response Surface (Contour)

Figure 13. Joint mIoU response surface over the λtg\lambda_{\mathrm{tg}}–λpg\lambda_{\mathrm{pg}} grid (26 evaluated points; dashed contour = SFT baseline) on LongVALE (left) and AVID (right).

Full Component Ablation

Table 12 reports the complete component ablation, including both addition results (upright rows, with individually optimized weights via progressive search) and subtraction results (italic rows, which inherit weights from the full configuration); the main text (Table 5) shows only the addition rows.

ConfigurationAVID [email protected]AVID mIoULongVALE [email protected]LongVALE mIoUΔ%
Baseline (CE only)0.35020.31530.14080.16040.00
Ltg\mathcal{L}_{\mathrm{tg}} only0.35430.32230.1563<u>0.1722</u>+4.79
Lpg\mathcal{L}_{\mathrm{pg}} only0.36490.32820.14790.1645+3.32
Ltp\mathcal{L}_{\mathrm{tp}} only<u>0.3671</u><u>0.3315</u>0.15140.1663+4.41
Full w/o Lpg\mathcal{L}_{\mathrm{pg}}, Ltp\mathcal{L}_{\mathrm{tp}}0.35580.32290.13600.1563−0.08
Full w/o Ltg\mathcal{L}_{\mathrm{tg}}, Ltp\mathcal{L}_{\mathrm{tp}}0.33500.30560.12930.1488−5.16
Full w/o Ltg\mathcal{L}_{\mathrm{tg}}, Lpg\mathcal{L}_{\mathrm{pg}}0.35340.32210.13660.1560−0.29
Ltg+Lpg\mathcal{L}_{\mathrm{tg}}+\mathcal{L}_{\mathrm{pg}}0.36710.33010.15080.1672+4.47
Full w/o Ltp\mathcal{L}_{\mathrm{tp}}0.32710.29890.13540.1579−3.38
Full w/o Lpg\mathcal{L}_{\mathrm{pg}}0.34810.31500.14460.1597−0.27
Full w/o Ltg\mathcal{L}_{\mathrm{tg}}0.37030.33150.13590.1570+1.51
Full TACG0.38140.3408<u>0.1546</u>0.1759+8.87

Table 12. Full component ablation of the TACG consistency loss. Upright rows: addition results with individually optimized weights; italic rows: subtraction results inheriting the full-model weights. Δ%\Delta\%: average relative improvement over the SFT baseline across both benchmarks. Bold/underline: best/second best per column.

Two observations stand out. First, among all single-edge removals from the full model, dropping the self-supervised attention–prediction edge Ltp\mathcal{L}_{\mathrm{tp}} causes the steepest degradation (−3.38%-3.38\%), reinforcing the main-text conclusion that it is the key contributor rather than a redundant term. Second, the subtraction rows—which reuse the full model’s weights instead of re-optimizing for the smaller edge subset—frequently under-perform their addition counterparts, and can even fall below the baseline. This shows that the edge weights do not transfer across subsets, and is why we report both the addition and subtraction views rather than a single ablation direction.

Multi-Seed Stability

The results in the main text come from a single run with a fixed seed (42). To verify that the reported margins are not seed artifacts—particularly the small full-scale gaps and the closely spaced single-edge ablations—we repeat the key experiments over three seeds and report the mean and standard deviation. Table 13 covers the 3B final model on the four grounding benchmarks, and Table 14 repeats the component ablation of Table 5. Across both tables, the mean stays close to the single-seed estimate and the ordering of configurations is preserved, indicating that the reported gains are stable rather than seed-specific.

Benchmark (mIoU)OmniGround (3B)
AVID56.9±0.456.9 \pm0.4
LongVALE28.6±0.528.6 \pm0.5
Charades-STA38.8±0.638.8 \pm0.6
ActivityNet-Captions23.7±0.423.7 \pm0.4

Table 13. Multi-seed stability of the 3B final model (mean ±\pm std over three seeds).

ConfigurationAVID mIoULongVALE mIoU
Baseline (CE only)0.3147±0.00420.3147 \pm0.00420.1598±0.00310.1598 \pm0.0031
Ltg\mathcal{L}_{\mathrm{tg}} only0.3218±0.00380.3218 \pm0.00380.1715±0.00350.1715 \pm0.0035
Lpg\mathcal{L}_{\mathrm{pg}} only0.3275±0.00450.3275 \pm0.00450.1640±0.00290.1640 \pm0.0029
Ltp\mathcal{L}_{\mathrm{tp}} only0.3309±0.00400.3309 \pm0.00400.1658±0.00330.1658 \pm0.0033
Ltg+Lpg\mathcal{L}_{\mathrm{tg}} + \mathcal{L}_{\mathrm{pg}}0.3296±0.00360.3296 \pm0.00360.1667±0.00300.1667 \pm0.0030
Full TACG0.3401±0.00340.3401 \pm0.00340.1753±0.00280.1753 \pm0.0028

Table 14. Multi-seed stability of the component ablation (cf. Table 5), mean ±\pm std over three seeds.

References

  1. [1]Caba Heilbron, F.; Escorcia, V.; Ghanem, B.; and Carlos Niebles, J. 2015. ActivityNet: A Large-Scale Video Benchmark for Human Activity Understanding. In CVPR.
  2. [2]Chen, Y.; Wu, Y.; Guan, K.; Ren, Y.; Wang, Y.; Song, R.; and Ru, L. 2025. ChronosOmni: Improving Time Awareness of Omni Large Language Models. arXiv preprint arXiv:2512.09841.arxiv.org/abs/2512.09841
  3. [3]Chen, Z.; Wang, D.; Lin, H.; Luo, L.; Xu, K.; Guo, Y.; Zhu, H.; Sun, T.; and Jiang, X. 2026. AVID: A Benchmark for Omni-Modal Audio-Visual Inconsistency Understanding via Agent-Driven Construction. arXiv preprint arXiv:2604.13593.arxiv.org/abs/2604.13593
  4. [4]Cheng, Z.; Leng, S.; Zhang, H.; Xin, Y.; Li, X.; Chen, G.; Zhu, Y.; Zhang, W.; Luo, Z.; Zhao, D.; and Bing, L. 2024. VideoLLaMA 2: Advancing Spatial-Temporal Modeling and Audio Understanding in Video-LLMs. arXiv preprint arXiv:2406.07476.arxiv.org/abs/2406.07476
  5. [5]Gao, J.; Sun, C.; Yang, Z.; and Nevatia, R. 2017. TALL: Temporal Activity Localization via Language Query. In ICCV. Charades-STA benchmark.arxiv.org/abs/1705.02101
  6. [6]Geng, T.; Wang, T.; Duan, J.; Cong, R.; and Zheng, F. 2023. Dense-Localizing Audio-Visual Events in Untrimmed Videos: A Large-Scale Benchmark and Baseline. In Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition, 22942–22951.arxiv.org/abs/2303.12930
  7. [7]Geng, T.; Zhang, J.; Wang, Q.; Wang, T.; Duan, J.; and Zheng, F. 2025. LongVALE: Vision-Audio-Language-Event Benchmark Towards Time-Aware Omni-Modal Perception of Long Videos. In CVPR.arxiv.org/abs/2411.19772
  8. [8]Google DeepMind. 2026. Gemini 3.1 Pro Model Card. Model card.
  9. [11]Hannan, T.; Islam, M. M.; Gu, J.; Seidl, T.; and Bertasius, G. 2025. ReVisionLLM: Recursive Vision-Language Model for Temporal Grounding in Hour-Long Videos. In Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR), 19012–19022.arxiv.org/abs/2411.14901
  10. [12]Hong, J.; Yan, S.; Cai, J.; Jiang, X.; Hu, Y.; and Xie, W. 2026. WorldSense: Evaluating Real-world Omnimodal Understanding for Multimodal LLMs. In International Conference on Learning Representations (ICLR). ArXiv preprint arXiv:2502.04326.arxiv.org/abs/2502.04326
  11. [13]Huang, B.; Wang, X.; Chen, H.; Song, Z.; and Zhu, W. 2024. VTimeLLM: Empower LLM to Grasp Video Moments. In CVPR.arxiv.org/abs/2311.18445
  12. [18]Lin, K. Q.; Zhang, P.; Chen, J.; Pramanick, S.; Gao, D.; Wang, A. J.; Yan, R.; and Shou, M. Z. 2023. UniVTG: Towards Unified Video-Language Temporal Grounding. In Proceedings of the IEEE/CVF International Conference on Computer Vision (ICCV), 2794–2804.arxiv.org/abs/2307.16715
  13. [19]Liu, Z.; Dong, Y.; Wang, J.; Liu, Z.; Hu, W.; Lu, J.; and Rao, Y. 2025. Ola: Pushing the Frontiers of Omni-Modal Language Model. arXiv preprint arXiv:2502.04328.arxiv.org/abs/2502.04328
  14. [21]Maaz, M.; Rasheed, H.; Khan, S.; and Khan, F. 2024. Video-ChatGPT: Towards Detailed Video Understanding via Large Vision and Language Models. In Proceedings of the 62nd Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers), 12585–12602. Association for Computational Linguistics.arxiv.org/abs/2306.05424
  15. [22]Mu, F.; Mo, S.; and Li, Y. 2024. SnAG: Scalable and Accurate Video Grounding. In Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR), 18930–18940.arxiv.org/abs/2404.02257
  16. [23]Palit, V.; Pandey, R.; Arora, A.; and Liang, P. P. 2023. Towards Vision-Language Mechanistic Interpretability: A Causal Tracing Tool for BLIP. In Proceedings of the IEEE/CVF International Conference on Computer Vision (ICCV) Workshops, 2856–2861.arxiv.org/abs/2308.14179
  17. [24]Pramanick, S.; Mavroudi, E.; Song, Y.; Chellappa, R.; Torresani, L.; and Afouras, T. 2025. Enrich and Detect: Video Temporal Grounding with Multimodal LLMs. In ICCV.arxiv.org/abs/2510.17023
  18. [26]Ren, S.; Yao, L.; Li, S.; Sun, X.; and Hou, L. 2024. TimeChat: A Time-sensitive Multimodal Large Language Model for Long Video Understanding. In CVPR.arxiv.org/abs/2312.02051
  19. [28]Wang, H.; Xu, Z.; Cheng, Y.; Diao, S.; Zhou, Y.; Cao, Y.; Wang, Q.; Ge, W.; and Huang, L. 2025a. Grounded-VideoLLM: Sharpening Fine-grained Temporal Grounding in Video Large Language Models. In EMNLP Findings.arxiv.org/abs/2410.03290
  20. [29]Wang, Y.; Meng, X.; Liang, J.; Wang, Y.; Liu, Q.; and Zhao, D. 2024. HawkEye: Training Video-Text LLMs for Grounding Text in Videos. arXiv preprint arXiv:2403.10228.arxiv.org/abs/2403.10228
  21. [30]Wang, Y.; Wang, Z.; Xu, B.; Du, Y.; Lin, K.; Xiao, Z.; Yue, Z.; Ju, J.; Zhang, L.; Yang, D.; Fang, X.; He, Z.; Luo, Z.; Wang, W.; Lin, J.; Luan, J.; and Jin, Q. 2025b. Time-R1: Post-Training Large Vision Language Model for Temporal Video Grounding. In NeurIPS.arxiv.org/abs/2503.13377
  22. [35]Yang, P.; Wang, X.; Duan, X.; Chen, H.; Hou, R.; Jin, C.; and Zhu, W. 2022. AVQA: A Dataset for Audio-Visual Question Answering on Videos. In ACM MM.DOI
  23. [36]Zeng, X.; Li, K.; Wang, C.; Li, X.; Jiang, T.; Yan, Z.; Li, S.; Shi, Y.; Yue, Z.; Wang, Y.; Wang, Y.; Qiao, Y.; and Wang, L. 2025. TimeSuite: Improving MLLMs for Long Video Understanding via Grounded Tuning. In ICLR.arxiv.org/abs/2410.19702
  24. [37]Zhang, H.; Li, X.; Bing, Z., Lidong; Leng, S.; Zhang, H.; Xin, Y.; and Li. 2023. Video-LLaMA: An Instruction-tuned Audio-Visual Language Model for Video Understanding. In Proceedings of the 2023 Conference on Empirical Methods in Natural Language Processing: System Demonstrations, 543–553. Association for Computational Linguistics.arxiv.org/abs/2306.02858
  25. [38]Zhang, S.; Peng, H.; Fu, J.; and Luo, J. 2020. Learning 2D Temporal Adjacent Networks for Moment Localization with Natural Language. In AAAI.arxiv.org/abs/1912.03590
  26. [39]Zhang, Y.; Wu, J.; Li, W.; Li, B.; Ma, Z.; Liu, Z.; and Li, C. 2024. Video Instruction Tuning with Synthetic Data. arXiv preprint arXiv:2410.02713.arxiv.org/abs/2410.02713
  27. [40]Zhao, Q.; Xu, M.; Gupta, K.; Asthana, A.; Zheng, L.; and Gould, S. 2024. The First to Know: How Token Distributions Reveal Hidden Knowledge in Large Vision-Language Models? arXiv preprint arXiv:2403.09037.arxiv.org/abs/2403.09037
  28. [41]Zhou, Z.; Wang, R.; Wu, Z.; and Jiang, Y.-G. 2025. Daily-Omni: Towards Audio-Visual Reasoning with Temporal Alignment across Modalities. arXiv preprint arXiv:2505.17862.arxiv.org/abs/2505.17862

Paper details

vlnc:W_tes6s8ws
v1 · latestHistory (1)

published · Fresh — added to the Hub within the last 30 days

added ·

Usage

views · 0

downloads · 0

A view counts once per browser per day, a download each time the PDF is requested.

CC BY 4.0

Original source

  • Computer Science

Not recorded for this paper · View statement →

Looking for related papers…
Contents