arXiv:2510.25113cs.LGcs.AI2025-10被引 1

让神经网络自带几何结构,提升可解释性与泛化能力

The Neural Differential Manifold: An Architecture with Explicit Geometric Structure

  • 将网络层视为流形上的坐标图,参数直接定义黎曼度量
  • 通过双目标损失抑制曲率与体积畸变,增强鲁棒性
  • 适合追求模型可解释性与科学建模的深度学习研究者

本文提出神经微分流形(Neural Differential Manifold, NDM),一种在基础设计中显式融入几何结构的新型神经网络架构。不同于传统欧几里得参数空间,NDM 将神经网络重新构想为一个可微流形,其中每一层作为局部坐标图,网络参数直接在每一点上参数化黎曼度量张量。该架构由三个协同层构成:坐标层通过受归一化流启发的可逆变换实现平滑的坐标切换;几何层通过辅助子网络动态生成流形度量;演化层则通过双目标损失函数优化任务性能与几何简洁性。这种几何正则化惩罚过度曲率与体积畸变,提供内在正则化以增强泛化与鲁棒性。该框架支持与学习到的流形几何一致的自然梯度下降,并通过赋予内部表示明确几何意义实现前所未有的可解释性。理论分析表明其在优化效率、持续学习及科学发现和可控生成建模方面具有潜力。尽管仍存在显著计算挑战,但 NDM 标志着向几何结构化、可解释且高效的深度学习系统的重要转变。

原文摘要 · Abstract (English)

This paper introduces the Neural Differential Manifold (NDM), a novel neural network architecture that explicitly incorporates geometric structure into its fundamental design. Departing from conventional Euclidean parameter spaces, the NDM re-conceptualizes a neural network as a differentiable manifold where each layer functions as a local coordinate chart, and the network parameters directly parameterize a Riemannian metric tensor at every point. The architecture is organized into three synergistic layers: a Coordinate Layer implementing smooth chart transitions via invertible transformations inspired by normalizing flows, a Geometric Layer that dynamically generates the manifold's metric through auxiliary sub-networks, and an Evolution Layer that optimizes both task performance and geometric simplicity through a dual-objective loss function. This geometric regularization penalizes excessive curvature and volume distortion, providing intrinsic regularization that enhances generalization and robustness. The framework enables natural gradient descent optimization aligned with the learned manifold geometry and offers unprecedented interpretability by endowing internal representations with clear geometric meaning. We analyze the theoretical advantages of this approach, including its potential for more efficient optimization, enhanced continual learning, and applications in scientific discovery and controllable generative modeling. While significant computational challenges remain, the Neural Differential Manifold represents a fundamental shift towards geometrically structured, interpretable, and efficient deep learning systems.

几何深度学习可解释性流形学习神经网络架构

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