AutoRec:一种自动化推荐系统
信息检索
2020-07-15 v1 机器学习
机器学习
摘要
现实中的推荐系统常需适应不断变化的数据与任务,或系统性地探索不同模型。为满足该需求,我们提出 AutoRec,一个基于 TensorFlow 生态系统扩展的开源自动化机器学习(AutoML, automated machine learning)平台,并且据我们所知,它是首个利用 AutoML 进行深度推荐模型中模型搜索与超参数调优的框架。AutoRec 还支持高度灵活的流水线,可容纳稀疏与稠密输入、评分预测与点击率(CTR, click-through rate)预测任务,以及一系列推荐模型。最后,AutoRec 提供了简单、用户友好的 API。在基准数据集上的实验表明 AutoRec 可靠,且能在无先验知识情况下识别出媲美最佳模型的模型。
引用
@article{arxiv.2007.07224,
title = {AutoRec: An Automated Recommender System},
author = {Ting-Hsiang Wang and Qingquan Song and Xiaotian Han and Zirui Liu and Haifeng Jin and Xia Hu},
journal= {arXiv preprint arXiv:2007.07224},
year = {2020}
}