Table 2.
Efficiency comparison. The “Memory” columns record GPU’s memory occupation of different models, while the “Time/Epoch” columns record average time consumption per epoch.
| Model | Criteo | Avazu | ||
|---|---|---|---|---|
| Memory | Time/Epoch | Memory | Seconds/Epoch | |
| Wide & Deep | 12.08 G | 983 s | 5.73 G | 380 s |
| IPNN | 11.76 G | 854 s | 5.19 G | 371 s |
| DeepFM | 12.07 G | 1040 s | 4.94 G | 394 s |
| DCN | 11.85 G | 915 s | 5.39 G | 375 s |
| xDeepFM | 11.49 G | 1388 s | 5.14 G | 494 s |
| AFN+ | 22.96 G | 1783 s | 10.98 G | 612 s |
| GAIN | 13.05 G | 881 s | 6.66 G | 351 s |