Papers
Topics
Authors
Recent
Search
2000 character limit reached

IIRG: Item–Item Relation Generation in GR

Updated 4 July 2026
  • 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 iki_k, denoted N(ik)\mathcal{N}(i_k), which is the multiset of all items that immediately follow iki_k in any training user sequence (Kim et al., 15 Jun 2026). For a user uu, the one-hop-based candidate set is formed by aggregating adjacent neighbors over the items in the user’s sequence: Mu=iksuN(ik),\mathcal{M}_u = \uplus_{i_k \in s_u} \mathcal{N}(i_k), where \uplus is multiset union. The top-KK most frequent items in Mu\mathcal{M}_u are then taken, with

K=min(50,iksuN(ik)),K = \min\left(50,\left|\bigcup_{i_k\in s_u}\mathcal{N}(i_k)\right|\right),

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 iki_k and N(ik)\mathcal{N}(i_k)0 is

N(ik)\mathcal{N}(i_k)1

with

N(ik)\mathcal{N}(i_k)2

Here, N(ik)\mathcal{N}(i_k)3 is the time-window length. The collaborative neighbor sequence N(ik)\mathcal{N}(i_k)4 is then defined as the top-N(ik)\mathcal{N}(i_k)5 items sorted by descending N(ik)\mathcal{N}(i_k)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 N(ik)\mathcal{N}(i_k)7, the item text description N(ik)\mathcal{N}(i_k)8 is encoded with a pretrained embedding model N(ik)\mathcal{N}(i_k)9: iki_k0 The semantic neighbor sequence iki_k1 is formed by selecting the top-iki_k2 items under cosine similarity,

iki_k3

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 iki_k4 is written as

iki_k5

In this expression, iki_k6 is the token-generation probability of the LLM, iki_k7 is the tokenized sequence representation, iki_k8 is the index of the final token for the anchor item iki_k9, and uu0 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 uu1 is

uu2

For semantic-neighbor generation, the loss for item uu3 is

uu4

The final batch loss is

uu5

where uu6 is batch size, and uu7 weight the auxiliary tasks. The reported settings are uu8 and uu9, 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 Mu=iksuN(ik),\mathcal{M}_u = \uplus_{i_k \in s_u} \mathcal{N}(i_k),0 next item, the model also learns item Mu=iksuN(ik),\mathcal{M}_u = \uplus_{i_k \in s_u} \mathcal{N}(i_k),1 collaborative neighbors and item Mu=iksuN(ik),\mathcal{M}_u = \uplus_{i_k \in s_u} \mathcal{N}(i_k),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, SMu=iksuN(ik),\mathcal{M}_u = \uplus_{i_k \in s_u} \mathcal{N}(i_k),3-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 Mu=iksuN(ik),\mathcal{M}_u = \uplus_{i_k \in s_u} \mathcal{N}(i_k),4, IIRG Mu=iksuN(ik),\mathcal{M}_u = \uplus_{i_k \in s_u} \mathcal{N}(i_k),5
  • Toys: Naive-TID Mu=iksuN(ik),\mathcal{M}_u = \uplus_{i_k \in s_u} \mathcal{N}(i_k),6, IIRG Mu=iksuN(ik),\mathcal{M}_u = \uplus_{i_k \in s_u} \mathcal{N}(i_k),7
  • Beauty: Naive-TID Mu=iksuN(ik),\mathcal{M}_u = \uplus_{i_k \in s_u} \mathcal{N}(i_k),8, IIRG Mu=iksuN(ik),\mathcal{M}_u = \uplus_{i_k \in s_u} \mathcal{N}(i_k),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 \uplus0 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.

Definition Search Book Streamline Icon: https://streamlinehq.com
References (1)

Topic to Video (Beta)

No one has generated a video about this topic yet.

Whiteboard

No one has generated a whiteboard explanation for this topic yet.

Follow Topic

Get notified by email when new papers are published related to IIRG.