Skip to main content

Table 3 IARM's performance was studied using ablation tests

From: Click-through rate prediction model integrating user interest and multi-head attention mechanism

Data sets

Model

AUC

LOSS

Criteo

IARM

0.7545

0.4830

IARM*

0.7371

0.4965

Avazu

IARM

0.7652

0.3994

IARM*

0.7591

0.3995

Movielens

IARM

0.8386

0.3296

IARM*

0.8349

0.3344

  1. IARM is a full model, whereas IARM* is a model that eliminates user interest