atlas

Research map / Large language models / Alignment and preferences

On Softmax Direct Preference Optimization for Recommendation

NeurIPS 2024 · Large language models › Alignment and preferences · approach: Active and token-level preference learning · 23 citations (OpenAlex)

Read the paperCheck your idea against 108,967 papers

AI-extracted summary. Written by AI from the paper's full text. Numbers and claims can be wrong or missing context, so check the paper before you cite or build on it.

Proposes S-DPO, a softmax-based extension of DPO that uses multiple sampled negatives to train LM-based recommenders for personalized ranking, outperforming SFT-based and other LM-based recommenders on three datasets.

Problem

LM-based recommenders are fine-tuned with a language modeling loss on positive items only, which ignores negatives and is not optimized for personalized ranking.

Approach

Extend DPO from pairwise Bradley-Terry to a Plackett-Luce-style partial-ranking loss over one positive and multiple sampled negatives, which connects to a softmax (InfoNCE-like) ranking loss and implicitly weights hard negatives.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
replace LLaRA with S-DPO preference alignmentLLaMA2-7B recommender · GoodreadsHR@1: 0.5292 -> 0.6609Helped
replace LLaRA with S-DPO preference alignmentLLaMA2-7B recommender · LastFMHR@1: 0.4508 -> 0.6628Helped
replace LLaRA with S-DPO preference alignmentLLaMA2-7B recommender · MovieLensHR@1: 0.4737 -> 0.5263Helped
add explicit ranking optimization to SFT (ablation)LLaMA2-7B recommender · Goodreads, LastFM, and MovieLensHR@1Helped
add multiple negative samples to pairwise DPO (ablation)LLaMA2-7B recommender · Goodreads, LastFM, and MovieLensHR@1Helped
use K negatives instead of one with DPO (ablation)LLaMA2-7B recommender · LastFMHitRatio@1: 0.6342 -> 0.6413Helped
use K negatives instead of one with DPO (ablation)LLaMA2-7B recommender · MovieLensHitRatio@1: 0.4947 -> 0.4947No clear effect
use K negatives instead of one with DPO (ablation)LLaMA2-7B recommender · GoodreadsHitRatio@1: 0.6381 -> 0.6628Helped
replace DPO with S-DPO while using K negatives (ablation)LLaMA2-7B recommender · LastFMHitRatio@1: 0.6413 -> 0.6477Helped
replace DPO with S-DPO while using K negatives (ablation)LLaMA2-7B recommender · MovieLensHitRatio@1: 0.4947 -> 0.5263Helped
replace DPO with S-DPO while using K negatives (ablation)LLaMA2-7B recommender · GoodreadsHitRatio@1: 0.6628 -> 0.6661Helped
increase the number of negatives from 1 to 15 (ablation)S-DPO · Recommendation datasetsHR@1Helped

Surprising findings

Limitations

Experimental setup

Datasets: MovieLens Goodreads LastFM

Benchmarks: MovieLens Goodreads LastFM

Metrics: HR@1 Valid Ratio

Baselines: GRU4Rec Caser SASRec MoRec LLaMA2 Chat-REC TALLRec LLaRA

Builds on: DPO Plackett-Luce model BPR TALLRec LLaRA

Closest papers by meaning

Writing a paper in this area? scime draft is a free LaTeX editor with AI included, real-time co-authors and a check of your references and venue rules before you submit. It also imports OpenAI Prism projects.

Open scime draft

Get the next scime Dispatch and a monthly field update

Free. We send a confirmation link first; every email has a one-click unsubscribe. Privacy