中文

GPU-Net:具有更丰富多样特征的轻量级 U-Net

图像与视频处理 2022-01-11 v1 计算机视觉与模式识别

摘要

图像分割是医学图像领域的一项重要任务,已有许多基于卷积神经网络(CNNs)的方法被提出,其中 U-Net 及其变体展现出良好性能。本文中,我们提出基于 U-Net 的 GP-module 与 GPU-Net,通过引入 Ghost module 与空洞空间金字塔池化(ASPP)可学习更丰富的多样特征。我们的方法以少于 4 倍的参数量与减半的 FLOPs 取得更优性能,为未来研究提供了新潜在方向。我们的即插即用模块也可应用于现有分割方法以进一步提升其性能。

关键词

引用

@article{arxiv.2201.02656,
  title  = {GPU-Net: Lightweight U-Net with more diverse features},
  author = {Heng Yu and Di Fan and Weihu Song},
  journal= {arXiv preprint arXiv:2201.02656},
  year   = {2022}
}