arXiv:2410.19504cs.LGcs.AI2024-10TPAMI被引 2

用专家混合模型提升高维数据降维的精度与可解释性

MoE-Enhanced Explainable Deep Manifold Transformation for Complex Data Embedding and Visualization

  • 通过稀疏专家路由和双曲映射结合,增强降维表征能力
  • 在多个数据集上同时实现更高准确率与更强可解释性
  • 适合需要理解降维结果的科研与工程用户

降维技术在数据工程与可视化中至关重要,能简化复杂数据集并保留关键信息。然而,在高维数据处理中,如何兼顾高精度与强可解释性仍是根本挑战。传统方法常在性能与透明度间权衡,优化精度往往牺牲可解释性。为此,本文提出基于专家混合(MoE)的可解释深度流形变换(DMT-ME),融合几何感知的双曲映射与MoE模型,利用稀疏专家专长实现主要表征提升,双曲组件则针对结构复杂数据提供额外优化。该方法通过MoE结构显式关联输入数据、嵌入结果与关键特征,显著提升可解释性。大量实验表明,DMT-ME在降维精度与模型可解释性方面均表现优异,适用于图像、表格与文本等复杂数据场景。代码已开源。

原文摘要 · Abstract (English)

Dimensionality reduction (DR) plays a crucial role in various fields, including data engineering and visualization, by simplifying complex datasets while retaining essential information. However, achieving both high DR accuracy and strong explainability remains a fundamental challenge, especially for users dealing with high-dimensional data. Traditional DR methods often face a trade-off between precision and transparency, where optimizing for performance can lead to reduced explainability, and vice versa. This limitation is especially prominent in real-world applications such as image, tabular, and text data analysis, where both accuracy and explainability are critical. To address these challenges, this work introduces the MoE-based Explainable Deep Manifold Transformation (DMT-ME). The proposed approach combines a geometry-aware hyperbolic mapper with Mixture of Experts (MoE) models, where sparse expert specialization provides the main representational gain and the hyperbolic component offers an additional refinement for structurally complex data. DMT-ME enhances DR accuracy primarily through MoE-based sparse routing and structure-aware matching, while also improving explainability by explicitly linking input data, embedding outcomes, and key features through the MoE structure. Extensive experiments demonstrate that DMT-ME consistently achieves superior performance in both DR accuracy and model explainability, making it a robust solution for complex data analysis. The code is available at https://github.com/zangzelin/code_dmtme.

降维可解释性MoE流形学习

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