CASE: Cadence-Aware Set Encoding for Large-Scale Next Basket Repurchase Recommendation
Organizations: Walmart Global Tech, Sunnyvale, CA, USA
Abstract
Repurchase behavior is a primary signal in large-scale retail recommendation, particularly in categories with frequent replenishment: many items in a user's next basket were previously purchased, and their timing follows stable, item-specific cadences. Yet most next basket repurchase recommendation models represent history as a sequence of discrete basket events indexed by visit order, which cannot explicitly model elapsed calendar time or update item rankings as days pass between purchases. We present CASE (Cadence-Aware Set Encoding) for next basket repurchase recommendation, which decouples item-level cadence learning from cross-item interaction, enabling explicit calendar-time modeling while remaining production-scalable. CASE represents each item's purchase history as a calendar-time signal over a fixed horizon, applies shared multi-scale temporal convolutions to capture recurring rhythms, and uses induced set attention to model cross-item dependencies with sub-quadratic complexity, allowing efficient batch inference at scale. Across three public benchmarks and a proprietary dataset, CASE consistently improves precision, recall, and NDCG at multiple cutoffs compared to strong next basket recommendation baselines. In a production-scale evaluation with tens of millions of users and a large item catalog, CASE achieves up to 8.6% relative precision lift and 9.9% relative recall lift at top-5, showing that scalable cadence-aware modeling yields measurable gains in both benchmark and industrial settings.
Figures & tables
| Datasets | Methods | ||||||||||||
|---|---|---|---|---|---|---|---|---|---|---|---|---|---|
| Prec | Rec | NDCG | Prec | Rec | NDCG | Prec | Rec | NDCG | Prec | Rec | NDCG | ||
| TaFeng | PersonalTop | 0.1771 | 0.0988 | 0.1771 | 0.1363 | 0.2100 | 0.2042 | 0.1104 | 0.2776 | 0.2263 | 0.0795 | 0.3839 | 0.2612 |
| TIFUKNN | 0.2146 | 0.1201 | 0.2146 | 0.1639 | 0.2646 | 0.2503 | 0.1332 | 0.3443 | 0.2769 | 0.0995 | 0.5007 | 0.3228 | |
| DNNTSP | 0.2387 | 0.1632 | 0.2387 | 0.1675 | 0.3196 | 0.2869 | 0.1406 | 0.4255 | 0.3275 | 0.1086 | 0.5809 | 0.3739 | |
| BERT4NBR | 0.2316 | 0.1527 | 0.2316 | 0.1662 | 0.3035 | 0.2783 | 0.1399 | 0.3986 | 0.3120 | 0.1079 | 0.5753 | 0.3638 | |
| PIETSP | 0.2507 | 0.1752 | 0.2507 | 0.1670 | 0.3165 | 0.2903 | 0.1421 | 0.4223 | 0.3317 | 0.1094 | 0.5853 | 0.3801 | |
| Model Components | ||||||||||||
|---|---|---|---|---|---|---|---|---|---|---|---|---|
| Prec | Rec | NDCG | Prec | Rec | NDCG | Prec | Rec | NDCG | Prec | Rec | NDCG | |
| CASE w/o CNN | 0.3333 | 0.0740 | 0.3333 | 0.2666 | 0.1793 | 0.3046 | 0.2382 | 0.2519 | 0.3065 | 0.2023 | 0.3784 | 0.3297 |
| CASE w/o Set Encoder | 0.5232 | 0.1263 | 0.5232 | 0.4485 | 0.2897 | 0.5038 | 0.3947 | 0.3915 | 0.5000 | 0.3122 | 0.5396 | 0.5152 |
| CASE w/o Item Embedding | 0.5390 | 0.1281 | 0.5390 | 0.4488 | 0.2848 | 0.5057 | 0.3919 | 0.3847 | 0.4985 | 0.3124 | 0.5360 | 0.5161 |
| CASE w/ PermEqMean | 0.5414 | 0.1326 | 0.5414 | 0.4538 | 0.2901 | 0.5115 | 0.3983 | 0.3917 | 0.5062 | 0.3147 | 0.5463 | 0.5225 |
| CASE (w/ ISAB) | 0.5478 | 0.1325 | 0.5478 | 0.4551 | 0.2901 | 0.5135 | 0.3989 | 0.3949 | 0.5086 | 0.3155 | 0.5464 | 0.5241 |
| k | Precision | Recall | NDCG |
|---|---|---|---|
| 5 | +8.63% | +9.90% | +10.46% |
| 10 | +6.78% | +7.95% | +9.40% |
| 20 | +5.27% | +6.32% | +8.75% |
Recommender Systems
Recommender systems predict and rank items for people using interaction histories, content, context, and feedback.
See all 514 papers