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

用于组合零样本学习的分层视觉基元专家

Hierarchical Visual Primitive Experts for Compositional Zero-Shot Learning

Yonsei University(延世大学);NAVER AI Lab(NAVER人工智能实验室);Korea Institute of Science and Technology (KIST)(韩国科学技术研究院)

Hanjae Kim, Jiyoung Lee, Seongheon Park, Kwanghoon Sohn

AI总结:

本文提出Composition Transformer(CoT)框架,通过分层物体与属性专家显式建模属性-物体语境关系,并结合少数属性增强策略缓解长尾偏置,在多个组合零样本学习基准上取得最先进性能。

摘要:

组合零样本学习(CZSL)旨在利用已知基元(属性和物体)的先验知识来识别未见过的组合。以往针对CZSL的工作通常在把握属性与物体之间的语境关系、视觉特征的判别力以及真实世界组合数据的长尾分布方面存在困难。我们提出了一个简单且可扩展的框架,称为Composition Transformer(CoT),以解决这些问题。CoT以独特的方式使用物体专家和属性专家,通过分层利用视觉网络来生成具有代表性的嵌入。物体专家以自底向上的方式从最后一层提取具有代表性的物体嵌入,而属性专家则以自顶向下的方式生成属性嵌入,并借助所提出的物体引导注意力模块显式建模语境关系。为了纠正由不平衡数据分布引起的偏置预测,我们开发了一种简单的少数属性增强(MAA)方法,通过混合两张图像并对少数属性类别进行过采样来合成虚拟样本。我们的方法在多个基准上取得了最先进(SoTA)性能,包括MIT-States、C-GQA和VAW-CZSL。我们还证明了CoT在改善视觉判别力和解决不平衡数据分布带来的模型偏置方面的有效性。代码可在https://github.com/HanjaeKim98/CoT获取。

英文摘要:

Compositional zero-shot learning (CZSL) aims to recognize unseen compositions with prior knowledge of known primitives (attribute and object). Previous works for CZSL often suffer from grasping the contextuality between attribute and object, as well as the discriminability of visual features, and the long-tailed distribution of real-world compositional data. We propose a simple and scalable framework called Composition Transformer (CoT) to address these issues. CoT employs object and attribute experts in distinctive manners to generate representative embeddings, using the visual network hierarchically. The object expert extracts representative object embeddings from the final layer in a bottom-up manner, while the attribute expert makes attribute embeddings in a top-down manner with a proposed object-guided attention module that models contextuality explicitly. To remedy biased prediction caused by imbalanced data distribution, we develop a simple minority attribute augmentation (MAA) that synthesizes virtual samples by mixing two images and oversampling minority attribute classes. Our method achieves SoTA performance on several benchmarks, including MIT-States, C-GQA, and VAW-CZSL. We also demonstrate the effectiveness of CoT in improving visual discrimination and addressing the model bias from the imbalanced data distribution. The code is available at https://github.com/HanjaeKim98/CoT.

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

↑