用自适应掩码统一处理移动智能中的个性化预测难题
U-MASK: User-adaptive Spatio-Temporal Masking for Personalized Mobile AI Applications
- 将用户行为建模为部分可观测的时空张量,通过可学习掩码分配证据预算
- 在数据极度稀疏时仍能保持稳定预测,长短期任务性能均优于现有方法
- 适合冷启动、动态变化的移动应用,尤其适用于资源受限场景
个性化移动人工智能应用广泛部署,但需在持续演化的时空背景下,从稀疏不规则的历史中推断用户行为。这带来三大需求的根本矛盾:快速响应最新行为、抵抗瞬时噪声干扰、支持长期预测与冷启动用户。现有方法最多满足其中两项,形成数据稀缺、非平稳个性化中的不可能三角。为此,本文将移动行为建模为部分可观测的时空张量,将短期适应、长期预测与冷启动推荐统一为条件补全问题,由用户与任务特定的掩码决定哪些坐标作为证据。提出U-MASK,根据用户可信度与任务敏感性动态分配证据预算。为应对稀疏观测,U-MASK通过U-SCOPE从应用和位置历史中学习紧凑的、任务无关的用户表示,作为唯一语义条件信号。共享扩散变压器在掩码引导下完成生成补全,同时保留已观测证据,使个性化与任务区分完全由掩码和用户表示决定。在真实移动数据集上的实验表明,该方法在短时预测、长周期预测及冷启动场景中均一致优于当前最优方法,尤其在严重数据稀疏条件下提升显著。代码与数据集将公开于https://github.com/NICE-HKU/U-MASK。
原文摘要 · Abstract (English)
Personalized mobile artificial intelligence applications are widely deployed, yet they are expected to infer user behavior from sparse and irregular histories under a continuously evolving spatio-temporal context. This setting induces a fundamental tension among three requirements, i.e., immediacy to adapt to recent behavior, stability to resist transient noise, and generalization to support long-horizon prediction and cold-start users. Most existing approaches satisfy at most two of these requirements, resulting in an inherent impossibility triangle in data-scarce, non-stationary personalization. To address this challenge, we model mobile behavior as a partially observed spatio-temporal tensor and unify short-term adaptation, long-horizon forecasting, and cold-start recommendation as a conditional completion problem, where a user- and task-specific mask specifies which coordinates are treated as evidence. We propose U-MASK, a user-adaptive spatio-temporal masking method that allocates evidence budgets based on user reliability and task sensitivity. To enable mask generation under sparse observations, U-MASK learns a compact, task-agnostic user representation from app and location histories via U-SCOPE, which serves as the sole semantic conditioning signal. A shared diffusion transformer then performs mask-guided generative completion while preserving observed evidence, so personalization and task differentiation are governed entirely by the mask and the user representation. Experiments on real-world mobile datasets demonstrate consistent improvements over state-of-the-art methods across short-term prediction, long-horizon forecasting, and cold-start settings, with the largest gains under severe data sparsity. The code and dataset will be available at https://github.com/NICE-HKU/U-MASK.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。