用奖励函数训练视频模型,让其识别操作与说明的不符,泛化能力强。
Post-Training VLMs for Video Mistake Detection

- 通过定制奖励函数后训练视频语言模型,识别指令与视频间的差异。
- 在未见流程上性能提升最高达11.6%,显著优于基线方法。
- 适合需要通用错误检测的应用,如自动化质检、智能辅导系统。
人类在执行任务时常犯错,可能导致严重后果。当前的错误检测方法多基于封闭集协议,在受控环境下表现良好,但受限于需重新收集数据和训练新模型,难以推广。本文主张错误检测应学习“错误”的通用概念,而非依赖具体步骤细节。为此,提出新的误操作检测视频问答(MD-VQA)协议及基准测试,评估模型对已见与未见动作的判断能力。我们首次提出面向视频-语言模型的后训练技术用于错误检测,采用定制奖励函数引导模型识别指令与视频间的不一致。大量实验表明,该方法在零样本、监督微调和后训练基线上均表现更优,尤其在未见流程上提升显著,例如在EP-VQA上比最优基线最高提升11.6%,为通用错误检测开辟道路。代码与数据集已开源。
原文摘要 · Abstract (English)
Human mistakes are inevitable when following instructions, yet they can lead to severe consequences. As such, there has been an increased interest in developing methods for detecting mistakes in videos, with current methods mostly focusing on closed-set protocols. While successful in controlled settings, the closed-set assumption limits their wider applicability, as any changes to the task require collecting new data and re-training models. Instead, we argue that mistake detection methods should learn the general concept of a mistake, rather than overfitting to step-specific details. To reflect this, we introduce the Mistake Detection Video Question Answering (MD-VQA) protocol and accompanying benchmark. MD-VQA tests whether methods can discern if a step was executed correctly with respect to its description, for both seen and unseen actions. To address this important challenge, we propose the first video-language-model post-training technique for mistake detection. Our method uses a tailored reward function to encourage the model to identify discrepancies between an instruction and the corresponding video. Extensive evaluations demonstrate that this approach outperforms zero-shot, supervised fine-tuning, and post-training baselines. Notably, our method generalizes especially well to unseen procedures, for instance, with an improvement of up to 11.6% over the best-performing baseline on EP-VQA, paving the way toward general mistake detection. We release our code and benchmark at https://github.com/FedeSpu/mstk.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。