中文

DisenSemi:基于解缠表示学习的半监督图分类

机器学习 2024-08-12 v2 人工智能 信息检索 社会与信息网络

摘要

图分类是众多多媒体应用中的关键任务,图被用于表示包括图像、视频和社交网络等多种类型的数据。然而,在实际场景中,标记图数据可能有限或稀缺。为此,我们聚焦于半监督图分类问题,即模型从标记和未标记数据中进行学习。与最近的做法不同,我们认为有效的迁移应仅保留与监督任务一致良好的相关语义。本文提出一种名为 DisenSemi 的新型框架,用于半监督图分类。具体而言,提出了解缠图编码器,用于为监督模型和非监督模型生成分量级图表示。随后,我们分别通过监督目标和互信息 (MI) 约束训练两个模型。为确保从非监督编码器到监督编码器之间知识的有意义迁移,我们进一步定义了基于 MI 的解缠一致性正则化,并确定了与当前图分类任务一致的依据。实验在多个公开数据集上表明,我们的 DisenSemi 方法有效。

关键词

引用

@article{arxiv.2407.14081,
  title  = {DisenSemi: Semi-supervised Graph Classification via Disentangled Representation Learning},
  author = {Yifan Wang and Xiao Luo and Chong Chen and Xian-Sheng Hua and Ming Zhang and Wei Ju},
  journal= {arXiv preprint arXiv:2407.14081},
  year   = {2024}
}

备注

Accepted by IEEE Transactions on Neural Networks and Learning Systems (TNNLS 2024)