dmlc / dmlc/xgboost

Training on GPU degrades F1 score

Open
#9,779 2 comments 0 reactions 0 assignees View on GitHub
Dominant language
C++
Stars
28.8k
Forks
8.9k
Avg merge
1d 12h
Merged PRs (30d)
54

Description

Hi folks, we have produced multiple experiments that shows some indicators that the training on GPU on a very skewed dataset will produce worse metrics than a training on CPU. Both training are distributed using https://github.com/ray-project/xgboost_ray.

On test set:
On CPU training, we are able to produces F1: ~0.15 and F2:~0.176 consistently.
On GPU training, we are only able to produce up to F1: 0.135 and F2:0.166 throughout at best over multiple runs

XGBoost version: 1.5.2 (both CPU and GPU)
Pre-processing steps: vectorize for dmatrix

Contributor guide

No contributing guide indexed for this repository

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.