基于蒸馏通道剪枝的轻量级 Alpha 抠像网络
计算机视觉与模式识别
2022-10-17 v1
摘要
近年来，由于 alpha 抠像在自拍等移动应用中的实用性，其受到了广泛关注。因此，由于商用便携设备计算资源有限，对轻量级 alpha 抠像模型的需求日益增长。为此，我们提出了一种面向 alpha 抠像网络的基于蒸馏的通道剪枝方法。在剪枝步骤中，我们移除对学生网络模仿教师网络知识影响较小的通道。随后，被剪枝的轻量级学生网络使用相同的蒸馏损失进行训练。由所提方法得到的轻量级 alpha 抠像模型优于现有的轻量级方法。为展示我们算法的优越性，我们提供了多种定量与定性实验及深入分析。此外，我们将所提基于蒸馏的通道剪枝方法应用于语义分割，证明了其通用性。
引用
@article{arxiv.2210.07760,
title = {Lightweight Alpha Matting Network Using Distillation-Based Channel Pruning},
author = {Donggeun Yoon and Jinsun Park and Donghyeon Cho},
journal= {arXiv preprint arXiv:2210.07760},
year = {2022}
}
备注
Accepted by ACCV2022