arXiv:2604.16052math.OCcs.LG2026-04

用最优传输几何统一解释突触可塑性与记忆动态

A Wasserstein Geometric Framework for Hebbian Plasticity

  • 将记忆建模为概率测度,通过最小化沃尔沙伊特路径演化
  • 揭示内部记忆轨迹与外部观测权重的几何分离机制
  • 适用于神经网络、记忆结构及多尺度相位对齐分析

我们提出Tan-HWG框架(赫布-沃尔沙伊特-几何),将记忆状态建模为在沃尔沙伊特几何空间中演化的概率测度。赫布学习规则被形式化为满足序列稳定性的赫布能量,确保纤维层上JKO更新的适定性、最优传输实现和能量下降不等式。该变分结构实现了内部动力学与可观测动力学的根本分离:内部记忆状态在隐含曲面空间中沿沃尔沙伊特测地线演化,而可观测量(如有效突触权重)则通过几何投影映射到外空间。单纯形投影恢复经典仿射方案(包括指数移动平均和镜像下降),揭示突触竞争与剪枝是质量重分布的几何结果;希尔伯特投影提供相位对齐与多尺度相干性的几何解释。经典神经网络表现为该曲率动力学的平坦投影,框架自然支持更丰富的分布表示,包括结构权重、嵌入记忆及其在复内空间中的谱扩展。在弱利普希茨正则性假设下,包括准静态‘睡眠模式’,我们证明了连续时间极限曲线的存在性,从而将记忆巩固形式化为受扰的沃尔沙伊特梯度流。该框架为突触可塑性、表征动力学与上下文依赖计算提供了统一的几何基础。

原文摘要 · Abstract (English)

We introduce the Tan-HWG framework (Hebbian-Wasserstein-Geometry), a geometric theory of Hebbian plasticity in which memory states are modeled as probability measures evolving through Wasserstein minimizing movements. Hebbian learning rules are formalized as Hebbian energies satisfying a sequential stability condition, ensuring well-posed fiberwise JKO updates, optimal-transport realizations, and an energy descent inequality. This variational structure induces a fundamental separation between internal and observable dynamics. Internal memory states evolve along Wasserstein geodesics in a latent curved space, while observable quantities, such as effective synaptic weights, arise through geometric projection maps into external spaces. Simplicial projections recover classical affine schemes (including exponential moving averages and mirror descent), while revealing synaptic competition and pruning as geometric consequences of mass redistribution. Hilbertian projections provide a geometric account of phase alignment and multi-scale coherence. Classical neural networks appear as flat projections of this curved dynamics, while the framework naturally accommodates richer distributional representations, including structural weights and embedding memories, and their spectral extensions in complex internal spaces. Under mild Lipschitz regularity assumptions, including a quasi-stationary "sleep-mode" regime, we establish the existence of continuous-time limit curves. This yields a variational formulation of memory consolidation as a perturbed Wasserstein gradient flow. The framework thus provides a unified geometric foundation for synaptic plasticity, representation dynamics, and context-dependent computation.

神经科学最优传输记忆建模几何学习

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