用大模型嵌入和属性平滑提升组合零样本识别准确率
Leveraging MLLM Embeddings and Attribute Smoothing for Compositional Zero-Shot Learning
- 用可学习掩码捕捉多粒度特征,分离属性与物体
- 在三个数据集上达到新最优,最高提升6.2%
- 适合做视觉-语言联合建模的开发者参考
组合零样本学习(CZSL)旨在识别由已见组合中未出现过的属性与物体构成的新组合。现有方法通过提取共享与独有部分并匹配预训练词向量来解耦属性与物体,但存在三大局限:(1) 背景干扰和属性-物体在同一区域的复杂纠缠导致解耦效果下降;(2) 传统词向量难以捕捉复杂多模态语义;(3) 模型对已见组合过度自信,影响对新组合的泛化能力。为此,我们提出一种新框架:利用特征自适应聚合模块抑制背景影响,通过可学习条件掩码捕捉多粒度特征以实现解耦;采用多模态大语言模型(MLLM)的最后隐藏状态作为词向量,提升表示能力;同时引入大语言模型生成的辅助属性对已见组合进行属性平滑,缓解过自信问题。大量实验表明,该方法在三个挑战性数据集上均取得当前最优性能,最高提升达6.2%。源代码将公开于https://github.com/xud-yan/Trident。
原文摘要 · Abstract (English)
Compositional zero-shot learning (CZSL) aims to recognize novel compositions of attributes and objects learned from seen compositions. Previous works disentangle attributes and objects by extracting shared and exclusive parts between the image pair sharing the same attribute (object), as well as aligning them with pretrained word embeddings to improve unseen attribute-object recognition. Despite the significant achievements of existing efforts, they are hampered by three limitations: (1) The efficacy of disentanglement is compromised due to the influence of the background and the intricate entanglement of attributes with objects in the same parts. (2) Existing word embeddings fail to capture complex multimodal semantic information. (3) Overconfidence exhibited by existing models in seen compositions hinders their generalization to novel compositions. Being aware of these, we propose a novel framework named multimodal large language model (MLLM) embeddings and attribute smoothing guided disentanglement for CZSL. First, we leverage feature adaptive aggregation modules to mitigate the impact of background, and utilize learnable condition masks to capture multi-granularity features for disentanglement. Moreover, the last hidden states of MLLM are employed as word embeddings for their superior representation capabilities. Furthermore, we propose attribute smoothing with auxiliary attributes generated by the large language model (LLM) for seen compositions to address the overconfidence challenge. Extensive experiments demonstrate that our method achieves state-of-the-art performance on three challenging datasets. The source code will be available at https://github.com/xud-yan/Trident .
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。