跳转至

Masked Diffusion Vision-Language Models for Temporal Action Localization

会议: NeurIPS 2026
arXiv: 2605.29858
代码: https://github.com/ycwfs/MDVLM-TAL
领域: 视频理解
关键词: 时序动作定位, 掩码扩散模型, 视觉语言模型, 双向去噪, 规划掩码策略

一句话总结

本文提出 MDVLM-TAL 框架,首次将时序动作定位转化为掩码扩散视觉语言序列生成任务,通过全局双向注意力打破自回归模型的左到右生成顺序,并配合规划训练目标与步级 tIoU 奖励实现语义描述与时序边界的协同迭代精细化。

研究背景与动机

时序动作定位(Temporal Action Localization, TAL)要求在未剪辑的长视频中准确识别动作类别并标定其实际发生的起止时间戳。近年来,基于大语言模型或视觉语言模型(LVLM)的生成式定位方案逐渐兴起,这类方法通过离散化时间戳将标签预测、边界回归与语言解释统一表述为序列生成任务,从而引入丰富的开放词表先验与推理能力。然而,现有生成式方案普遍依赖自回归(Autoregressive, AR)解码架构,必须严格遵循从左至右逐 token 生成的固定顺序,已生成的早期预测在后续步骤中完全不可变更。

这种自回归生成范式与时序动作定位任务的内在机理存在根本矛盾:在复杂的未剪辑视频中,动作起止边界的精准推断高度依赖于全局时序线索和后文呈现的完整动作语义。自回归模型在预测早期时间戳时,往往尚未接收到后文充分展开的动作演化信息;当后文语义逐步生成后,模型却受限于因果因果掩码而无法回过头去修正此前粗糙或产生偏差的时序边界。这种生成顺序的不可逆性,严重阻碍了语义理解上下文对时序定位边界的修正能力。

掩码扩散视觉语言模型(MDVLM)通过双向全局注意力进行迭代去噪生成,为突破这一瓶颈提供了全新契机。在双向去噪过程中,所有语义 token 与边界时间 token 均在生成全流程中保持可编辑状态。但直接套用通用掩码扩散模型存在两大不适应:一是训练掩码策略与去噪动力学的失配,通用扩散模型在空间与语义上执行均匀随机掩码,但时序定位中时间 token 必须依赖于已有语义上下文的建立才能稳定,过早解掩时间 token 违背了实际解码规律;二是常规交叉熵损失仅衡量离散 token 类别分类,无法反映连续时序区间重合度(tIoU)的数值距离关系。核心 idea:将 TAL 建模为掩码扩散序列生成,提出基于边界延迟解掩与步级加权的规划训练目标,并结合可导软时序交并比(Step-Level IoU Reward)协同迭代精细化动作语义与时间边界。

方法详解

整体框架

MDVLM-TAL 的整体流程包括多模态特征编码、帧级视觉 token 压缩、基于掩码扩散的双向迭代去噪生成三个主要阶段。对于给定的未剪辑视频和定位指令,系统首先利用视觉编码器与文本分词器提取多模态表征,并通过专门的时序压缩模块将每帧特征提炼为固定数量的紧凑视觉 token,与指令及特殊占位符拼接构成多模态上下文;语言骨干网络接收被噪声破坏的响应序列,在双向全局注意力下执行多步迭代去噪。

在训练阶段,模型采用针对时序动作特征定制的规划训练目标,使时间边界 token 比常规文本语义 token 保持更长时间的掩码状态,同时向低噪声重建步骤赋予更大权重;此外,模型引入步级软 IoU 奖励项,在起止边界均被激活时提供基于连续时间重叠的强化梯度,配合标准序列交叉熵损失共同优化模型参数。

%%{init: {'flowchart': {'rankSpacing': 24, 'nodeSpacing': 28, 'padding': 6, 'wrappingWidth': 400}}}%%
flowchart TD
    A["输入未剪辑视频与文本指令"] --> B["多模态编码与帧级 Token 压缩"]
    B --> C["构建多模态上下文与加噪响应序列"]
    C --> D["双向掩码扩散骨干网络迭代去噪"]
    D --> E["规划训练目标:边界感知延迟解掩与步级加权"]
    D --> F["步级 IoU 奖励:软边界期望与双边界门控"]
    E --> G["输出精细化时序动作边界与语义解析"]
    F --> G

关键设计

1. 规划训练目标:解决扩散掩码时序与边界稳定规律的动力学失配

常规掩码扩散模型采用完全均匀的随机独立掩码策略,然而在时序动作定位任务中,统计表明模型通常必须先构建出动作的语义线索(如类别和行为动作属性),时间边界 token 才能逐步稳定收敛。若在训练过程中让时间 token 与文本 token 具有相同的解掩概率,模型将频繁在缺乏语义上下文的极端噪声状态下强行预测时间,严重破坏去噪轨迹的稳定性。为此,本文设计了边界感知的差异化掩码策略:将响应序列中的 token 划分为时间边界集合 \(\mathcal{B}\) 与非时间集合 \(\mathcal{N}\),在扩散步长 \(t \in \{1, \dots, S\}\) 下赋予不同的保留概率:

\[\kappa_t^{\mathcal{B}} = \left(1 - \frac{t}{S}\right)^\gamma, \quad \kappa_t^{\mathcal{N}} = \left(1 - \frac{t}{S}\right)^\eta, \quad (\gamma > \eta > 0)\]

通过设置 \(\gamma > \eta\)(实验中取 \(\gamma=2, \eta=1\)),保证了在任意中间步数 \(t\) 下始终有 \(\kappa_t^{\mathcal{B}} < \kappa_t^{\mathcal{N}}\),即时间边界 token 比普通文本 token 经历更长周期的掩码遮蔽,从而强制模型在去噪过程中复现“先推断动作语义框架、后精准雕刻时间边界”的推理时自然解掩规律。同时,配套采用线性衰减的步级重构加权机制 \(w_t = \frac{2(S-t+1)}{S(S+1)}\),赋予接近真实分布的低噪声步(较小 \(t\))更大的梯度权重,强化高信噪比阶段对微小时序边缘差异的刻画能力。

2. 步级 IoU 奖励:基于软边界期望与双边界门控的连续时序重叠引导

离散 token 级别的交叉熵损失仅能判别当前预测的 token ID 是否与标签完全一致,对预测时间戳的数值距离毫无感知(例如当真实标签为 \(\langle t_{43} \rangle\) 时,预测为邻近的 \(\langle t_{47} \rangle\) 与严重偏离的 \(\langle t_0 \rangle\) 在常规交叉熵下遭受同等力度的惩罚),这与时序动作定位基于 tIoU 评价标准的连续几何特性产生脱节。本文提出可微的步级 IoU 奖励机制:首先从预测的起止位置 logits 中提取归一化的时间 token 概率分布 \(p_t^s(k)\) 和 \(p_t^e(k)\),通过加权期望计算出连续意义下的软起止时间边界 \(\bar{b}_{s,t} = \frac{\sum_{k=0}^{N-1} k p_t^s(k)}{N-1}\) 与 \(\bar{b}_{e,t} = \frac{\sum_{k=0}^{N-1} k p_t^e(k)}{N-1}\),并据此构建平滑可微的软重叠度 \(r_t^{\text{soft}} = \frac{\ell_t^\cap}{\ell_t^\cup + \epsilon}\)。

为了避免在仅解掩单一边界或边界尚未生成的初始去噪阶段引入剧烈梯度扰动,该机制引入了双边界门控指示因子 \(g_t = \mathbb{I}[\hat{y}_t^{(s)} \neq \texttt{[MASK]} \land \hat{y}_t^{(e)} \neq \texttt{[MASK]}}\),仅当起止双边界均解掩时激活奖励。进一步地,设计了统一的 IoU 损失函数 \(\mathcal{L}_{\text{iou}}\),其不仅直接最小化各激活步的未重叠惩罚 \((1 - r_t^{\text{soft}})\),而且包含时序单调提升折页损失 \(\max(0, r_t^{\text{soft}} - r_{t-1}^{\text{soft}} + \delta)\)(强制要求去噪轨迹沿反向时间轴朝向真实状态逐步提升 IoU),并利用相对于群体均值的标准化优势度 \(\tilde{A}_t\) 强化高增益去噪步的目标对数似然。

损失函数 / 训练策略

MDVLM-TAL 的总训练目标由标准序列重构交叉熵损失 \(\mathcal{L}_{\text{ce}}\)、规划训练目标损失 \(\mathcal{L}_{\text{plan}}\) 以及步级 IoU 奖励损失 \(\mathcal{L}_{\text{iou}}\) 联合加权构成:

\[\mathcal{L} = \lambda_{\text{ce}} \mathcal{L}_{\text{ce}} + \lambda_{\text{plan}} \mathcal{L}_{\text{plan}} + \lambda_{\text{iou}} \mathcal{L}_{\text{iou}}\]

默认超参数设置为 \((\lambda_{\text{plan}}, \lambda_{\text{iou}}, \lambda_{\text{ce}}) = (1.0, 0.5, 0.2)\)。视觉前端采用冻结权重的 SigLIP-SO400M-Patch14-384,语言骨干基于冻结的 LLaDA-8B-Instruct,仅在语言模型中插入 rank 16、\(\alpha=32\)、dropout 0.05 的 LoRA 适配层,并联合微调跨注意力多模态视觉压缩层。视频输入均匀采样 100 帧,时间戳划分为 \(N=100\) 个相对离散时间区间 \(\langle t_0 \rangle \sim \langle t_{99} \rangle\),去噪迭代总步数设定为 \(S=64\)。模型使用 AdamW 优化器,学习率 \(2 \times 10^{-4}\),在 4 张 NVIDIA L40S GPU 上配合余弦退火策略进行高效微调。

实验关键数据

主实验

在经典闭集时序动作定位基准 THUMOS-14、ActivityNet-1.3 以及推理型动作定位基准 ActivityNet-RTL 上全面评估模型性能。相比判别式 SOTA 模型(如 ActionFormer、TriDet、CLTDR)和自回归视觉语言基准(如 LITA、Qwen3.5-9B),MDVLM-TAL 取得了显著且全面的精度优势,尤其在高 tIoU 阈值下提升幅度更为突出。

数据集 评估指标 MDVLM-TAL (本文) 次优 SOTA 模型与配置 相对提升
THUMOS-14 [email protected] 74.4% 73.6% (CLTDR-GMG) / 71.2% (DiffTAD) +0.8% / +3.2%
THUMOS-14 [email protected] 60.1% 48.2% (CLTDR-GMG) / 58.5% (DiffTAD) +11.9% / +1.6%
THUMOS-14 Avg mAP [0.3:0.7] 72.7% 69.9% (CLTDR-GMG) / 68.0% (DiffTAD) +2.8% / +4.7%
ActivityNet-1.3 [email protected] 58.2% 55.0% (CLTDR-GMG) / 56.1% (DiffTAD) +3.2% / +2.1%
ActivityNet-1.3 [email protected] 38.8% 38.0% (CLTDR-GMG) / 36.9% (DiffTAD) +0.8% / +1.9%
ActivityNet-1.3 [email protected] 9.8% 8.6% (CLTDR-GMG) / 9.0% (DiffTAD) +1.2% / +0.8%
ActivityNet-1.3 Avg mAP [0.5:0.95] 37.9% 37.1% (CLTDR-GMG) / 36.1% (DiffTAD) +0.8% / +1.8%
ActivityNet-RTL mIoU 31.24% 30.09% (Qwen3.5-9B AR) / 28.6% (LITA-13B) +1.15% / +2.64%
ActivityNet-RTL [email protected] 27.12% 26.59% (Qwen3.5-9B AR) / 25.9% (LITA-13B) +0.53% / +1.22%
ActivityNet-RTL Score (推理质量) 48.9 48.3 (Qwen3.5-9B AR) / 46.3 (LITA-13B) +0.6 / +2.6

消融实验

在相同的 LLaDA-8B 掩码扩散骨干、SigLIP 视觉前端及训练数据下开展系统消融,验证规划训练目标(PT)、步级 IoU 奖励及各子机制的独立贡献:

配置 THUMOS-14 Avg mAP ActivityNet-1.3 Avg mAP ActivityNet-RTL mIoU 说明
MDVLM-TAL (完整模型) 72.7% 37.9% 31.24% 完整模型配置
w/o 规划训练目标 (Planned Training) 71.0% (-1.7%) 36.9% (-1.0%) 28.21% (-3.03%) 退化为普通随机均匀掩码与无步级加权
w/o 边界感知掩码 (Planned Masking) 71.7% (-1.0%) 37.2% (-0.7%) 29.35% (-1.89%) 移除边界时间延迟解掩,时间与文本同分布掩码
w/o 步级加权 (Step Weighting) 72.1% (-0.6%) 37.3% (-0.6%) 30.31% (-0.93%) 移除对低噪声 steps 的加权提升
w/ 反向规划 (Reversed Plan) 70.5% (-2.2%) 36.7% (-1.2%) 27.86% (-3.38%) 强制时间 token 先于语义 token 解掩,破坏生成动力学
w/o 步级 IoU 奖励 (IoU Reward) 71.5% (-1.2%) 37.2% (-0.7%) 28.74% (-2.50%) 仅依靠离散交叉熵作为监督信号
w/ 每步强制奖励 (Reward at Every Step) 72.0% (-0.7%) 37.3% (-0.6%) 30.12% (-1.12%) 移除双边界解掩门控,未解掩完全时强行计算奖励引入噪声
仅边界交叉熵 (Boundary-only CE) 71.8% (-0.9%) 37.3% (-0.6%) 29.68% (-1.56%) 丢弃非时间 token 的交叉熵,导致语言推理能力退化
全参数微调 (Full fine-tuning) 72.2% (-0.5%) 37.4% (-0.5%) 30.58% (-0.66%) 语言模型全量更新,由于过拟合泛化表现不及 LoRA
基础 MDVLM (Base MDVLM, w/o both) 69.9% (-2.8%) 36.3% (-1.6%) 26.94% (-4.30%) 同时移除规划目标和 IoU 奖励的纯扩散基线

关键发现

  • 规划训练是扩散定位的核心基石:完全去除规划训练目标会导致 ActivityNet-RTL 上的 mIoU 剧烈下滑 3.03 个百分点;若恶意将掩码顺序颠倒(反向规划,先解掩时间再解掩文本),性能甚至低于无规划基线(mIoU 仅 27.86%),从反面强有力地证明了“先宏观语义理解、后微观时序定界”是视觉语言扩散生成的必然内在机理。
  • 高重叠阈值下的决定性飞跃:在 THUMOS-14 上,MDVLM-TAL 相比以往最顶尖的判别式检测器 CLTDR-GMG,在 [email protected] 上取得了从 48.2% 到 60.1% 的飞跃式增长(+11.9%),这直接证明了全局双向可编辑性彻底摆脱了自回归累积误差与判别式提议回归在精确边缘上的界限模糊问题。
  • 双边界门控的去噪滤波作用:若在每个扩散步无差别计算软 IoU 奖励(移除门控 \(g_t\)),RTL mIoU 会显著下降 1.12 个百分点,表明在单侧边界缺失或未生成有效时间槽位时,盲目计算空间重叠会向下游反向传播有害的虚假梯度。

亮点与洞察

  • 打破自回归左至右的时序枷锁:将时序定位重塑为掩码扩散建模,赋予时序边界全局可编辑性。这从机制上解决了大模型定位领域长期存在的“前文定界即锁死、后文理解难修正”的核心症结。
  • 任务感知的非均匀扩散去噪动力学:作者敏锐捕获到时序动作中“动作语义早收敛、起止边界晚稳定”的数据集级动力学规律,通过两参数幂律控制(\(\gamma > \eta\))巧妙设计出贴合真实推理规律的延迟解掩训练目标,构成了离散扩散模型领域通用有效的范式参考。
  • 离散时间 token 的连续空间可微桥梁:通过在受限时序词表上以期望方式定义连续软边界,成功打通了离散 token 概率分布与连续时序重叠几何度量(tIoU)之间的梯度传导通道,使模型在生成离散 token 的同时兼具直接回归连续边界的重叠感知能力。

局限与展望

  • 迭代去噪步数带来的推理延迟开销:模型在推断阶段需要执行 \(S=64\) 步全局自注意力反向去噪,尽管单步计算量低于完整自回归的大上下文自回归自增量解码,但固定 64 步解码相比传统一次性前向传播的判别式检测器(如 ActionFormer、TriDet)在纯推理时延上仍有数倍至一个数量级的劣势。
  • 密集长视频时序离散化精度受限:时间戳被统一离散映射为 \(N=100\) 个分箱(bin),在长达数十分钟的高密度未剪辑视频中,单个时间 bin 对应的实际秒数跨度较大,理论上存在不可消除的量化截断误差。
  • 未来方向:探索与快速少步扩散采样器(如一致性模型 Consistency Models 或流匹配 Rectified Flow)的结合以压低推理步数,并探索自适应多分辨率时间尺度分箱策略以适应超长视频超精细定位。

相关工作与启发

  • vs LITA / VTimeLLM / VTG-LLM: 现有生成式视频时序定位方案均基于自回归因果语言解码器,时间边界生成后无法结合后文语义动态更新;MDVLM-TAL 基于双向全局注意力的掩码扩散生成,支持时间 token 与语义文本在全去噪轨迹中协同修正。
  • vs DiffTAD / DenoiseLoc: 这类连续扩散定位方法属于传统判别式架构的去噪扩展,将连续候选框或边界偏移量进行高斯加噪与去噪,无法处理自然语言查询且不支持生成动作解释文本;MDVLM-TAL 在离散文本与时间 token 统一的表征空间内运作,保留了多模态大模型的语言理解与开放词表推理优势。
  • vs ActionFormer / TriDet: 经典的基于金字塔特征的多尺度判别式时序定位网络在闭集基准上具备极高的单次前向效率,但在严苛的高 tIoU 阈值(如 tIoU=0.7)下边界回归易受特征混淆影响;MDVLM-TAL 借助大语言模型的丰富先验和迭代式全局去噪,在高精度边界判定上展现出大幅超越经典判别式模型的精准度。

评分

  • 新颖性: ⭐⭐⭐⭐⭐ 首次系统性将掩码扩散模型引入视觉语言时序动作定位,提出的规划训练与软 IoU 奖励机制兼具开创性与任务特异性。
  • 实验充分度: ⭐⭐⭐⭐⭐ 覆盖 THUMOS-14、ActivityNet-1.3 和 ActivityNet-RTL 三大主流权威基准,跨模型/跨组件消融与动力学统计异常完备。
  • 写作质量: ⭐⭐⭐⭐⭐ 逻辑严密,图文对照精确,符号定义与公式推导高度自洽清晰。
  • 价值: ⭐⭐⭐⭐⭐ 为解决长视频中多模态生成式时序感知与定位任务开辟了极具前景的去中心化/非自回归技术路径。