用检索增强大模型统一处理电动车充电数据缺失问题
A Unified Variational Imputation Framework for Electric Vehicle Charging Data Using Retrieval-Augmented Language Model
- 基于大模型与检索记忆的统一框架,融合时序、日历和地理信息
- 在4个公开数据集上显著提升填补精度并保持原始分布特性
- 适合电力基建、交通规划等需要高质量充电数据的研究者
电动汽车基础设施中的数据驱动应用(如充电需求预测)依赖于完整且高质量的充电数据。然而,真实世界中的电动车数据集常存在缺失记录,现有填补方法难以应对充电数据复杂的多模态特征,通常采用每站点一个模型的限制性范式,忽略站点间的相关性。为此,我们提出一种基于概率变分的统一填补框架——PRAIM,利用预训练语言模型将时间序列需求、日历特征与地理空间上下文编码为统一的语义丰富表示,并通过检索增强记忆动态获取整个充电网络的相关实例,使单一模型具备变分神经架构能力,有效缓解数据稀疏问题。在四个公开数据集上的大量实验表明,PRAIM在填补准确性和保留原始数据统计分布方面显著优于现有基线,大幅提升了下游预测性能。
原文摘要 · Abstract (English)
The reliability of data-driven applications in electric vehicle (EV) infrastructure, such as charging demand forecasting, hinges on the availability of complete, high-quality charging data. However, real-world EV datasets are often plagued by missing records, and existing imputation methods are ill-equipped for the complex, multimodal context of charging data, often relying on a restrictive one-model-per-station paradigm that ignores valuable inter-station correlations. To address these gaps, we develop a novel PRobabilistic variational imputation framework that leverages the power of large lAnguage models and retrIeval-augmented Memory (PRAIM). PRAIM employs a pre-trained language model to encode heterogeneous data, spanning time-series demand, calendar features, and geospatial context, into a unified, semantically rich representation. This is dynamically fortified by retrieval-augmented memory that retrieves relevant examples from the entire charging network, enabling a single, unified imputation model empowered by variational neural architecture to overcome data sparsity. Extensive experiments on four public datasets demonstrate that PRAIM significantly outperforms established baselines in both imputation accuracy and its ability to preserve the original data's statistical distribution, leading to substantial improvements in downstream forecasting performance.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。