Skip to main content

使用 AI 模型预测缺失值的工作原理

现实世界的空间和表格数据集常因传感器中断、调查无响应、数据集成问题或手动数据输入错误而缺失或不完整。 缺失值会降低数据质量,限制分析,并可能对下游建模和决策产生负面影响。 丢弃缺失值记录通常会导致重大信息丢失,因此智能填充成为关键的预处理步骤。

使用 AI 模型预测缺失值工具使用先进的机器学习和深度学习模型,填充要素图层和独立表中的缺失数值。 与独立处理单个字段的统计隙填方法不同,该工具捕捉所有可用字段之间的关系,生成统计一致且具上下文感知的填充。

该工具在填充缺失值时不考虑相邻要素之间的空间或时间关系。 当缺失数据主要受空间邻域或时间连续性影响时,填充缺失值工具更为适用。

工具目前支持基于 XGBoost 的填充基于 DistilGPT-2 的填充模型,这些模型以 Esri 模型定义文件 (.emd) 或深度学习包 (.dlpk) 的形式打包,允许您根据数据集大小、复杂度和系统资源选择最合适的方法。

基于 AI 的方法

统计填充方法如最小值、最大值、平均值、中位数或众数,通过全局统计量或来自空间或时间上邻域位置的值来替代缺失值。 虽然这些方法快速且易于应用,但存在以下限制:

  • 忽略多个字段之间的关系。

  • 独立处理每个缺失值。

  • 无法捕捉非线性交互或依赖关系。

  • 当缺失值非随机时,结果可能会有偏差。

使用 AI 模型预测缺失值工具通过建模要素联合分布,实现更准确、更真实的填充,尤其适用于复杂高维数据集。

支持的填充模型

工具支持两种主要模型类型,均可通过 ArcGIS Living Atlas of the World 访问。

基于 XGBoost 的填充

使用基于 XGBoost 的填充模型时,工具执行预测填充如下:

  • 目标字段中具有观察值的记录用于训练。

  • 剩余字段作为目标字段的预测因子变量。

  • 经过训练的模型会为目标字段为空的记录预测缺失值。

  • 数据集会使用预测值进行更新。

模型的关键特征包含以下几点:

  • 对于大型表和高字段计数来说高效。

  • 捕捉非线性关系和要素交互。

  • 相比深度学习模型,内存需求更低。

  • 适用于生产工作流和大型企业数据集。

基于 DistilGPT-2 的填充

使用基于 DistilGPT-2 的填充模型时,工具采用基于深度学习的生成式、按行填充策略。 该策略可以归纳如下:

  • 每一行都被序列化成类似句子的要素值对序列。

  • 模型通过自回归下一个令牌预测对输入数据集进行微调。

  • 缺失值在推断过程中会被掩膜。

  • 模型基于同一行中的观测字段,生成缺失值。

  • 生成的数值输出会被转换回原始的数值格式。

这种方法使模型能够同时学习所有字段上的复杂依赖关系,并且在复杂填充方案下,误差指标(如 RMSE 和 MAE)相比传统机器学习基线更低。

模型的关键特征包含以下几点:

  • 学习联合要素分布。

  • 对复杂且相互依赖的数据集有效。

  • 支持未来扩展至更多转换器模型。

  • GPU 内存需求比 XGBoost 高。

性能与内存注意事项

基于深度学习的模型需要占用大量内存。 请考虑以下方法:

  • DistilGPT-2

    • 推荐用于字段数不超过 40 的表。

    • 需要高显存 GPU(24GB 或以上)。

    • 在表较宽时可能会遇到内存错误。

  • XGBoost

    • 支持更大的表和更高的字段数量。

    • 降低内存占用和计算需求。

    • 推荐用于企业级数据集。

以下是针对内存限制的解决方法:

  • 通过要素选择减少输入字段数量。

  • 将宽表分割成更小的子集。

  • 在云或高内存系统上运行工具。

何时使用此工具

当以下情况成立时,使用使用 AI 模型预测缺失值

  • 缺失值出现在多个相互关联的字段中。

  • 保持多元关系非常重要。

  • 统计填充会产生不切实际的结果。

  • 数据质量对于下游分析或建模至关重要。

在使用较简单的数据集、单个字段间隙,或缺失值最好用空间或时间邻近性来解释的情况时,像填充缺失值这样的工具通常很合适。

参考文献

  • Vadim Borisov, Kathrin Seßler, Tobias Leemann, Martin Pawelczyk, Gjergji Kasneci. "Language Models are Realistic Tabular Data Generators." October 12, 2022. https://arxiv.org/abs/2210.06280

  • Tianqi Chen, and Carlos Guestrin. "XGBoost: A Scalable Tree Boosting System." March 9, 2016. https://arxiv.org/abs/1603.02754

  • Ashish Vaswani, Noam Shazeer, Niki Parmar, Jakob Uszkoreit, Llion Jones, Aidan N. Gomez, Lukasz Kaiser, Illia Polosukhin. "Attention Is All You Need." June 12, 2017. https://arxiv.org/abs/1603.02754