arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2606.14631cs.CV

SED: 基于蒸馏的轻量级事件数据显著性预测

SED:Lightweight Saliency prediction for Event-based data via Distillation

i3S/CNRS, Université Côte d’Azur(法国蔚蓝海岸大学i3S/CNRS实验室);ETH Zürich(苏黎世联邦理工学院)

Romaric Mazna, Jean Martinet, Michele Magno

AI总结:

提出轻量级网络SED,通过知识蒸馏和深度时空块(DSTconv)实现事件数据显著性预测,模型大小减少562倍,参数减少554倍,性能匹配或超越教师模型。

摘要:

基于事件的显著性预测最近受到关注,因为将事件相机与显著性估计结合可以作为上游阶段,自然提高边缘端下游事件感知的效率。然而,当前的方法要么是神经形态的,在基于事件的显著性基准上表现不佳,要么由于依赖Transformer或3D卷积而对资源受限的边缘应用来说过于沉重。受高效卷积模块的启发,SED旨在利用事件数据中的时间信息,我们提出了一种轻量级网络,通过知识蒸馏训练,构建于深度时空块(DSTconv)之上——这是3D深度可分离卷积的分解。相对于其教师模型,我们的模型将模型大小从180 MB减少到0.32 MB(562倍),参数数量从45M减少到81k(554倍),同时在N-DHF1K和N-UCF Sports数据集上匹配或超越其性能。此外,它在训练分布之外具有很强的泛化能力,从合成事件数据迁移到真实事件数据,而从头训练的模型则失败。

英文摘要:

Event-based saliency prediction has gained attention recently, as combining event cameras with saliency estimation can act as an upstream stage that naturally improves the efficiency of downstream eventbased perception at the edge. However, current approaches are either neuromorphic, underperforming on event-based saliency benchmarks, or too heavy for resource-constrained edge applications due to their reliance on transformers or 3D convolutions. Drawing inspiration from efficient convolutional modules, SED and aiming to exploit the temporal information in event data, we propose a lightweight network, trained through knowledge distillation, built on a Depthwise Spatio-Temporal Block (DSTconv) -- a factorization of the 3D depthwise separable convolution. Relative to its teacher, our model reduces the model size from 180 MB to 0.32 MB (562x) and the parameter count from 45M to 81k (554x), while matching or outperforming it on the N-DHF1K and N-UCF Sports datasets. Moreover, it generalizes strongly beyond its training distribution, transferring from synthetic to real event data where a model trained from scratch fails.

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

↑