IIRG: Item–Item Relation Generation in GR
- IIRG is a multi-task training strategy that supplements next-item prediction with collaborative and semantic neighbor generation to capture richer item–item relations.
- It leverages auxiliary tasks to mitigate one-hop memorization and improve recommendations for cold-start and non-local target items.
- Empirical results on Amazon Reviews and Yelp show IIRG outperforms baselines by redistributing gains, with improvements of up to 21% in Recall@5.
IIRG, short for Item–Item Relation Generation, is a training strategy for LLM-based generative recommendation (GR) that augments standard next-item prediction with two auxiliary item-to-item generation tasks: collaborative neighbor generation and semantic neighbor generation. It was introduced to address a specific failure mode identified in LLM recommenders, namely one-hop memorization, in which models disproportionately recommend items that appeared as direct successors in training sequences. The central objective of IIRG is to teach LLMs to model richer item–item relations beyond one-hop transitions in user sequences, thereby improving recommendation quality, especially for users whose target items are not recoverable from memorized adjacent transitions (Kim et al., 15 Jun 2026).
1. Definition and research context
IIRG is situated in the literature on generative recommendation, where an LLM is fine-tuned to generate item identifiers rather than score candidates using a conventional retrieval model. In the formulation reported for IIRG, standard training uses next-item prediction: given a tokenized representation of a user sequence, the model is trained to generate the next item ID. The paper argues that this objective, by itself, encourages LLMs to rely too heavily on one-hop memorization, and that such reliance narrows the effective use of pretrained knowledge (Kim et al., 15 Jun 2026).
The motivating comparison is between LLM-based GR models and a non-LLM baseline, TIGER. The reported finding is that LLMs exhibit stronger one-hop memorization than TIGER, and that the gains of LLMs over TIGER are concentrated among users whose target items can already be reached via memorized one-hop transitions. This observation leads to the paper’s main hypothesis: improving recommendation for users not covered by one-hop memorization requires LLMs to learn richer item–item relations than direct successor patterns.
Within that framing, IIRG is not a decoding heuristic or post hoc reranker. It is a multi-task training strategy. The auxiliary supervision is intended to expose the model to two additional relational structures: multi-hop collaborative co-occurrence patterns in user sequences and semantic similarity derived from item text.
2. One-hop memorization and its operationalization
The paper defines one-hop memorization through the adjacent neighbor set for an item , denoted , which is the multiset of all items that immediately follow in any training user sequence (Kim et al., 15 Jun 2026). For a user , the one-hop-based candidate set is formed by aggregating adjacent neighbors over the items in the user’s sequence: where is multiset union. The top- most frequent items in are then taken, with
and ties broken by popularity.
This construction is used to quantify the fraction of a model’s top-5 predictions that fall in the one-hop candidate set. On that basis, the paper reports three central observations. First, LLM-based GR models consistently have a higher one-hop memorization ratio than TIGER. Second, this remains true for both TIDs and SIDs. Third, the effect persists even after applying regularization and scaling TIGER up to near-LLM size, which the paper uses to argue that neither model size alone nor overfitting alone explains the behavior (Kim et al., 15 Jun 2026).
The evaluation further partitions users into two groups: one-hop-memorization-benefiting users, whose target item lies in the one-hop candidate set, and non-benefiting users, whose target item does not. The reported pattern is that LLMs obtain much larger gains over TIGER on the former group than on the latter. This motivates IIRG’s design: the method is intended to supply recommendation signals that remain useful when direct train-time transitions do not cover the target item.
3. Relational supervision in IIRG
IIRG introduces two auxiliary generation targets for each item: collaborative neighbors and semantic neighbors (Kim et al., 15 Jun 2026).
Collaborative relations
Collaborative neighbors are defined from item co-occurrences across multiple hops in user sequences. The pairwise weight between items and 0 is
1
with
2
Here, 3 is the time-window length. The collaborative neighbor sequence 4 is then defined as the top-5 items sorted by descending 6.
The intended effect is to capture collaborative behavior beyond direct transitions: items that frequently appear near each other in sequences and may therefore be predictive even when they are not immediate successors. The paper explicitly notes that a variant excluding one-hop transitions from this collaborative set was tested, but that the version including them performed better overall.
Semantic relations
Semantic neighbors are built from textual similarity. For each item 7, the item text description 8 is encoded with a pretrained embedding model 9: 0 The semantic neighbor sequence 1 is formed by selecting the top-2 items under cosine similarity,
3
These neighbors are intended to encode similar themes, including semantically related products such as related LEGO products or compatible accessories. The paper characterizes them as signals for thematic similarity and for substitutable or complementary items.
A key empirical observation motivating the joint use of these two signals is that collaborative neighbors and semantic neighbors each cover a fraction of users comparable to one-hop memorization, while their overlap with one-hop neighbors is limited: the pairwise Jaccard similarity is reported to be below 0.5. The paper therefore treats them as predictive but non-redundant sources of supervision.
4. Objective function and multitask formulation
Under standard next-item prediction, the next-item loss for user 4 is written as
5
In this expression, 6 is the token-generation probability of the LLM, 7 is the tokenized sequence representation, 8 is the index of the final token for the anchor item 9, and 0 is the instruction prompt for next-item prediction (Kim et al., 15 Jun 2026).
IIRG adds two auxiliary losses.
For collaborative-neighbor generation, the loss for item 1 is
2
For semantic-neighbor generation, the loss for item 3 is
4
The final batch loss is
5
where 6 is batch size, and 7 weight the auxiliary tasks. The reported settings are 8 and 9, tuned by dataset and ID type (Kim et al., 15 Jun 2026).
Conceptually, the distinction from standard next-item prediction is exact: instead of learning only user history 0 next item, the model also learns item 1 collaborative neighbors and item 2 semantic neighbors. This suggests a shift from purely sequential imitation toward explicit relational modeling within the same generative backbone.
5. Experimental protocol and empirical findings
The main experiments are conducted on Amazon Reviews sequential recommendation datasets: Sports, Toys, and Beauty, with additional evaluation on Yelp (Kim et al., 15 Jun 2026). The reported dataset statistics are:
| Dataset | Users | Items | Interactions |
|---|---|---|---|
| Sports | 35,598 | 18,357 | 296,337 |
| Toys | 19,412 | 11,924 | 167,597 |
| Beauty | 22,363 | 12,101 | 198,502 |
| Yelp | 30,431 | 20,033 | 316,354 |
The evaluation protocol is standard leave-one-out: the last item is used for test, the second-to-last for validation, and the remainder for training. Evaluation is in the full-item ranking setting, with Recall@5, Recall@10, NDCG@5, and NDCG@10 as reported metrics.
The comparison covers 17 baselines in four groups: traditional recommender systems (LightGCN, SimGCL, SASRec, FDSA, S3-Rec), non-LLM-based GR models (TIGER, LETTER), GR methods with auxiliary tasks (P5, ReAT, LC-Rec, EAGER-LLM), and LLM-based GR models using SIDs or TIDs (PLUM, OneRec-Think, Naive, GRAM, GRLM, AgenticTagger). For the main results, the same LLM backbone is used across comparisons: Qwen-3.5-4B (Kim et al., 15 Jun 2026).
The principal empirical findings are fourfold.
First, IIRG is reported as the best-performing method on every dataset and metric in the main table. Relative to Naive, it improves by an average of 21% in Recall@5 across datasets. Example values reported in the paper are:
- Sports: Naive-TID 4, IIRG 5
- Toys: Naive-TID 6, IIRG 7
- Beauty: Naive-TID 8, IIRG 9
Second, IIRG reduces the one-hop memorization ratio of Naive and, more importantly, yields larger gains for non-one-hop-memorization-benefiting users than for benefiting users. The reported average gain is 40% for users who do not benefit from one-hop memorization, versus 16% for users who do benefit. This is the paper’s main evidence that the method succeeds in shifting the model toward non-local recommendation signals (Kim et al., 15 Jun 2026).
Third, the method is reported to generalize across identifier schemes. Under SIDs, IIRG still improves over Naive and outperforms LC-Rec, which the paper identifies as the strongest SID-applicable baseline. The paper further reports that both the memorization findings and IIRG’s effectiveness hold under Residual K-Means SIDs as well.
Fourth, IIRG shows larger gains for cold-start target items, defined in the paper as target items with 0 interactions. The reported interpretation is that the semantic-neighbor task is particularly useful for sparse items. The Yelp results are used to argue that the method is not limited to Amazon Reviews.
6. Ablations, limitations, and significance
The ablation study supports the necessity of both auxiliary tasks (Kim et al., 15 Jun 2026). Removing collaborative neighbor generation hurts performance, and removing semantic neighbor generation also hurts performance; using both gives the best results. A random-neighbor variant performs worse than IIRG and can even degrade Naive, which the paper takes as evidence that the auxiliary neighbors must be predictive, not arbitrary. The paper also evaluates heuristics that directly recommend from collaborative or semantic neighbors; these are worse than IIRG, indicating that it is more effective to train the LLM to learn the relations than to use the relations directly as recommendations. The alternative design that excludes one-hop successors from collaborative neighbors performs worse than the current design, suggesting that completely removing one-hop signals may overcorrect and discard useful predictive information.
The paper states two principal limitations. The first is that the cause of one-hop memorization remains unknown. The behavior is documented, but the underlying mechanism is not explained, and the experiments suggest that model size alone does not account for it. The second is that the experiments are limited to the Qwen-3.5 family; testing whether the same memorization pattern and the same effectiveness of IIRG hold for other LLM families such as Llama or Mistral is identified as important future work. An additional caveat reported in the appendix is that LoRA fine-tuning performed poorly in this setting and produced many invalid outputs, which suggests that sufficient full-model capacity may be required for reliable generative recommendation (Kim et al., 15 Jun 2026).
In the generative-recommendation literature, IIRG is therefore best understood as a targeted response to a specific empirical diagnosis: training LLM recommenders only with next-item prediction encourages shortcut-like one-hop memorization. IIRG retains next-item prediction but supplements it with explicit supervision over non-local collaborative structure and semantic item similarity. A plausible implication is that the method’s main contribution is not only higher aggregate ranking performance, but also a redistribution of performance gains toward users and items that are poorly served by direct transition memory alone.