Unsupervised learned embeddings with LightGBM model downstream evaluations:
mean |
|
---|---|
Gender | AUROC |
baseline | 0.877 |
cpc_embeddings | 0.851 |
mles2_embeddings | 0.882 |
mles_embeddings | 0.881 |
nsp_embeddings | 0.852 |
random_encoder | 0.593 |
rtd_embeddings | 0.855 |
sop_embeddings | 0.785 |
barlow_twins | 0.865 |
Age group (age_pred) | Accuracy |
baseline | 0.629 |
cpc_embeddings | 0.602 |
mles2_embeddings | 0.643 |
mles_embeddings | 0.640 |
mles_longformer | 0.630 |
nsp_embeddings | 0.621 |
random_encoder | 0.375 |
rtd_embeddings | 0.631 |
sop_embeddings | 0.512 |
barlow_twins | 0.634 |
coles_transformer | 0.646 |
Churn (rosbank) | AUROC |
baseline | 0.827 |
cpc_embeddings | 0.792 |
mles2_embeddings | 0.837 |
mles_embeddings | 0.841 |
nsp_embeddings | 0.828 |
random_encoder | 0.725 |
rtd_embeddings | 0.771 |
sop_embeddings | 0.780 |
barlow_twins | 0.839 |
Assessment (bowl2019) | Accuracy |
barlow_twins | 0.595 |
baseline | 0.592 |
cpc_embeddings | 0.593 |
mles2_embeddings | 0.588 |
mles_embeddings | 0.597 |
nsp_embeddings | 0.579 |
random_encoder | 0.574 |
rtd_embeddings | 0.574 |
sop_embeddings | 0.567 |
Retail (x5) | Accuracy |
baseline | 0.547 |
cpc_embeddings | 0.525 |
mles_embeddings | 0.539 |
nsp_embeddings | 0.425 |
rtd_embeddings | 0.520 |
sop_embeddings | 0.428 |
Scoring (alpha battle) | AUROC |
baseline | 0.7792 |
random_encoder | 0.6456 |
barlow_twins | 0.7878 |
cpc | 0.7919 |
mles | 0.7921 |
nsp | 0.7655 |
rtd | 0.7910 |
sop | 0.7238 |
mlmnsp | 0.7591 |
tabformer | 0.7862 |
gpt | 0.7737 |
coles_transformer | 0.7968 |
Supervised finetuned encoder with MLP head evaluation:
mean |
|
---|---|
Gender | AUROC |
barlow_twins | 0.865 |
cpc_finetuning | 0.865 |
mles_finetuning | 0.879 |
rtd_finetuning | 0.868 |
target_scores | 0.867 |
Age group (age_pred) | Accuracy |
barlow_twins | 0.619 |
cpc_finetuning | 0.625 |
mles_finetuning | 0.624 |
rtd_finetuning | 0.622 |
target_scores | 0.620 |
Churn (rosbank) | AUROC |
barlow_twins | 0.830 |
cpc_finetuning | 0.804 |
mles_finetuning | 0.819 |
nsp_finetuning | 0.806 |
rtd_finetuning | 0.791 |
target_scores | 0.818 |
Assessment (bowl2019) | Accuracy |
barlow_twins | 0.561 |
cpc_finetuning | 0.594 |
mles_finetuning | 0.577 |
rtd_finetuning | 0.571 |
target_scores | 0.585 |
Retail (x5) | Accuracy |
cpc_finetuning | 0.549 |
mles_finetuning | 0.552 |
rtd_finetuning | 0.544 |
target_scores | 0.542 |