This paper presents a novel Clustering-based Graph Neural Network algorithm for weakly supervised regression. The proposed approach constructs a robust graph representation of the data using a weighted co-association matrix derived from a cluster ensemble, enabling the model to effectively capture complex relationships and reduce the impact of noise and outliers. A graph neural network is then trained on this structure, with manifold regularization via the graph Laplacian allowing the model to effectively utilize both labeled and unlabeled data. This approach improves stability and enhances robustness to label noise. Additionally, Truncated Loss is employed to mitigate the influence of outliers during training, and a Balanced Batch Sampling algorithm is introduced to ensure effective mini-batch training on the constructed graph. Numerical experiments on several real-world regression datasets demonstrate that CBGNN outperforms classical supervised, semi-supervised, and other weakly supervised learning methods, particularly in settings with significant label noise.

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

Clustering-Based Graph Neural Network in a Weakly Supervised Regression Problem

  • Kirill Kalmutskiy,
  • Vladimir Berikov

摘要

This paper presents a novel Clustering-based Graph Neural Network algorithm for weakly supervised regression. The proposed approach constructs a robust graph representation of the data using a weighted co-association matrix derived from a cluster ensemble, enabling the model to effectively capture complex relationships and reduce the impact of noise and outliers. A graph neural network is then trained on this structure, with manifold regularization via the graph Laplacian allowing the model to effectively utilize both labeled and unlabeled data. This approach improves stability and enhances robustness to label noise. Additionally, Truncated Loss is employed to mitigate the influence of outliers during training, and a Balanced Batch Sampling algorithm is introduced to ensure effective mini-batch training on the constructed graph. Numerical experiments on several real-world regression datasets demonstrate that CBGNN outperforms classical supervised, semi-supervised, and other weakly supervised learning methods, particularly in settings with significant label noise.