Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

OneTrans: Unified Feature Interaction and Sequence Modeling with One Transformer in Industrial Recommender

About

In recommendation systems, scaling up feature-interaction modules (e.g., Wukong, RankMixer) or user-behavior sequence modules (e.g., LONGER) has achieved notable success. However, these efforts typically proceed on separate tracks, which not only hinders bidirectional information exchange but also prevents unified optimization and scaling. In this paper, we propose OneTrans, a unified Transformer backbone that simultaneously performs user-behavior sequence modeling and feature interaction. OneTrans employs a unified tokenizer to convert both sequential and non-sequential attributes into a single token sequence. The stacked OneTrans blocks share parameters across similar sequential tokens while assigning token-specific parameters to non-sequential tokens. Through causal attention and cross-request KV caching, OneTrans enables precomputation and caching of intermediate representations, significantly reducing computational costs during both training and inference. Experimental results on industrial-scale datasets demonstrate that OneTrans scales efficiently with increasing parameters, consistently outperforms strong baselines, and yields a 5.68% lift in per-user GMV in online A/B tests.

Zhaoqi Zhang, Haolei Pei, Jun Guo, Tianyu Wang, Yufei Feng, Hui Sun, Shaowei Liu, Aixin Sun• 2025

Related benchmarks

TaskDatasetResultRank
Click-Through Rate PredictionIndustrial
AUC74.82
104
CTR PredictionIndustrial max sequence length 1,600 (test)
AUC0.7239
18
CTR PredictionTaobao latest 600 clicked items (test)
AUC63.51
18
CTR PredictionSQS
AUC90.89
16
Add-to-Cart PredictionIndustrial-scale Recommender Dataset
AUC86.77
14
Conversion Rate (CVR) PredictionIndustrial-scale Recommender Dataset
AUC91.12
14
Finish Click-Through Rate PredictionDouyin Offline
AUC1.05
12
Skip Click-Through Rate PredictionDouyin Offline
AUC130
12
Click-Through Rate PredictionAmazon Product Reviews Sports (test)
AUC0.6775
11
Click-Through Rate PredictionIndustrial 7 billion user interaction records (test)
AUC0.6956
11
Showing 10 of 18 rows

Other info

Follow for update