arXiv:2602.15330cs.LGcs.AI2026-02

用游戏博弈+好奇机制,让模型自动关注罕见标签,提升长尾多标签分类效果。

A Scalable Curiosity-Driven Game-Theoretic Framework for Long-Tail Multi-Label Learning in Data Mining

  • 将多标签分类设计为玩家协作游戏,各子模型专注不同标签组,通过好奇度驱动学习。
  • 在含3万+标签的Wiki10-31K数据集上,P@3指标提升最高达+1.6%。
  • 无需人工调参或重采样,适合电商、医疗等长尾标签场景。

长尾分布下,少数头部标签主导,大量尾部标签稀少,严重挑战大规模多标签分类任务。现有重采样与重加权策略常破坏标签间依赖关系,且在标签数达数万时需繁琐调参。为此,我们提出可扩展的好奇心驱动博弈论多标签学习框架(CD-GTMLL),将长尾多标签分类重构为多玩家博弈:每个子预测器(“玩家”)负责标签空间的一部分,协同优化全局准确率,并基于尾部标签稀有性和玩家间分歧获取内在好奇奖励。该机制无需人工平衡即可自适应向低频标签注入学习信号。理论分析表明,CD-GTMLL收敛至关注尾部的均衡点,且优化动态与罕见类别F1指标提升直接相关。在7个基准数据集上的实验,包括含30,000+标签的极端多标签数据集,均显示其优于当前最优方法,最高在Wiki10-31K上实现+1.6% P@3提升。消融研究证实博弈合作与好奇探索对尾部性能的关键贡献。融合博弈论与好奇机制,不仅提升资源受限环境下的模型效率,也为工业领域不平衡数据的自适应学习提供新路径。

原文摘要 · Abstract (English)

The long-tail distribution, where a few head labels dominate while rare tail labels abound, poses a persistent challenge for large-scale Multi-Label Classification (MLC) in real-world data mining applications. Existing resampling and reweighting strategies often disrupt inter-label dependencies or require brittle hyperparameter tuning, especially as the label space expands to tens of thousands of labels. To address this issue, we propose Curiosity-Driven Game-Theoretic Multi-Label Learning (CD-GTMLL), a scalable cooperative framework that recasts long-tail MLC as a multi-player game - each sub-predictor ("player") specializes in a partition of the label space, collaborating to maximize global accuracy while pursuing intrinsic curiosity rewards based on tail label rarity and inter-player disagreement. This mechanism adaptively injects learning signals into under-represented tail labels without manual balancing or tuning. We further provide a theoretical analysis showing that our CD-GTMLL converges to a tail-aware equilibrium and formally links the optimization dynamics to improvements in the Rare-F1 metric. Extensive experiments across 7 benchmarks, including extreme multi-label classification datasets with 30,000+ labels, demonstrate that CD-GTMLL consistently surpasses state-of-the-art methods, with gains up to +1.6% P@3 on Wiki10-31K. Ablation studies further confirm the contributions of both game-theoretic cooperation and curiosity-driven exploration to robust tail performance. By integrating game theory with curiosity mechanisms, CD-GTMLL not only enhances model efficiency in resource-constrained environments but also paves the way for more adaptive learning in imbalanced data scenarios across industries like e-commerce and healthcare.

多标签分类长尾学习游戏博弈好奇机制

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。