arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2109.12342cs.CV
收藏

TreeNet:一种轻量级的一次性聚合卷积网络

TreeNet: A lightweight One-Shot Aggregation Convolutional Network

Lu Rao

摘要:

深度卷积网络(CNN)的架构已经发展了多年,变得更加准确和快速。然而,它仍然是具有挑战性的设计合理的网络结构,旨在获得最佳的精度在有限的计算预算。在本文中,我们提出了一个树块,命名的外观,它扩展了一次聚合(OSA)模块,同时更轻量级和灵活。具体地,树块将OSA中的每个$3\times3$ Conv层替换为浅残差块(SRB)和$1\times1$ Conv层的堆叠。$1\times1$ Conv层负责增加维数,SRB被送入下一步骤。通过这样做,当聚合相同数量的后续特征图时,树块具有更深的网络结构,同时具有更低的模型复杂度。此外,剩余连接和有效的信道注意(ECA)被添加到树块,以进一步提高网络的性能。在Tree模块的基础上,我们建立了高效的主干模型TreeNets。TreeNet具有与ResNet类似的网络架构,因此可以灵活地在各种计算机视觉框架中取代ResNet。我们在常用的基准测试中全面评估了TreeNet,包括用于分类的ImageNet-1 k,用于目标检测的MS COCO和实例分割。实验结果表明,TreeNet是更有效的,并表现出良好的对当前国家的最先进的骨干方法。

英文摘要:

The architecture of deep convolutional networks (CNNs) has evolved for years, becoming more accurate and faster. However, it is still challenging to design reasonable network structures that aim at obtaining the best accuracy under a limited computational budget. In this paper, we propose a Tree block, named after its appearance, which extends the One-Shot Aggregation (OSA) module while being more lightweight and flexible. Specifically, the Tree block replaces each of the $3\times3$ Conv layers in OSA into a stack of shallow residual block (SRB) and $1\times1$ Conv layer. The $1\times1$ Conv layer is responsible for dimension increasing and the SRB is fed into the next step. By doing this, when aggregating the same number of subsequent feature maps, the Tree block has a deeper network structure while having less model complexity. In addition, residual connection and efficient channel attention(ECA) is added to the Tree block to further improve the performance of the network. Based on the Tree block, we build efficient backbone models calling TreeNets. TreeNet has a similar network architecture to ResNet, making it flexible to replace ResNet in various computer vision frameworks. We comprehensively evaluate TreeNet on common-used benchmarks, including ImageNet-1k for classification, MS COCO for object detection, and instance segmentation. Experimental results demonstrate that TreeNet is more efficient and performs favorably against the current state-of-the-art backbone methods.

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

↑