We gratefully acknowledge support from
the Simons Foundation and member institutions.
Full-text links:

Download:

Current browse context:

cs.IR

Change to browse by:

cs

References & Citations

DBLP - CS Bibliography

Bookmark

(what is this?)
CiteULike logo BibSonomy logo Mendeley logo del.icio.us logo Digg logo Reddit logo ScienceWISE logo

Computer Science > Information Retrieval

Title: AIM: Automatic Interaction Machine for Click-Through Rate Prediction

Abstract: Feature embedding learning and feature interaction modeling are two crucial components of deep models for Click-Through Rate (CTR) prediction. Most existing deep CTR models suffer from the following three problems. First, feature interactions are either manually designed or simply enumerated. Second, all the feature interactions are modeled with an identical interaction function. Third, in most existing models, different features share the same embedding size which leads to memory inefficiency. To address these three issues mentioned above, we propose Automatic Interaction Machine (AIM) with three core components, namely, Feature Interaction Search (FIS), Interaction Function Search (IFS) and Embedding Dimension Search (EDS), to select significant feature interactions, appropriate interaction functions and necessary embedding dimensions automatically in a unified framework. Specifically, FIS component automatically identifies different orders of essential feature interactions with useless ones pruned; IFS component selects appropriate interaction functions for each individual feature interaction in a learnable way; EDS component automatically searches proper embedding size for each feature. Offline experiments on three large-scale datasets validate the superior performance of AIM. A three-week online A/B test in the recommendation service of a mainstream app market shows that AIM improves DeepFM model by 4.4% in terms of CTR.
Comments: Accepted by IEEE Transactions on Knowledge and Data Engineering (TKDE), the codes are availiable at this https URL arXiv admin note: text overlap with arXiv:2003.11235
Subjects: Information Retrieval (cs.IR)
Cite as: arXiv:2111.03318 [cs.IR]
  (or arXiv:2111.03318v2 [cs.IR] for this version)

Submission history

From: Chenxu Zhu [view email]
[v1] Fri, 5 Nov 2021 08:39:34 GMT (19405kb,D)
[v2] Mon, 13 Dec 2021 06:51:22 GMT (4693kb,D)

Link back to: arXiv, form interface, contact.