跳到正文
OpenReview · State space models· Md Mohaiminul Islam; Mahmudul Hasan; Kishan Shamsundar Athrey; Tony Braskich; Gedas Bertasius·· 2022-01-01精选AI 评分77

使用状态空间 Transformer 的高效电影场景检测

Efficient Movie Scene Detection using State-Space Transformers

AI 导读

基于摘要分析。本文研究电影场景检测:场景理解需要跨越较长电影片段整合线索,而多数视频识别模型侧重短时分析。作者提出可端到端训练的 State-Space Transformer 模型 TranS4mer,通过结合局部自注意力与长程状态空间运算,建模电影中的短程和长程依赖。 核心模块 S4A 结合 structured state-space sequence(S4)与自注意力(A)。输入是已划分为电影镜头的帧序列;模块先用自注意力捕获镜头内短程依赖,再用状态空间运算聚合镜头间长程线索。TranS4mer 由多个 S4A 模块堆叠构成,具体堆叠数量、特征提取方式、损失函数及训练配置尚未验证。该设计的关键在于利用镜头这一层级结构,分别处理局部关系与跨镜头上下文。 作者报告,在 MovieNet、BBC 和 OVSD 三个电影场景检测数据集上,TranS4mer 优于此前方法;相较标准 Transformer,速度为其 2 倍,GPU 显存需求约为其三分之一。摘要未给出检测指标、划分协议、具体对照配置及效率测量条件,因此这些结果仅适用于作者所述电影场景检测评估,实验协议、消融及统计信息尚未验证。作者表示将发布代码和模型,材料不能确认其已公开。 平台推测(待验证):可检验的迁移假设是,将局部交互与长程状态聚合分离,可能缓解长时 EEG 建模的计算负担。可复用的是 S4A 的分工机制,但电影镜头结构不能直接等同于 EEG 时间窗;需要改造信号编码、时间分段与任务监督。摘要也不足以判断原方法的数据规模要求。低信噪比、跨被试差异、通道布局变化及小数据训练均可能使局部注意力学习到噪声,或使长程聚合混合不同生理状态。一个小规模可证伪实验是在同一 EEG 数据集、固定跨被试划分及相同输入长度和训练预算下,对比局部注意力加 S4、纯注意力和纯 S4,核对任务指标与速度、显存,并测试时间窗长度变化。已有 EEG 相关工作尚未检索确认。

阅读价值

值得阅读的是 TranS4mer 将镜头内自注意力与镜头间 S4 状态空间聚合分工结合的长序列建模机制;其效率优势由作者报告,对 EEG 的迁移价值尚未验证。

来源:OpenReview · State space models · openreview.net