计算机科学 > 机器学习
[提交于 2020年3月10日
(v1)
,最后修订 2021年8月10日 (此版本, v2)]
标题: GPCA:高斯过程嵌入通道注意力的概率框架
标题: GPCA: A Probabilistic Framework for Gaussian Process Embedded Channel Attention
摘要: 通道注意力机制已被广泛应用于许多视觉任务中,以有效提升性能。它能够增强有信息量的通道并抑制无用的通道。最近,不同的通道注意力模块已被提出并以各种方式实现。一般来说,它们主要基于卷积和池化操作。在本文中,我们提出了嵌入高斯过程的通道注意力(GPCA)模块,并进一步以概率方式解释通道注意力方案。GPCA模块旨在建模通道之间的相关性,这些相关性被假设由贝塔分布变量捕获。由于贝塔分布无法以数学上可处理的方式集成到卷积神经网络(CNNs)的端到端训练中,我们利用贝塔分布的近似来解决这个问题。具体来说,我们采用了一种Sigmoid-高斯近似,其中高斯分布变量被转换到区间[0,1]。然后利用高斯过程来建模不同通道之间的相关性。在这种情况下,得到了一个数学上可处理的解决方案。GPCA模块可以高效实现并集成到CNNs的端到端训练中。实验结果证明了所提出的GPCA模块的出色性能。代码可在https://github.com/PRIS-CV/GPCA获取。
文献和引用工具
与本文相关的代码,数据和媒体
alphaXiv (什么是 alphaXiv?)
CatalyzeX 代码查找器 (什么是 CatalyzeX?)
DagsHub (什么是 DagsHub?)
Gotit.pub (什么是 GotitPub?)
Hugging Face (什么是 Huggingface?)
带有代码的论文 (什么是带有代码的论文?)
ScienceCast (什么是 ScienceCast?)
演示
推荐器和搜索工具
arXivLabs:与社区合作伙伴的实验项目
arXivLabs 是一个框架,允许合作伙伴直接在我们的网站上开发和分享新的 arXiv 特性。
与 arXivLabs 合作的个人和组织都接受了我们的价值观,即开放、社区、卓越和用户数据隐私。arXiv 承诺这些价值观,并且只与遵守这些价值观的合作伙伴合作。
有一个为 arXiv 社区增加价值的项目想法吗? 了解更多关于 arXivLabs 的信息.