atlas

Research map / Architectures and efficiency / Attention and transformers

Long-Sequence Recommendation Models Need Decoupled Embeddings

ICLR 2025 · Architectures and efficiency › Attention and transformers · approach: State space and recurrent sequence models · 0 citations (OpenAlex)

Read the paperCheck your idea against 108,967 papers

AI-extracted summary. Written by AI from the paper's full text. Numbers and claims can be wrong or missing context, so check the paper before you cite or build on it.

Proposes DARE, which uses two separate embedding tables (one for attention, one for representation) in long-sequence CTR models to fix gradient domination and conflict between the two modules, improving AUC and enabling faster search.

Problem

Long-sequence recommendation models share one embedding table for both the attention (behavior retrieval) and representation modules, causing gradient domination and conflict that degrade both attention accuracy and representation discriminability; common fixes like linear projections fail due to lo

Approach

Decouple attention and representation at the embedding level by using two independent embedding tables, with target-aware representation (element-wise product) for the representation branch; this also allows a smaller attention embedding dimension to accelerate the search stage.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Decouple attention and representation into separate embedding tablesTWIN · Taobao and Tmall, embedding dimensions 16, 64, and 128AUC: AUC improvements of 1\textperthousand~and 9\textperthousand~Helped
Decouple attention and representation embeddingsTWIN · Behavior retrieval on test dataNDCG: 0.8124 vs. 0.5545Helped
Decouple attention and representation embeddingsTWIN · Behavior retrieval on test dataNDCG: 0.8124 vs. 0.6382Helped
Reduce the attention embedding dimension (ablation)DARE · Long-sequence inferencesearch speed: 50%Helped
Reduce the attention embedding dimension (ablation)DARE · Long-sequence inferenceAUCNo clear effect
Reduce the attention embedding dimension further (ablation)DARE · Long-sequence inferencesearch speed: 75%Helped
Remove target-aware representation (ablation)TWIN · Taobao and TmallAUC: 0.91688 -> 0.90732; 0.95812 -> 0.96170Mixed
Replace shared history/target embeddings with four independent embeddings (ablation)TWIN · Taobao and TmallAUC: 0.91688 -> 0.90414; 0.95812 -> 0.96124Hurt
Add linear projections for attention and representation (ablation)TWIN · Taobao and TmallAUC: 0.91688 -> 0.89642; 0.95812 -> 0.96152Mixed
Apply linear projection to category features (ablation)TWIN · Taobao and TmallAUC: 0.91688 -> 0.91738; 0.95812 -> 0.96658Helped
Replace linear projection with MLP projection (ablation)TWIN · Taobao and TmallAUC: 0.91688 -> 0.86960; 0.95812 -> 0.95678Hurt
Use gradient normalization to avoid representation-gradient domination (ablation)TWIN · Taobao and TmallAUC: 0.91688 -> 0.91976; 0.95812 -> 0.93887Mixed

Surprising findings

Limitations

Experimental setup

Datasets: Taobao Tmall Amazon

Benchmarks: Taobao Tmall Amazon

Metrics: AUC GAUC Logloss NDCG mutual information GMV lift

Baselines: ETA SDIM DIN TWIN TWIN (hard) TWIN (w/ proj.) TWIN (w/o TR) TWIN-V2 TWIN-4E

Builds on: TWIN SIM DIN TIN Transformer self-attention

Closest papers by meaning

Writing a paper in this area? scime draft is a free LaTeX editor with AI included, real-time co-authors and a check of your references and venue rules before you submit. It also imports OpenAI Prism projects.

Open scime draft

Get the next scime Dispatch and a monthly field update

Free. We send a confirmation link first; every email has a one-click unsubscribe. Privacy