arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2602.15330cs.LGcs.AI
收藏

可扩展的基于好奇心驱动的游戏理论框架用于数据挖掘中的长尾多标签学习

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

Sun Yat-sen University(中山大学)

Jing Yang, Keze Wang

AI总结:

本文提出了一种基于好奇心驱动的游戏理论框架,用于解决数据挖掘中的长尾多标签学习问题,通过协作机制和好奇心奖励提升模型性能。

摘要:

长尾分布,其中少数头部标签占主导地位而稀有尾部标签众多,对大规模多标签分类(MLC)在现实数据挖掘应用中提出了持续的挑战。现有的重采样和再加权策略往往破坏标签之间的依赖关系或需要脆弱的超参数调整,尤其是在标签空间扩展到数万标签时。为了解决这个问题,我们提出了好奇心驱动的游戏理论多标签学习(CD-GTMLL),一个可扩展的协作框架,将长尾MLC重新解释为多玩家游戏——每个子预测器(“玩家”)专门负责标签空间的一部分,协作以最大化全局准确性,同时追求基于尾部标签稀有性和玩家之间分歧的内在好奇心奖励。这种机制能够自适应地将学习信号注入到代表性不足的尾部标签中,而无需手动平衡或调整。我们进一步提供了一个理论分析,显示我们的CD-GTMLL收敛到一个尾部感知的均衡点,并正式将优化动态与Rare-F1指标的改进联系起来。在7个基准测试中进行的广泛实验,包括具有30,000多个标签的极端多标签分类数据集,证明CD-GTMLL在Wiki10-31K上始终优于最先进的方法,最大收益达到+1.6% P@3。消融研究进一步确认了游戏理论合作和好奇心驱动探索对稳健尾部性能的贡献。通过将博弈论与好奇心机制相结合,CD-GTMLL不仅在资源受限环境中提高了模型效率,还为跨行业(如电子商务和医疗保健)的不平衡数据场景中的更适应性学习铺平了道路。

英文摘要:

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.

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

↑