arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2305.12349cs.LGcs.IR

PINA:通过预测实例邻域聚合在极端多标签分类中利用侧信息

PINA: Leveraging Side Information in eXtreme Multi-label Classification via Predicted Instance Neighborhood Aggregation

University of Illinois Urbana-Champaign(伊利诺伊大学厄巴纳-香槟分校);Amazon(亚马逊);University of California, Los Angeles(加利福尼亚大学洛杉矶分校)

Eli Chien, Jiong Zhang, Cho-Jui Hsieh, Jyun-Yu Jiang, Wei-Cheng Chang, Olgica Milenkovic, Hsiang-Fu Yu

AI总结:

提出PINA方法,通过利用标签元数据和训练实例间的相关性等侧信息进行数据增强,在多个XMC基准数据集上显著提升了分类准确率。

摘要:

极端多标签分类(XMC)问题旨在从极大的标签空间中寻找相关标签。现有大多数XMC学习器专注于从输入查询文本中提取语义特征。然而,传统XMC研究通常忽略实例和标签的侧信息,而这些信息在推荐系统和电商产品搜索等许多实际应用中具有利用价值。我们提出了预测实例邻域聚合(PINA),这是一种针对通用XMC问题的数据增强方法,能够利用有益的侧信息。与大多数将标签和输入实例视为无特征指示器和独立条目的现有XMC框架不同,PINA从标签元数据和训练实例间的相关性中提取信息。大量实验结果表明,与最先进的方法相比,PINA在各种XMC任务上均表现出一致的提升:在五个公开基准数据集上,PINA的准确率优于标准的XR-Transformers。此外,在最大数据集LF-AmazonTitles-1.3M上,PINA实现了约5%的准确率提升。我们的实现已公开可用。

英文摘要:

The eXtreme Multi-label Classification~(XMC) problem seeks to find relevant labels from an exceptionally large label space. Most of the existing XMC learners focus on the extraction of semantic features from input query text. However, conventional XMC studies usually neglect the side information of instances and labels, which can be of use in many real-world applications such as recommendation systems and e-commerce product search. We propose Predicted Instance Neighborhood Aggregation (PINA), a data enhancement method for the general XMC problem that leverages beneficial side information. Unlike most existing XMC frameworks that treat labels and input instances as featureless indicators and independent entries, PINA extracts information from the label metadata and the correlations among training instances. Extensive experimental results demonstrate the consistent gain of PINA on various XMC tasks compared to the state-of-the-art methods: PINA offers a gain in accuracy compared to standard XR-Transformers on five public benchmark datasets. Moreover, PINA achieves a $\sim 5\%$ gain in accuracy on the largest dataset LF-AmazonTitles-1.3M. Our implementation is publicly available.

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

↑