paper-with-me

홈 › Papers

Automated Loss function Search for Class-imbalanced Node Classification

2024-05-23 · Xinyu Guo, Kai Wu, XiaoYu Zhang, Jing Liu

Class-imbalanced node classification tasks are prevalent in real-world scenarios. Due to the uneven distribution of nodes across different classes, learning high-quality node representations remains a challenging endeavor. The engineering of loss functions has shown promising potential in addressing this issue. It involves the meticulous design of loss functions, utilizing information about the quantities of nodes in different categories and the network's topology to learn unbiased node representations. However, the design of these loss functions heavily relies on human expert knowledge and exhibits limited adaptability to specific target tasks. In this paper, we introduce a high-performance, flexible, and generalizable automated loss function search framework to tackle this challenge. Across 15 combinations of graph neural networks and datasets, our framework achieves a significant improvement in performance compared to state-of-the-art methods. Additionally, we observe that homophily in graph-structured data significantly contributes to the transferability of the proposed framework.

📄 PDF Abstract BibTeX arXiv:2405.14133

Code (0)

등록된 구현이 없습니다.

Tasks

Node Classification

Similar Papers 제목 키워드 기반

Automated Focal Loss for Image based Object Detection

2019-04-19 · Michael Weber, Michael Fürst, J. Marius Zöllner

Current state-of-the-art object detection algorithms still suffer the problem of imbalanced distribution of training data over object classes and background. Recent work introduced a new loss function called focal loss t…

Objectobject-detectionObject Detectionregression+1

Enlarged Large Margin Loss for Imbalanced Classification

2023-06-15 · Sota Kato, Kazuhiro Hotta

We propose a novel loss function for imbalanced classification. LDAM loss, which minimizes a margin-based generalization bound, is widely utilized for class-imbalanced image classification. Although, by using LDAM loss, …

Classificationimage-classificationImage Classificationimbalanced classification

Balanced MSE for Imbalanced Visual Regression

2022-03-30 · CVPR 2022 1 · Jiawei Ren, Mingyuan Zhang, Cunjun Yu, Ziwei Liu

Data imbalance exists ubiquitously in real-world visual regressions, e.g., age estimation and pose estimation, hurting the model's generalizability and fairness. Thus, imbalanced regression gains increasing research atte…

Age EstimationFairnessimbalanced classificationPose Estimation+1

Improving GBDT Performance on Imbalanced Datasets: An Empirical Study of Class-Balanced Loss Functions

2024-07-19 · Jiaqi Luo, Yuan Yuan, Shixin Xu

Class imbalance remains a significant challenge in machine learning, particularly for tabular data classification tasks. While Gradient Boosting Decision Trees (GBDT) models have proven highly effective for such tasks, t…

ClassificationMulti-Label ClassificationMUlTI-LABEL-ClASSIFICATIONtabular-classification

Deep Metric Learning Model for Imbalanced Fault Diagnosis

2021-07-08 · Xingtai Gui, Jiyang Zhang

Intelligent diagnosis method based on data-driven and deep learning is an attractive and meaningful field in recent years. However, in practical application scenarios, the imbalance of time-series fault is an urgent prob…

Fault Diagnosisimbalanced classificationMetric Learningmodel+2