中文

面向类别不平衡节点分类的自动损失函数搜索

机器学习 2024-05-24 v1 人工智能 符号计算

摘要

类别不平衡节点分类任务在现实场景中广泛存在。由于节点在不同类别之间的分布不均,学习高质量的节点表示仍然具有挑战性。损失函数的工程化设计在解决这一问题方面显示出巨大的潜力,涉及对损失函数进行精细设计,利用不同类别节点数量及网络拓扑信息来学习无偏的节点表示。然而,这些损失函数的设计高度依赖人类专家知识,且对特定目标任务的适应性有限。本文引入了一个高性能、灵活且可泛化的自动化损失函数搜索框架,以应对这一挑战。在 15 种图神经网络与数据集的组合上,本文框架的性能显著优于现有最先进方法。此外,我们观察到,图结构数据中的同质性(homophily)对本文框架的可迁移性贡献显著。

关键词

引用

@article{arxiv.2405.14133,
  title  = {Automated Loss function Search for Class-imbalanced Node Classification},
  author = {Xinyu Guo and Kai Wu and Xiaoyu Zhang and Jing Liu},
  journal= {arXiv preprint arXiv:2405.14133},
  year   = {2024}
}

备注

ICML 2024