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

基于多分支架构变换与分组卷积的低成本自集成

Low-Cost Self-Ensembles Based on Multi-Branch Transformation and Grouped Convolution

School of Integrated Technology, Yonsei University(延世大学融合技术学院)

Hojung Lee, Jong-Seok Lee

AI总结:

本文提出一种将 CNN 无额外开销地变换为多分支结构、结合差异化分组卷积与自蒸馏训练的低成本自集成方法,在保持单模型复杂度的同时提升了分类准确率与不确定性估计性能。

摘要:

低成本集成学习的最新进展已证明其在图像分类任务上的效率提升。然而,与传统集成学习相比,现有的低成本集成方法准确率相对较低。本文提出一种新型低成本集成学习方法,可同时实现高效率与优异分类性能。该方法将 CNN 变换为多分支结构,且不引入任何额外组件,从而保持与原始单一模型相同的计算复杂度;同时,通过让各分支的不同通路充分分离,增强分支输出之间的多样性。此外,本文提出一种新策略,在各分支中采用分组卷积,并让不同分支使用不同的分组数量,以进一步提升分支输出的多样性。训练阶段,采用知识蒸馏方法,将各分支输出的集成作为教师信号。输出之间的高度多样性能够形成强大的教师,从而提升每个独立分支的分类性能,并最终提升整体集成性能。实验结果表明,与以往低成本集成方法相比,本文方法取得了最先进的分类准确率以及更优的不确定性估计性能。代码可在 https://github.com/hjdw2/SEMBG 获取。

英文摘要:

Recent advancements in low-cost ensemble learning have demonstrated improved efficiency for image classification. However, the existing low-cost ensemble methods show relatively lower accuracy compared to conventional ensemble learning. In this paper, we propose a new low-cost ensemble learning, which can simultaneously achieve high efficiency and classification performance. A CNN is transformed into a multi-branch structure without introduction of additional components, which maintains the computational complexity as that of the original single model and also enhances diversity among the branches' outputs via sufficient separation between different pathways of the branches. In addition, we propose a new strategy that applies grouped convolution in the branches with different numbers of groups in different branches, which boosts the diversity of the branches' outputs. For training, we employ knowledge distillation using the ensemble of the outputs as the teacher signal. The high diversity among the outputs enables to form a powerful teacher, enhancing the individual branch's classification performance and consequently the overall ensemble performance. Experimental results show that our method achieves state-of-the-art classification accuracy and higher uncertainty estimation performance compared to previous low-cost ensemble methods. The code is available at https://github.com/hjdw2/SEMBG.

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

↑