Skip to content

This is the official implementation of the IJCAI 2023 paper "Feature Staleness Aware Incremental Learning for CTR Prediction".

Notifications You must be signed in to change notification settings

SJTU-DMTai/FeSAIL

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

16 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

FeSAIL

Introduction

This is the official implementation for the IJCAI 2023 paper "Feature Staleness Aware Incremental Learning for CTR Prediction".

Abstract

Click-through Rate (CTR) prediction in real-world recommender systems often deals with billions of user interactions every day. To improve the training efficiency, it is common to update the CTR prediction model incrementally using the new incremental data and a subset of historical data. However, the feature embeddings of a CTR prediction model often get stale when the corresponding features do not appear in current incremental data. In the next period, the model would have a performance degradation on samples containing stale features, which we call the feature staleness problem. To mitigate this problem, we propose a Feature Staleness Aware Incremental Learning method for CTR prediction (FeSAIL) which adaptively replays samples containing stale features. We first introduce a staleness aware sampling algorithm (SAS) to sample a fixed number of stale samples with high sampling efficiency. We then introduce a staleness aware regularization mechanism (SAR) for a fine-grained control of the feature embedding updating. We instantiate FeSAIL with a general deep learningbased CTR prediction model and the experimental results demonstrate FeSAIL outperforms various state-of-the-art methods on four benchmark datasets.

Architecture

Requirement

pytorch == 1.14
python == 3.7

Instruction

1, Unzip the datasets into the root dir first.

2, You can run the code by:

python code/main.py

3, You can change ifrandom into 1 as random selection, and change oldrate into 1 to activate old sampling.

4, result on Criteo:

Reference

Please cite our paper if you use this code.

@inproceedings{wang2023fesail,
  title={Feature Staleness Aware Incremental Learning for CTR Prediction},
  author={Zhikai Wang and Yanyan Shen},
  booktitle={IJCAI},
  year={2023}
}

About

This is the official implementation of the IJCAI 2023 paper "Feature Staleness Aware Incremental Learning for CTR Prediction".

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages