Skip to content

Fotiligner/ILRec-KDD

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

4 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

ILRec

The official implementation of ILRec. The implementation of ILRec is based on PyTorch and RecBole.

Overview

  • We present ILRec, a novel LLM-based recommendation frame-work that extracts fine-grained self-hard negative signals from intermediate layers for preference optimization.
  • We propose the cross-layer preference fine-tuning to penalize negative tokens based on the self-hard signals during LLM fine-tuning. Moreover, we introduce collaborative reward regularization to prevent over-penalty and integrate collaborative information.

img

Quick Start

  1. Pre-processing
python data_processing.py
  1. Train the model with ILRec framework LC-Rec:
bash ./train/ilrec-lcrec.sh

BIGRec:

bash ./train/ilrec-bigrec.sh

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published