2 个月前

MobileVOS:实时视频对象分割——对比学习与知识蒸馏的结合

Roy Miles; Mehmet Kerim Yucel; Bruno Manganelli; Albert Saa-Garriga
MobileVOS:实时视频对象分割——对比学习与知识蒸馏的结合
摘要

本文解决了在资源受限设备(如手机)上进行半监督视频对象分割的问题。我们将这一问题表述为一种知识蒸馏任务,通过实验证明,具有有限内存的小型时空网络可以以较低的计算成本(三星Galaxy S22每帧32毫秒)实现与最先进方法相媲美的结果。具体而言,我们提供了一个理论基础框架,将知识蒸馏与有监督对比表示学习统一起来。这些模型能够同时从像素级对比学习和预训练教师模型的知识蒸馏中获益。我们在标准DAVIS和YouTube基准测试中验证了该损失函数的有效性,尽管运行速度提高了多达5倍,并且参数量减少了32倍,但仍能取得与最先进方法相当的J&F分数。

MobileVOS:实时视频对象分割——对比学习与知识蒸馏的结合 | 最新论文 | HyperAI超神经