GOD: Enhancing Generalization via Deep Grafting for Sequential Recommendation
Organizations: Pohang University of Science and Technology Pohang, Republic of Korea
Abstract
Sequential recommenders often struggle with sparse and noisy histories, limiting generalization to unseen interactions. Knowledge distillation mitigates this by transferring dense supervision from a teacher to a student. However, most distillation methods run teacher and student independently, then match student outputs or representations to the teacher. Such supervision entangles student-component effects, blurring whether weak generalization stems from unreliable embeddings, overfitted encoding, or co-adaptation to sparse histories. In this paper, we propose Graft-Oriented Distillation (GOD), a component-level distillation framework for improved generalization through grafting. Grafting denotes replacing selected frozen-teacher components with trainable student counterparts to build hybrid source models. GOD uses these hybrid models to evaluate student embeddings with the teacher encoder and the student encoder with teacher embeddings, providing component-level feedback. At inference, GOD uses only the student, incurring no additional cost. Across three real-world datasets, GOD outperforms state-of-the-art baselines by up to 13.92%.
Figures & tables
| Datasets | # Interactions | # Users | # Items | Avg.SeqLen | Sparsity |
|---|---|---|---|---|---|
| Amazon Beauty | 198,502 | 22,363 | 12,101 | 8.9 | 99.93 % |
| Yelp | 345,186 | 29,915 | 21,572 | 11.5 | 99.95 % |
| MovieLens 1M | 999,611 | 6,040 | 3,416 | 165.5 | 95.16 % |
| Dataset | Backbone | Metric | Teacher | Student | RD | CD | DE | RRD | HTD | BD | AdaRec | MSKDIK | EMKD | GOD | ||
|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|
| Amazon Beauty | GRU4Rec | HR@10 | 0.0506 | 0.0384 | 0.0392 | 0.0403 | 0.0413 | 0.0426 | 0.0438 | 0.0464 | 0.0450 | 0.0454 | 0.0472 | 0.0508* | 32.29 % | 7.63 % |
| HR@20 | 0.0788 | 0.0622 | 0.0626 | 0.0639 | 0.0652 | 0.0668 | 0.0682 | 0.0711 | 0.0692 | 0.0703 | 0.0722 | 0.0777* | 24.92 % | 7.62 % | ||
| NDCG@10 | 0.0257 | 0.0187 | 0.0190 | 0.0196 | 0.0203 | 0.0210 | 0.0217 | 0.0232 | 0.0224 | 0.0227 | 0.0237 | 0.0257* | 37.69 % | 8.44 % | ||
| NDCG@20 | 0.0322 | 0.0249 | 0.0262 | 0.0266 | 0.0270 | 0.0275 | 0.0281 | 0.0290 | 0.0285 | 0.0286 | 0.0294 | 0.0316* | 26.91 % | 7.48 % | ||
| FMLPRec | HR@10 | 0.0667 | 0.0524 | 0.0540 | 0.0547 | 0.0555 | 0.0564 | 0.0573 | 0.0583 | 0.0583 | 0.0590 | 0.0596 | 0.0634* | 20.99 % | 6.38 % | |
| HR@20 | 0.0959 | 0.0813 | 0.0843 | 0.0849 | 0.0856 | 0.0864 | 0.0871 | 0.0883 | 0.0873 | 0.0886 | 0.0892 | 0.0951* | 16.97 % | 6.61 % |
| Dataset | Metric | SASRec | RRD | HTD | BD | MSKDIK | EMKD | CL4SRec | MCLRec | CT4Rec | DuoRec | RCL | GOD | |
|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|
| Amazon Beauty | HR@10 | 0.0727 | 0.0718 | 0.0721 | 0.0749 | 0.0763 | 0.0785 | 0.0796 | 0.0806 | 0.0837 | 0.0856 | 0.0853 | 0.0923* | 7.83 % |
| HR@20 | 0.1030 | 0.1010 | 0.1018 | 0.1047 | 0.1062 | 0.1078 | 0.1084 | 0.1099 | 0.1139 | 0.1164 | 0.1172 | 0.1249* | 6.57 % | |
| NDCG@10 | 0.0401 | 0.0396 | 0.0400 | 0.0426 | 0.0438 | 0.0454 | 0.0464 | 0.0474 | 0.0490 | 0.0506 | 0.0517 | 0.0559* | 8.12 % | |
| NDCG@20 | 0.0472 | 0.0461 | 0.0469 | 0.0499 | 0.0513 | 0.0525 | 0.0544 | 0.0554 | 0.0568 | 0.0580 | 0.0596 | 0.0641* | 7.55 % | |
| Yelp | HR@10 | 0.0568 | 0.0559 | 0.0571 | 0.0574 | 0.0566 | 0.0584 | 0.0588 | 0.0591 | 0.0601 | 0.0606 | 0.0607 | 0.0640* | 5.44 % |
| HR@20 | 0.0917 | 0.0903 | 0.0931 | 0.0934 | 0.0917 | 0.0945 | 0.0951 | 0.0957 | 0.0968 | 0.0975 | 0.0978 | 0.1032* | 5.52 % |
| Case | (T,T) | (S,T) | (T,S) | Amazon Beauty | Yelp | MovieLens 1M |
|---|---|---|---|---|---|---|
| (1) | 0.0474 | 0.0405 | 0.1218 | |||
| (2) | 0.0451 | 0.0375 | 0.1141 | |||
| (3) | 0.0448 | 0.0376 | 0.1135 | |||
| (4) | 0.0410 | 0.0342 | 0.1084 | |||
| (5) | 0.0421 | 0.0349 | 0.1111 | |||
| (6) | 0.0401 | 0.0336 | 0.1074 |
| Variant | Amazon Beauty | Yelp | MovieLens 1M |
|---|---|---|---|
| GE | 0.0474 | 0.0405 | 0.1218 |
| T S only | 0.0466 | 0.0394 | 0.1184 |
| S T only | 0.0459 | 0.0386 | 0.1166 |
| GE (half length) | 0.0462 | 0.0396 | 0.1152 |
| w/o GE | 0.0452 | 0.0379 | 0.1126 |
| Variant | Amazon Beauty | Yelp | MovieLens 1M |
|---|---|---|---|
| GCL | 0.0474 | 0.0405 | 0.1218 |
| Source-Target only | 0.0460 | 0.0386 | 0.1111 |
| Unweighted | 0.0455 | 0.0379 | 0.1090 |
| Learnable weight | 0.0427 | 0.0359 | 0.1074 |
| Model | Amazon Beauty | MovieLens 1M | ||||||
|---|---|---|---|---|---|---|---|---|
| Mem. | Time/Epoch | Train | NDCG@20 | Mem. | Time/Epoch | Train | NDCG@20 | |
| Teacher | 0.7 GB | 8.35 s | 6.12 m | 0.0472 | 1.1 GB | 15.88 s | 23.82 m | 0.1417 |
| Student | 0.6 GB | 7.75 s | 24.03 m | 0.0390 | 0.8 GB | 15.13 s | 18.66 m | 0.1052 |
| BD | 1.0 GB | 12.98 s | 27.69 m | 0.0444 | 1.5 GB | 26.67 s | 59.12 m | 0.1094 |
| EMKD | 1.1 GB | 15.57 s | 34.77 m | 0.0447 | 1.6 GB | 32.90 s | 51.00 m | 0.1098 |
| GOD | 0.9 GB | 10.06 s | 15.26 m | 0.0474 | 1.2 GB | 21.00 s | 17.85 m | 0.1218 |
Sequential Recommendation
Sequential recommendation predicts what a user will choose next from ordered interactions, adapting to changing interests and context.
See all 138 papers