跳到正文
Apple Machine Learning Research·· 2026-07-06AI 评分42

Fortress:一项关于通过时序数据增强和特征剪枝稳定搜索推荐的案例研究

Fortress: A Case Study in Stabilizing Search Recommendations via Temporal Data Augmentation and Feature Pruning

AI 导读

Fortress是一个提升搜索推荐模型稳定性与准确性的通用框架。其通过收集历史数据快照、识别预测不稳定样本、剔除导致波动的特征,并仅用稳定特征重新训练模型,来抑制来自用户交互信号(如engagement特征)的波动性,同时保留其预测能力。该方法在大型应用商店的查询到应用相关性模型上验证,离线实验表明预测稳定性(以变异系数衡量)和分类性能(以PR-AUC衡量)均有提升。

正文 · 原文

AuthorsMilind Pandurang Jagre, Jia Huang**, Dayvid V. R. Oliveira, Zhinan Cheng**, Babak Seyed Aghazadeh**, Puja Das, Chris Alvino, Jinda Han, Kailash Thiyagarajan

In search and recommendation systems, predictive models often suffer from temporal instability when certain input features introduce volatility in output scores. This instability can degrade model reliability and user experience especially in multi-stage systems where consistent predictions are critical for downstream decision making. We introduce Fortress, a general framework for enhancing model stability and accuracy by identifying and pruning features that contribute to inconsistent prediction scores over time. Fortress leverages historical snapshots temporally partitioned datasets capturing score fluctuations for the same entity across periods and follows a four-step process: (1) collect historical snapshots, (2) identify samples with unstable predictions, (3) isolate and remove instability-inducing features, and (4) retrain models using only stable features. While semantic features from LLMs and BERT-based models improve generalization, they often lack full query or entity coverage. Engagement-based features offer strong predictive power but tend to introduce temporal instability. Fortress mitigates this trade-off by suppressing the volatility of engagement signals while retaining their predictive value leading to more stable and accurate models. We validate Fortress on a query-to-app relevance model in a large-scale app marketplace. Offline experiments demonstrate notable improvements in prediction stability (measured by Coefficient of Variation) and classification performance (measured by PR-AUC).

  • ** Work done while at Apple

Related readings and updates.

NarrativeTrack: Evaluating Entity-Centric Reasoning for Narrative Understanding

Multimodal large language models (MLLMs) have achieved impressive progress in vision-language reasoning, yet their ability to understand temporally unfolding narratives in videos remains underexplored. True narrative understanding requires grounding who is doing what, when, and where, maintaining coherent entity representations across dynamic visual and temporal contexts. We introduce NarrativeTrack, the first benchmark to evaluate narrative…

Model Stability with Continuous Data Updates

In this paper, we study the “stability” of machine learning (ML) models within the context of larger, complex NLP systems with continuous training data updates. For this study, we propose a methodology for the assessment of model stability (which we refer to as jitter under various experimental conditions. We find that model design choices, including network architecture and input representation, have a critical impact on stability through…

Bottom banner

Discover opportunities in Machine Learning.

来源:Apple Machine Learning Research · machinelearning.apple.com