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

使用自适应图卷积网络的多标签图像分类:从单一域到多域

Multi-label Image Classification using Adaptive Graph Convolutional Networks: from a Single Domain to Multiple Domains

Interdisciplinary Centre for Security, Reliability and Trust (SnT), University of Luxembourg(卢森堡大学安全、可靠性与信任跨学科中心(SnT))

Indel Pal Singh, Enjie Ghorbel, Oyebade Oyedotun, Djamila Aouada

AI总结:

本文提出一种端到端学习的自适应图卷积网络,通过注意力机制和相似性保持策略优化图拓扑,并扩展至多域,在单域和多域基准上以较小模型取得有竞争力的mAP。

摘要:

本文提出了一种基于自适应图的方法用于多标签图像分类。图方法在多标签分类领域已被广泛利用,因为它们能够对标签相关性进行建模。具体来说,其有效性不仅在考虑单一域时得到证明,而且在考虑多个域时也得到了证明。然而,所使用的图的拓扑结构并非最优,因为它是启发式预定义的。此外,连续的图卷积网络(GCN)聚合往往会破坏特征相似性。为了克服这些问题,本文引入了一种以端到端方式学习图连接性的架构。这是通过整合基于注意力的机制和相似性保持策略来实现的。所提出的框架随后使用对抗训练方案扩展到多个域。在著名的单域和多域基准上报告了大量实验。结果表明,与最先进的方法相比,我们的方法在平均精度均值(mAP)和模型大小方面取得了有竞争力的结果。代码将公开提供。

英文摘要:

This paper proposes an adaptive graph-based approach for multi-label image classification. Graph-based methods have been largely exploited in the field of multi-label classification, given their ability to model label correlations. Specifically, their effectiveness has been proven not only when considering a single domain but also when taking into account multiple domains. However, the topology of the used graph is not optimal as it is pre-defined heuristically. In addition, consecutive Graph Convolutional Network (GCN) aggregations tend to destroy the feature similarity. To overcome these issues, an architecture for learning the graph connectivity in an end-to-end fashion is introduced. This is done by integrating an attention-based mechanism and a similarity-preserving strategy. The proposed framework is then extended to multiple domains using an adversarial training scheme. Numerous experiments are reported on well-known single-domain and multi-domain benchmarks. The results demonstrate that our approach achieves competitive results in terms of mean Average Precision (mAP) and model size as compared to the state-of-the-art. The code will be made publicly available.

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

↑