arXiv:2507.23267cs.IR2025-07被引 5

用Transformer模型提升金融行为预测,无需新增数据即可显著优化推荐效果。

Your Spending Needs Attention: Modeling Financial Habits with Transformers

  • 将Transformer用于交易数据的自监督学习,自动提取用户行为特征。
  • 在Nubank大规模推荐任务中,性能超越传统方法,提升明显。
  • 适合金融风控、个性化推荐等场景的算法研发人员阅读。

预测模型在金融领域至关重要,可影响风险评估、欺诈检测和个性化推荐,模型性能微小提升可能带来数十亿美元的收益或损失。尽管金融机构掌握海量用户数据(如银行交易、应用内事件、客服日志),但因数据复杂且规模大,有效利用仍具挑战。目前多数生产模型仍依赖传统机器学习,通过人工设计表格特征处理非结构化数据。而其他领域(如自然语言处理)已成功应用自监督学习(SSL)从原始数据中学习丰富表征,无需人工特征工程。本文研究使用基于Transformer的表示学习模型处理交易数据,假设其在大规模数据上训练后能提供理解客户行为的新范式。提出一种新方法nuFormer,通过适配Transformer模型以同时处理文本与结构化属性,并采用端到端微调策略,融合用户嵌入与现有表格特征。实验表明,在Nubank的大规模推荐任务中取得显著性能提升,且仅靠增强表征学习实现,未引入新数据源。

原文摘要 · Abstract (English)

Predictive models play a crucial role in the financial industry, enabling risk prediction, fraud detection, and personalized recommendations, where slight changes in core model performance can result in billions of dollars in revenue or losses. While financial institutions have access to enormous amounts of user data (e.g., bank transactions, in-app events, and customer support logs), leveraging this data effectively remains challenging due to its complexity and scale. Thus, in many financial institutions, most production models follow traditional machine learning (ML) approaches by converting unstructured data into manually engineered tabular features. Conversely, other domains (e.g., natural language processing) have effectively utilized self-supervised learning (SSL) to learn rich representations from raw data, removing the need for manual feature extraction. In this paper, we investigate using transformer-based representation learning models for transaction data, hypothesizing that these models, trained on massive data, can provide a novel and powerful approach to understanding customer behavior. We propose a new method enabling the use of SSL with transaction data by adapting transformer-based models to handle both textual and structured attributes. Our approach, denoted nuFormer, includes an end-to-end fine-tuning method that integrates user embeddings with existing tabular features. Our experiments demonstrate improvements for large-scale recommendation problems at Nubank. Notably, these gains are achieved solely through enhanced representation learning rather than incorporating new data sources.

金融建模Transformer自监督学习用户行为

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。