Adaptive Item-based Collaborative Structures via Noise Rescheduling in Diffusion for Generative Recommendation
Organizations: Fudan University, Shanghai, China · Tongji University, Shanghai, China
Abstract
Discrete Diffusion Models (DDMs) have recently been introduced to recommendation systems, modeling user history as a token generation process via iterative denoising. However, while effective at capturing user-level sequential patterns, these methods often fail to explicitly integrate item-based collaborative filtering information, a critical component for accurate recommendation. This deficiency manifests in two key aspects: (1) the item representation is often semantic-focused, lacking collaborative priors for diffusion training; and (2) the denoising process employs a uniform noise schedule, treating all tokens indiscriminately and ignoring item-level adaptive structural dependencies. To bridge this gap, we propose ANR-DiffRec, a unified framework designed to encode item-based collaborative structures into discrete diffusion for generative recommendation. First, we explicitly incorporate an item co-occurrence matrix to guide semantic ID generation, providing a structured collaborative prior for discrete diffusion training. Second, we introduce an item-based adaptive noise rescheduling mechanism that dynamically adjusts denoising weights according to both local contextual recoverability and behavior-aware item dependencies. Specifically, the proposed strategy jointly models intra-item structural context and inter-item collaborative signals, enabling structure-aware denoising during diffusion training. Extensive experiments on multiple benchmarks demonstrate that our method consistently outperforms state-of-the-art generative recommendation models. Code: https://github.com/CalmaQi/ANR-DiffRec.
Figures & tables
| Dataset | #Users | #Items | #Interactions | Sparsity | Avg.Len |
| Scientific | 50,985 | 25,848 | 412,947 | 99.969% | 8.10 |
| Instrument | 57,439 | 24,587 | 511,836 | 99.964% | 8.91 |
| Game | 94,762 | 25,612 | 814,586 | 99.966% | 8.60 |
| Steam | 39,795 | 9,265 | 2,949,605 | 99.200% | 74.12 |
| MovieLens | 6,040 | 3,883 | 1,001,456 | 95.730% | 165.57 |
| Category | Model | Scientific | Instrument | Game | ||||||||||||
| R@1 | R@5 | R@10 | N@5 | N@10 | R@1 | R@5 | R@10 | N@5 | N@10 | R@1 | R@5 | R@10 | N@5 | N@10 | ||
| Item ID based | GRU4Rec ( 13 ) | 0.0071 | 0.0184 | 0.0272 | 0.0128 | 0.0156 | 0.0094 | 0.0297 | 0.0453 | 0.0196 | 0.0246 | 0.0149 | 0.0461 | 0.0712 | 0.0307 | 0.0387 |
| SASRec ( 20 ) | 0.0063 | 0.0240 | 0.0379 | 0.0152 | 0.0197 | 0.0089 | 0.0331 | 0.0525 | 0.0211 | 0.0273 | 0.0128 | 0.0516 | 0.0823 | 0.0323 | 0.0421 | |
| BERT4Rec ( 37 ) | 0.0045 | 0.0157 | 0.0264 | 0.0100 | 0.0134 | 0.0065 | 0.0255 | 0.0412 | 0.0160 | 0.0211 | 0.0082 | 0.0315 | 0.0530 | 0.0199 | 0.0267 | |
| FMLP-Rec ( 54 ) | 0.0046 | 0.0181 | 0.0300 | 0.0113 | 0.0151 | 0.0086 | 0.0299 | 0.0496 | 0.0193 | 0.0257 | 0.0099 | 0.0395 | 0.0649 | 0.0246 | 0.0328 | |
| LRURec ( 50 ) | 0.0049 | 0.0169 | 0.0267 | 0.0110 | 0.0141 | 0.0071 | 0.0272 | 0.0431 | 0.0172 | 0.0223 | 0.0134 | 0.0480 | 0.0753 | 0.0308 | 0.0396 | |
| Dataset | Model | NDCG@5 | NDCG@10 | Recall@5 | Recall@10 |
| MovieLens | PreferGrow ( 18 ) | 0.0910 | 0.1177 | 0.1409 | 0.2237 |
| RPG ( 17 ) | 0.1197 | 0.1440 | 0.1750 | 0.2503 | |
| DiffGRM ( 28 ) | 0.1038 | 0.1260 | 0.1541 | 0.2235 | |
| LLaDA-Rec ( 35 ) | 0.1359 | 0.1628 | 0.1980 | 0.2820 | |
| ANR-DiffRec | 0.1468 | 0.1738 | 0.2141 | 0.2977 | |
| Steam | PreferGrow ( 18 ) | 0.0406 | 0.0508 | 0.0615 | 0.0935 |
| Dataset | Method | NDCG@1 | NDCG@5 | NDCG@10 | Recall@5 | Recall@10 |
| Scientific | RQ-VAE | 0.0120 | 0.0221 | 0.0269 | 0.0320 | 0.0487 |
| w/o II | 0.0109 | 0.0217 | 0.0268 | 0.0314 | 0.0479 | |
| w/o Schedule | 0.0099 | 0.0205 | 0.0259 | 0.0312 | 0.0477 | |
| w/o Schedule and II | 0.0097 | 0.0203 | 0.0253 | 0.0310 | 0.0474 | |
| Full | 0.0122 | 0.0226 | 0.0279 | 0.0327 | 0.0494 | |
| Instrument | RQ-VAE | 0.0128 | 0.0270 | 0.0341 | 0.0408 | 0.0626 |
| Stage | Method | Time/Epoch (s) | Total (h) | Cost |
| Item Tokenization | RQ-VAE | 0.648 | 1.799 | 1.000 |
| Ours | – | 0.106 | 0.059 | |
| Generative Training | LLaDA-Rec | 87.402 | 3.641 | 1.000 |
| ANR-DiffRec | 88.351 | 3.681 | 1.011 |
| Method | Metric | Beam=1 | Beam=10 | Beam=20 | Beam=30 | Beam=40 | Beam=50 |
| LLaDA-Rec | Total Time (s) | 0m 13s | 1m 46s | 3m 22s | 5m 04s | 6m 46s | 8m 33s |
| Time / User (ms) | 0.25 | 2.08 | 3.96 | 5.96 | 7.96 | 10.06 | |
| ANR-DiffRec | Total Time (s) | 0m 13s | 1m 46s | 3m 24s | 5m 06s | 6m 50s | 8m 39s |
| Time / User (ms) | 0.25 | 2.08 | 4.00 | 6.00 | 8.04 | 10.18 |
| Item | Metric Space | Co-occur Hits@20 | Co-occurring Neighbors (Post-integration) | ||
| Pre | Post | Gain ( ) | |||
| Item #710 | Embedding | 3 | 5 | +2 | {757, 732, 9894, 750, 1104 } |
| SIDs | 1 | 8 | +7 | { 732, 719, 750, 1104, 755, 758, 751 , 757} | |
| Item #6278 | Embedding | 5 | 9 | +4 | {9430, 3671, 23939, 2229, 3021, 17400, 24692, 21922 , 17350} |
| SIDs | 4 | 12 | +8 | {3671, 3717 , 879 , 2371, 2250, 2230, 1613, 2268, 1169 , 998, 2229, 1774 } | |
| Item #219 | Embedding | 4 | 8 | +4 | {975, 971 , 20665, 977 , 214, 13070, 9644, 6499 } |
Generative Recommendation
Generative recommendation models user histories and item catalogs to generate personalized items, lists, or queries rather than only scoring candidates.
See all 106 papers