arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2306.01792cs.IRcs.AIcs.LG
收藏

Task Relation-aware 持续用户表征学习

Task Relation-aware Continual User Representation Learning

KAIST(韩国科学技术院);NAVER Corporation(Naver公司)

Sein Kim, Namkyeong Lee, Donghyun Kim, Minchul Yang, Chanyoung Park

AI总结:

针对现有通用用户表征学习存在的灾难性遗忘、新增任务学习能力受限等问题,提出结合任务嵌入软掩码与伪标签知识保留模块的TERACON方法,在多数据集上验证了有效性。

摘要:

用户建模旨在基于用户的历史行为将用户表征映射到低维表征空间,因能为用户提供个性化服务而在工业界掀起研究热潮。现有用户建模研究大多聚焦于学习为单一任务设计的任务专属用户表征。然而,由于为每个任务都学习任务专属用户表征并不现实,近期研究提出了通用用户表征的概念,即一种与多种任务相关的、更具泛化性的用户表征。尽管这类方法具备一定有效性,但现有通用用户表征学习方法因数据需求高、存在灾难性遗忘问题,且对持续新增任务的学习能力有限,在实际应用中并不实用。本文提出一种名为TERACON的新型持续用户表征学习方法,该方法的学习能力不会随已学习任务数量的增加而受限,同时还能捕捉任务之间的关联。其核心思路是为每个任务引入一个嵌入(即任务嵌入),用于生成任务专属软掩码,该掩码不仅允许整个模型的参数在训练序列结束前持续更新,还能助力捕捉任务之间的关联。此外,我们提出了一种结合伪标签策略的新型知识保留模块,成功缓解了持续学习中长期存在的灾难性遗忘问题。在公开数据集和专有真实数据集上开展的大量实验证明了TERACON的优越性与实用性。我们的代码已发布于https://github.com/Sein-Kim/TERACON。

英文摘要:

User modeling, which learns to represent users into a low-dimensional representation space based on their past behaviors, got a surge of interest from the industry for providing personalized services to users. Previous efforts in user modeling mainly focus on learning a task-specific user representation that is designed for a single task. However, since learning task-specific user representations for every task is infeasible, recent studies introduce the concept of universal user representation, which is a more generalized representation of a user that is relevant to a variety of tasks. Despite their effectiveness, existing approaches for learning universal user representations are impractical in real-world applications due to the data requirement, catastrophic forgetting and the limited learning capability for continually added tasks. In this paper, we propose a novel continual user representation learning method, called TERACON, whose learning capability is not limited as the number of learned tasks increases while capturing the relationship between the tasks. The main idea is to introduce an embedding for each task, i.e., task embedding, which is utilized to generate task-specific soft masks that not only allow the entire model parameters to be updated until the end of training sequence, but also facilitate the relationship between the tasks to be captured. Moreover, we introduce a novel knowledge retention module with pseudo-labeling strategy that successfully alleviates the long-standing problem of continual learning, i.e., catastrophic forgetting. Extensive experiments on public and proprietary real-world datasets demonstrate the superiority and practicality of TERACON. Our code is available at https://github.com/Sein-Kim/TERACON.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑