Source-linked AI summary

Sequential Recommendation with Self-Attentive Multi-Adversarial Network

Ruiyang Ren, Zhaoyang Liu, Yaliang Li, Wayne Xin Zhao, Hui Wang, Bolin Ding, Ji-Rong Wen

arXiv:2005.10602v1cs.IRcs.LG

TL;DR

Sequential recommendation must capture evolving user behavior, while the effects of individual context factors are difficult to analyze. MFGAN separates sequence prediction from factor-specific evaluation, and experiments on three real-world datasets report better performance than competitive baselines, with improved stability and interpretability.

  • Problem

    Sequential recommenders need to capture evolving user behavior, while the effects of individual context factors are difficult to analyze.

    Source: §1 · pp. 1–2
  • Method

    MFGAN uses a Transformer-based generator to predict next items and multiple factor-specific discriminators to evaluate generated recommendations.

    Source: §1 · p. 2
  • Results

    Across three real-world datasets, MFGAN outperforms competitive baselines; multiple discriminators help stabilize adversarial training and enhance interpretability.

    Source: §1 · p. 2, §6 · p. 10
  • Takeaways & Limitations

    Factor-specific discriminator rewards can help characterize how different factors contribute to recommendations over time.

    Source: §1 · p. 2, §3.5 · p. 6

    The current setting uses separately designed discriminators; future work will investigate sharing discrimination information and incorporating explicit user preference.

    Source: §6 · p. 10

Abstract

from arXiv · show

Recently, deep learning has made significant progress in the task of sequential recommendation. Existing neural sequential recommenders typically adopt a generative way trained with Maximum Likelihood Estimation (MLE). When context information (called factor) is involved, it is difficult to analyze when and how each individual factor would affect the final recommendation performance. For this purpose, we take a new perspective and introduce adversarial learning to sequential recommendation. In this paper, we present a Multi-Factor Generative Adversarial Network (MFGAN) for explicitly modeling the effect of context information on sequential recommendation. Specifically, our proposed MFGAN has two kinds of modules: a Transformer-based generator taking user behavior sequences as input to recommend the possible next items, and multiple factor-specific discriminators to evaluate the generated sub-sequence from the perspectives of different factors. To learn the parameters, we adopt the classic policy gradient method, and utilize the reward signal of discriminators for guiding the learning of the generator. Our framework is flexible to incorporate multiple kinds of factor information, and is able to trace how each factor contributes to the recommendation decision over time. Extensive experiments conducted on three real-world datasets demonstrate the superiority of our proposed model over the state-of-the-art methods, in terms of effectiveness and interpretability.

1 INTRODUCTION

MFGAN addresses the difficulty of identifying how individual context factors affect sequential recommendations by separating factor evaluation from sequence generation. It uses adversarial training, and experiments report benefits in recommendation effectiveness and interpretability.

  • MLE-based recommenders combine context factors with sequence representations or a black-box module, obscuring when and how each factor affects recommendations.Source: §1 · p. 1
  • MFGAN uses a Transformer-based generator for next-item prediction and factor-specific discriminators to evaluate generated recommendations.Discriminator rewards guide generator learning through policy-gradient training.Source: §1 · p. 2
  • The proposed multi-discriminator structure decouples factors and is designed to improve sequential recommendation performance.The paper also analyzes the architecture's effectiveness and stability.Source: §1 · p. 2
  • Experiments on three real-world datasets show MFGAN benefits recommendation effectiveness and interpretability over state-of-the-art methods.Source: §1 · p. 2

2 PROBLEM DEFINITION

The task is to predict a user's next item from their interaction history and item context. Items may have multiple kinds of contextual information, called factors.

  • Each item is associated with m kinds of contextual information, corresponding to m factors.The paper gives artist, album, and popularity as examples in music recommendation.Source: §2 · p. 2
  • Given a user's chronologically ordered item interactions and item context, the task is to predict the item they will interact with next.Source: §2 · p. 2

3 METHODOLOGY

The methodology section outlines MFGAN's framework, its generator and discriminators, and its training process.

  • The methodology section introduces the MFGAN framework and its generator and discriminators, then discusses their design and training process.Source: §3 · p. 2

3.1 Multi-Factor Generative Adversarial Network Framework

MFGAN separates sequential next-item prediction from factor-based evaluation: a generator predicts items, while multiple discriminators evaluate generated sequences. Discriminator feedback guides training, and separating factors is presented as more flexible and interpretable.

3.1.1 Basic Components.

  • The generator predicts next items from the historical sequence, while multiple discriminators judge generated sequences using their respective factors.The generator does not use item-side context; examples of discriminator factors include category, popularity, artist, and album.Source: §3.1.1 · p. 2, Figure 1 · p. 3

3.1.2 Overall Procedure.

  • At each step, factor-specific discriminators evaluate the generated sequence and return feedback to guide the generator's next learning round.Discriminators are updated using generated sequences and actual user behavior sequences.Source: §3.1.2 · pp. 2–3
  • Separate discriminators let the framework incorporate factors flexibly, keep the generator focused on sequence generation, and distinguish each factor's effect for interpretation.The paper also says this separation can make the model more resistant to useless or noisy context information.Source: §3.1.3 · p. 3

3.2 The Generator Component

MFGAN’s generator turns user interaction histories into next-item recommendations using embedded sequences and stacked self-attention blocks. It masks future positions and converts the final representation into preferences over the item set.

3.2 The Generator Component

  • The generator stacks an embedding layer, self-attention blocks, and a prediction layer to generate target items.Its input is a sequence of historical interactions.Source: §3.2 · p. 3

3.2.1 Embedding Layer.

  • Historical item embeddings are combined with learnable position encodings to form the generator’s input representations.The item embedding matrix maps items to dense vectors, which are summed with position encodings.Source: §3.2.1 · p. 3

3.2.2 Self-attention Block.

  • Multi-head self-attention extracts sequence information from different representation subspaces, with future-position connections masked out.A point-wise feed-forward network adds non-linearity to the self-attention block.Source: §3.2.2 · pp. 3–4

3.2.3 Prediction Layer.

  • The final generator layer applies softmax to calculate user preferences over the item set.The prediction uses the maintained item embedding matrix.Source: §3.2.3 · p. 4

3.3 Factor-specific Discriminator Components

MFGAN uses a separate discriminator for each factor to judge generated recommendation sequences. These discriminators use factor-specific embeddings and bidirectional attention to produce rationality scores that guide the generator.

3.3 Factor-specific Discriminator Components

  • Each factor has an independent discriminator that classifies generated versus actual recommendation sequences as rational or not.The framework decouples context information from the generator by using factor-specific discriminators.Source: §3.3 · p. 4
  • Each discriminator embeds discretized factor values with bin-specific vectors and adds positional encodings.The factor values are mapped to bins, and their embeddings are concatenated by input sequence.Source: §3.3 · p. 4

3.3.2 Architecture.

  • A discriminator uses one self-attention block to avoid becoming too strong to provide suitable feedback to the generator.Its architecture is intentionally relatively weak compared with the generator.Source: §3.3.2 · p. 4
  • Unlike the generator, a discriminator can attend to subsequent positions, modeling interactions between any two positions in the sequence.The discriminator removes the attention mask and uses overall sequential characteristics for its judgment.Source: §3.3.2 · p. 4
  • An MLP turns each discriminator’s attention output into a rationality score, and the scores provide supervision signals for the generator.A rationality score reflects the discriminator’s estimate that a sequence comes from the actual data distribution.Source: §3.3.2 · p. 4

3.4 Multi-adversarial Training

MFGAN casts next-item generation as a reinforcement-learning problem because item sampling is discrete. It combines factor-discriminator rewards to guide generator policy updates, alternating generator and discriminator training.

3.4 Multi-adversarial Training

  • Discrete item sampling prevents direct gradient descent on the original GAN formulation, so recommendation is formalized as reinforcement learning.The state is the previously generated subsequence, the action is the next item, and the generator defines the policy.Source: §3.4.1 · p. 4

3.4.1 RL-based Formalization.

  • The expected return combines discriminator rationality scores using λ-parameterized softmax weights.λ is a tuning parameter, and discriminator scores provide guidance from different factor perspectives.Source: §3.4.1 · p. 5
  • The generator maximizes expected reward with the classic policy-gradient method.Its objective is to maximize the expected reward at each step.Source: §3.4.2 · p. 5

3.4.2 Learning Algorithm.

  • Training pretrains the generator and discriminators, then updates the generator using discriminator rewards and policy gradients.For each discriminator update, actual sequences serve as positives and generated sequences as negatives.Source: §3.4.2 · p. 5

3.5 Discussion and Analysis

The analysis compares ways of combining discriminator rewards and explains why multiple discriminators can stabilize generator learning. Their reward values also provide a view of factor importance over time.

3.5 Discussion and Analysis

  • A λ-parameterized softmax combines discriminator rewards as minimum, maximum, mean, or a soft combination, depending on λ.The minimum and maximum arise as λ approaches −∞ and +∞; λ = 0 gives the mean.Source: §3.5 · p. 5
  • Minimum rewards can indicate where to decrease generator probability without identifying where to increase it, while maximum rewards can let a prominent factor dominate and leave others undertrained.The maximum strategy alleviates the stated training issue with the minimum strategy.Source: §3.5 · pp. 5–6
  • Multiple discriminators make generator learning more stable; the authors adopt mean rewards because the last two reward cases show no significant difference.The generator's Q-function is zero only when all discriminators give zero reward.Source: §3.5 · p. 6
  • Discriminator rewards can be treated as the importance of influencing factors at each time step, and the framework leaves room for more advanced generator or discriminator designs.The approach decouples factors from the prediction component.Source: §3.5 · p. 6

4 EXPERIMENTS

MFGAN is evaluated on three recommendation datasets, with comparisons, factor and model ablations, convergence analysis, and illustrative cases. The reported results favor MFGAN over competitive baselines and show how factor signals vary across a sequence.

4.1 Dataset Construction

  • The experiments use MovieLens-1M, a 10,000-user Yahoo! subset, and Steam, with user sequences sorted by time and filtered using dataset-specific k-core thresholds.The thresholds are k = 5, 10, and 5, respectively; the final item is test data, the preceding item validation data, and earlier interactions training data.Source: §4.1 · p. 6

4.2 Experimental Settings

  • Evaluation uses MRR, NDCG@10, and HR@10, ranking each test positive against 100 sampled unobserved items and averaging scores over test users.MFGAN is implemented with two generator self-attention blocks and one discriminator block, with separate pretraining followed by alternating adversarial training.Source: §4.2.1 · p. 6, §4.2.2 · p. 6

4.3 Results and Analysis

  • MFGAN consistently outperforms the competitive baselines in sequential recommendation, including when its generator uses no context directly at test time.The comparison is reported in Table 2; the authors attribute the improvement over the self-attentive generator to multi-adversarial training guided by factor information.Source: Table 2 · p. 7, §4.3 · p. 7

4.2.3 Comparison Methods.

  • Among sequential baselines, SASRec performs best, while GRUF improves on GRU with context; MFGAN is compared with SASRec and other competitive methods.The authors omit other context-aware baselines because they found the Transformer architecture superior to other sequence neural networks for this task.Source: §4.2.3 · pp. 6–7, §4.3 · p. 7

4.4.1 Effect of Different Factors.

  • Removing factors generally hurts MFGAN, except developer on Steam; knowledge graph, artist, and popularity are the most useful factors on MovieLens-1M, Yahoo!, and Steam, respectively.The factor-removal experiments report NDCG@10 while retaining the other factor information.Source: §4.4.1 · p. 8, Figure 2 · p. 8

4.4.2 Comparisons of Model Variants.

  • The Uni-directional, SDSF, and SDAF variants all underperform complete MFGAN, supporting the value of bidirectional discriminators and multiple-discriminator adversarial training.The reported numerical improvement from bidirectionality appears small; the comparison is on MovieLens-1M.Source: §4.4.2 · p. 8, Table 3 · p. 8
  • About 2,000 epochs: three- and four-discriminator losses stabilize, while the other configurations remain unstable after 4,000 epochs.The authors also observe larger loss with more discriminators and speculate that this may act like regularization against overfitting.Source: §4.4.2 · p. 8, Figure 3 · p. 8

4.5 Qualitative Analysis on the Recommendation Interpretability

  • In the MovieLens case, popularity is prominent early and category at selected later steps; in the Yahoo! case, artist signals become dominant when a previously heard artist recurs.The examples illustrate factor-specific reward signals over time; the authors contrast this decoupling with methods that combine context into unified representations.Source: §4.5 · p. 9, Figure 4 · p. 9

5 RELATED WORK

The related work covers Markov-chain, recurrent, and self-attention approaches to sequential recommendation, alongside GAN applications in recommendation. MFGAN is positioned as applying adversarial training to sequential recommendation to characterize factor effects over time.

Sequential Recommendation

  • Prior sequential recommenders include Markov-chain methods, recurrent networks and variants, context-aware approaches, and self-attention methods such as SASRec.The cited examples include matrix factorization with first-order Markov chains, GRU recommendation [10], and self-attention recommendation [14].Source: §5 · p. 9

Generative Adversarial Network

  • GAN work spans framework and training variants as well as recommendation applications, including information retrieval and adversarial approaches to recommendation diversity.The passage names IRGAN [28], Chae et al., and personalized determinantal point process.Source: §5 · p. 9
  • The paper presents its work as the first application of adversarial training to sequential recommenders, using a multi-adversarial network to trace individual factor effects over time.The stated aim is improved recommendation interpretability through explicit characterization of factor effects.Source: §5 · p. 9

6 CONCLUSIONS

The conclusion presents MFGAN as a sequential recommender with a sequence-based generator and factor-specific discriminators. Experiments report stronger baseline performance, training stabilization, and improved interpretability; the discriminator-sharing design remains future work.

6 CONCLUSIONS

  • MFGAN uses a user-sequence generator to propose next items and multiple factor-specific discriminators to evaluate generated subsequences.Experiments on three real-world datasets are reported to outperform several competitive baselines.Source: §6 · p. 10
  • Multiple discriminators are reported as useful for stabilizing adversarial training and enhancing recommendation interpretability.The conclusion identifies separate discriminator design as the current simple setting and proposes principled sharing of discrimination information as future work.Source: §6 · p. 10
Loading 2005.10602v1…