arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2001.05864cs.CVcs.LGcs.MM
收藏

基于分层强化学习的弱监督视频摘要

Weakly Supervised Video Summarization by Hierarchical Reinforcement Learning

The University of Tokyo(东京大学)

Yiyan Chen, Li Tao, Xueting Wang, Toshihiko Yamasaki

摘要:

传统的基于强化学习的视频摘要方法存在的问题是只有在生成完整摘要后才能收到奖励。这种奖励是稀疏的,它使得强化学习难以收敛。另一个问题是标记每个帧是繁琐和昂贵的,这通常会阻止大规模数据集的构建。为了解决这些问题,我们提出了一个弱监督的分层强化学习框架,它将整个任务分解成几个子任务,以提高摘要质量。该框架由管理者网络和工作者网络组成。对于每个子任务,管理者被训练成仅通过任务级二进制标签来设置子目标,这比传统方法需要少得多的标签。在子目标的指导下,工作者根据全局奖励和新定义的子奖励,通过策略梯度预测子任务中视频帧的重要性得分,克服了稀疏问题.在两个基准数据集上的实验表明,我们的建议取得了最好的性能,甚至优于监督方法。

英文摘要:

Conventional video summarization approaches based on reinforcement learning have the problem that the reward can only be received after the whole summary is generated. Such kind of reward is sparse and it makes reinforcement learning hard to converge. Another problem is that labelling each frame is tedious and costly, which usually prohibits the construction of large-scale datasets. To solve these problems, we propose a weakly supervised hierarchical reinforcement learning framework, which decomposes the whole task into several subtasks to enhance the summarization quality. This framework consists of a manager network and a worker network. For each subtask, the manager is trained to set a subgoal only by a task-level binary label, which requires much fewer labels than conventional approaches. With the guide of the subgoal, the worker predicts the importance scores for video frames in the subtask by policy gradient according to both global reward and innovative defined sub-rewards to overcome the sparse problem. Experiments on two benchmark datasets show that our proposal has achieved the best performance, even better than supervised approaches.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑