Figure 1.
Figure 2.
Figure 3.
Figure 4.
Figure 5.
Figure 6.
Figure 7.
Figure 8.
Figure 9.
Figure 10.
Figure 11.
Figure 12.
Figure 13.
PARAMETER SETTINGS
| Parameter Name | Value |
|---|---|
| batch_size | 256 |
| optimizer | adam |
| learning_rate | 0.001 |
| embedding_size | 4 |
| dnn_layers | (512, 256) |
| dnn_use_bn | True |
Results of the UCI Census Income Dataset
| Models | AUC/ Income | AUC/ Marital | Mean |
|---|---|---|---|
| Single-Task | 0.9198 | 0.9748 | 0.9473 |
| Shared-Bottom | 0.9148 | 0.9754 | 0.9451 |
| MMoE | 0.9152 | 0.9756 | 0.9454 |
| PLE | 0.9161 | 0.9764 | 0.9463 |
| Base | 0.9134 | 0.9706 | 0.9420 |
| FG | 0.9146 | 0.9693 | 0.9420 |
| FS | 0.9216 | 0.9756 | 0.9486 |
Parameter Settings
| Models | AUC/ Income | AUC/ Marital | Mean |
|---|---|---|---|
| Single-Task | 0.7192 | 0.6635 | 0.6914 |
| DeepFM | 0.719 | 0.658 | 0.689 |
| Shared-Bottom | 0.7184 | 0.6916 | 0.705 |
| ESMM | 0.7189 | 0.6897 | 0.7043 |
| MMoE | 0.7201 | 0.7057 | 0.7129 |
| Base | 0.7193 | 0.7086 | 0.714 |
| FG | 0.7201 | 0.7084 | 0.7143 |
| FS | 0.7204 | 0.7114 | 0.7159 |
| FSMD | 0.7205 | 0.7117 | 0.7161 |