提出EVINET框架,实现图学习中的误判与未知类检测。
EVINET: Towards Open-World Graph Learning via Evidential Reasoning Network
- 用贝塔嵌入结合主观逻辑,构建双推理模块
- 在多个数据集上优于现有方法,提升检测准确率
- 适合开放世界图学习、需要可信预测的场景
图学习在诸多实际任务中至关重要,但传统方法多基于封闭世界假设,即所有标签已知。为在开放且嘈杂环境中有效进行图学习,必须让模型在对已知类别数据误判时发出警告(误分类检测),或在遇到新类别时识别其为分布外数据(分布外检测)。本文提出证据推理网络(EVINET),通过在主观逻辑框架中引入贝塔嵌入,集成两种关键模块:用于误分类检测的矛盾推理模块和用于分布外检测的空缺推理模块。大量实验表明,EVINET在分布内分类、误分类检测和分布外检测任务中,多项指标均超越当前最优方法。结果验证了不确定性估计与逻辑推理对两类检测的重要性,为开放世界图学习奠定基础。代码与数据已在https://github.com/SSSKJ/EviNET公开。
原文摘要 · Abstract (English)
Graph learning has been crucial to many real-world tasks, but they are often studied with a closed-world assumption, with all possible labels of data known a priori. To enable effective graph learning in an open and noisy environment, it is critical to inform the model users when the model makes a wrong prediction to in-distribution data of a known class, i.e., misclassification detection or when the model encounters out-of-distribution from novel classes, i.e., out-of-distribution detection. This paper introduces Evidential Reasoning Network (EVINET), a framework that addresses these two challenges by integrating Beta embedding within a subjective logic framework. EVINET includes two key modules: Dissonance Reasoning for misclassification detection and Vacuity Reasoning for out-of-distribution detection. Extensive experiments demonstrate that EVINET outperforms state-of-the-art methods across multiple metrics in the tasks of in-distribution classification, misclassification detection, and out-of-distribution detection. EVINET demonstrates the necessity of uncertainty estimation and logical reasoning for misclassification detection and out-of-distribution detection and paves the way for open-world graph learning. Our code and data are available at https://github.com/SSSKJ/EviNET.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。