Intention-Modulated Personalized ACT-R — a hybrid sequential/cognitive-memory recommender system for music that beats SASRec (established baseline) by 114% NDCG@10 on the 4.8B-interaction Yambda-5B dataset. The full pipeline — dataset download, model training and system evaluation — runs within 16GB of RAM with no GPU (and no OOM), though SASRec was trained on a GPU for speed.
Built for master's thesis at Wrocław University of Science and Technology.
Sequential recommender systems, such as SASRec [1], model what a user wants right now, within a short window of recent sessions. However, music listening is dominated by repetition — people return to their favourite tracks for months or even years — and this long-term signal is completely outside the scope of a short attention window. IMPACT adds it back in by using a personalised popularity signal derived from ACT-R [2], a cognitive architecture that models how human declarative memory strengthens and weakens with use over time. This signal is weighted by whether each listen was actively chosen by the user or just passively served by the streaming platform. The two models operate independently and are combined at inference time, so the final "blend" can be adjusted/retuned without retraining SASRec (ACT-R is purely deterministic either way).
Experiments showed that ACT-R achieves the highest ranking accuracy (NDCG@10), but can only recommend tracks already present in the user's listening history — its global exploration score is exactly 0.000 by definition. SASRec can recommend novel, previously unheard tracks, but at substantially lower ranking accuracy. IMPACT is "the best of both worlds", and on the full-scale Yambda-5B dataset (4.8 billion interactions) [3] improves NDCG@10 by +114% relative to SASRec, while still being able to recommend tracks that the user has never heard before.
The two components — SASRec and ACT-R — are fully decoupled and independent of each other. Fusion happens only at inference time, via a hyperparameter alpha.
Three configurations of IMPACT were evaluated on the held-out Yambda-5B test set: pure ACT-R, the optimal (empirically found) hybrid and pure SASRec.
| Metric | ACT-R (alpha=0) | IMPACT (alpha=0.4) | SASRec (alpha=1) |
|---|---|---|---|
| NDCG@10 | 0.0894 | 0.0650 | 0.0304 |
| Recall@10 | 0.0805 | 0.0624 | 0.0258 |
The results presented above, which use only overall metrics, tell a bit misleading story: ACT-R performs better than the hybrid with SASRec on both NDCG@10 and Recall@10, which might suggest that the SASRec component is unnecessary. To understand why ACT-R is superior and whether that superiority is desirable, the NDCG@10 metric is analysed along two dimensions — whether the recommended track is one the user already knows (repetition) or genuinely new to them (exploration), and whether the "knownness" is judged against the user's recent session window (local) or their entire listening history (global):
| NDCG@10, decomposed | ACT-R (alpha=0) | IMPACT (alpha=0.4) | SASRec (alpha=1) |
|---|---|---|---|
| Repetition, local | 0.1445 | 0.1024 | 0.0444 |
| Exploration, local | 0.0138 | 0.0098 | 0.0070 |
| Repetition, global | 0.1004 | 0.0720 | 0.0328 |
| Exploration, global | 0.0000 | 0.0022 | 0.0035 |
The NDCG@10 metric decomposition reveals the trade-off introduced earlier. ACT-R's accuracy advantage is entirely driven by repetition — it performs well because music consumption is repetitive by nature, and ACT-R is specifically designed to model such behaviour. However, the downside is that its global exploration score is 0.000, meaning it cannot recommend a track outside the user's listening history (which leads to a filtering bubble). SASRec is at the other extreme — it can recommend genuinely new content, but with only a fraction of ACT-R's accuracy. IMPACT bridges this gap by retaining most of ACT-R's repetition-driven accuracy while maintaining a non-zero exploration ability — the outcome the hybrid was designed to produce.
Selecting alpha. The three configurations evaluated above use
The plot makes the tradeoff clear: overall accuracy and repetition-based metrics decline monotonically as alpha increases, while global exploration is exactly 0.000 for small alpha values and only becomes non-zero once SASRec contributes enough weight to the fusion.
Statistical significance. To test whether the improvement is statistically significant, a Wilcoxon signed-rank test for paired data was applied to per-session NDCG@10 differences
-
$H_0$ : The median of$d_i$ is zero — IMPACT does not outperform SASRec. -
$H_1$ : The median of$d_i$ is positive — IMPACT systematically achieves a higher NDCG@10 than SASRec.
The test rejected
Details that are worth stating explicitly:
- Effect size scope: an effect size of r = 0.771 describes the strength and consistency of the difference among the 5,960 pairs (26.8% of the test set) in which the two systems actually differed. For the remaining 73.2%, both models returned identical scores — predominantly NDCG@10 = 0 for both, which is typical in sparse recommendation scenarios. The effect is large where it exists, but not uniformly large across the full test set.
-
Direction: of the 5,960 non-zero pairs, IMPACT outperformed SASRec in 5,182 cases (86.9%) and underperformed in 778 (13.1%) — a ratio of approximately 6.7:1 in favour of IMPACT, which is consistent with the one-sided hypothesis
$H_1$ . This could be understood like the following: where introducing the personalised popularity signal changes the prediction at all, that change is nearly seven times more likely to help than to hurt.
The accompanying master's thesis is undergoing formal review and defence at Wrocław University of Science and Technology, and cannot be shared publicly yet.
This repository serves as the primary citable artifact. It contains the complete implementation of IMPACT, together with the experimental pipelines, diagrams, and results required to reproduce the findings reported in the thesis.
If you use this software, please cite it using the entry provided in CITATION.cff.
[1] Wang-Cheng Kang and Julian McAuley, 2018. Self-Attentive Sequential Recommendation. DOI: 10.48550/arXiv.1808.09781
[2] John R. Anderson et al., 2004. Psychological Review 111.4. DOI: 10.1037/0033-295X.111.4.1036
[3] Alexander Ploshkin et al., 2025. Yambda-5B — A Large-Scale Multi-Modal Dataset for Ranking and Retrieval. DOI: 10.1145/3705328.3748163
MIT — see LICENSE.