math-skill 2.0.1 → 3.0.0

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (206) hide show
  1. package/README.en-US.md +167 -175
  2. package/README.md +167 -175
  3. package/agents/math-critic.en.md +235 -0
  4. package/agents/math-critic.md +10 -8
  5. package/commands/ask.md +3 -12
  6. package/design-patterns/attention/equivariant-attention.en.md +72 -0
  7. package/design-patterns/attention/equivariant-attention.md +72 -0
  8. package/design-patterns/attention/geometry-aware-attention.en.md +73 -0
  9. package/design-patterns/attention/geometry-aware-attention.md +72 -0
  10. package/design-patterns/attention/information-bottleneck-attention.en.md +72 -0
  11. package/design-patterns/attention/information-bottleneck-attention.md +72 -0
  12. package/design-patterns/attention/projection-attention.en.md +68 -0
  13. package/design-patterns/attention/projection-attention.md +68 -0
  14. package/design-patterns/attention/spectral-attention.en.md +71 -0
  15. package/design-patterns/attention/spectral-attention.md +71 -0
  16. package/design-patterns/compression/leverage-score-selection.en.md +71 -0
  17. package/design-patterns/compression/leverage-score-selection.md +71 -0
  18. package/design-patterns/compression/low-rank-kv-cache.en.md +60 -0
  19. package/design-patterns/compression/low-rank-kv-cache.md +60 -0
  20. package/design-patterns/compression/spectral-token-pruning.en.md +61 -0
  21. package/design-patterns/compression/spectral-token-pruning.md +61 -0
  22. package/design-patterns/compression/topology-preserving-compression.en.md +64 -0
  23. package/design-patterns/compression/topology-preserving-compression.md +64 -0
  24. package/design-patterns/loss/constraint-penalty.en.md +67 -0
  25. package/design-patterns/loss/constraint-penalty.md +78 -0
  26. package/design-patterns/loss/contrastive-loss.en.md +59 -0
  27. package/design-patterns/loss/contrastive-loss.md +68 -0
  28. package/design-patterns/loss/information-bottleneck-loss.en.md +69 -0
  29. package/design-patterns/loss/information-bottleneck-loss.md +74 -0
  30. package/design-patterns/loss/orthogonality-loss.en.md +56 -0
  31. package/design-patterns/loss/orthogonality-loss.md +60 -0
  32. package/design-patterns/loss/variational-loss.en.md +61 -0
  33. package/design-patterns/loss/variational-loss.md +71 -0
  34. package/design-patterns/representation/equivariant-split.en.md +87 -0
  35. package/design-patterns/representation/equivariant-split.md +89 -0
  36. package/design-patterns/representation/manifold-representation.en.md +85 -0
  37. package/design-patterns/representation/manifold-representation.md +87 -0
  38. package/design-patterns/representation/shared-private-decomposition.en.md +84 -0
  39. package/design-patterns/representation/shared-private-decomposition.md +86 -0
  40. package/design-patterns/representation/subspace-alignment.en.md +90 -0
  41. package/design-patterns/representation/subspace-alignment.md +92 -0
  42. package/design-patterns/routing/graph-routing.en.md +79 -0
  43. package/design-patterns/routing/graph-routing.md +81 -0
  44. package/design-patterns/routing/moe-routing.en.md +85 -0
  45. package/design-patterns/routing/moe-routing.md +87 -0
  46. package/design-patterns/routing/optimal-transport-routing.en.md +68 -0
  47. package/design-patterns/routing/optimal-transport-routing.md +72 -0
  48. package/design-patterns/routing/spectral-clustering-routing.en.md +88 -0
  49. package/design-patterns/routing/spectral-clustering-routing.md +90 -0
  50. package/knowledge-base/differential-geometry/connection.en.md +51 -0
  51. package/knowledge-base/differential-geometry/connection.md +51 -0
  52. package/knowledge-base/differential-geometry/curvature.en.md +49 -0
  53. package/knowledge-base/differential-geometry/curvature.md +49 -0
  54. package/knowledge-base/differential-geometry/geodesic.en.md +51 -0
  55. package/knowledge-base/differential-geometry/geodesic.md +51 -0
  56. package/knowledge-base/differential-geometry/manifold.en.md +47 -0
  57. package/knowledge-base/differential-geometry/manifold.md +47 -0
  58. package/knowledge-base/differential-geometry/metric-tensor.en.md +50 -0
  59. package/knowledge-base/differential-geometry/metric-tensor.md +50 -0
  60. package/knowledge-base/differential-geometry/tangent-space.en.md +48 -0
  61. package/knowledge-base/differential-geometry/tangent-space.md +48 -0
  62. package/knowledge-base/information-geometry/fisher-metric.en.md +56 -0
  63. package/knowledge-base/information-geometry/fisher-metric.md +56 -0
  64. package/knowledge-base/information-geometry/natural-gradient.en.md +54 -0
  65. package/knowledge-base/information-geometry/natural-gradient.md +54 -0
  66. package/knowledge-base/lie-theory/equivariance.en.md +54 -0
  67. package/knowledge-base/lie-theory/equivariance.md +54 -0
  68. package/knowledge-base/lie-theory/group-action.en.md +51 -0
  69. package/knowledge-base/lie-theory/group-action.md +51 -0
  70. package/knowledge-base/lie-theory/lie-algebra.en.md +51 -0
  71. package/knowledge-base/lie-theory/lie-algebra.md +51 -0
  72. package/knowledge-base/lie-theory/lie-group.en.md +51 -0
  73. package/knowledge-base/lie-theory/lie-group.md +51 -0
  74. package/knowledge-base/lie-theory/representation.en.md +52 -0
  75. package/knowledge-base/lie-theory/representation.md +52 -0
  76. package/knowledge-base/matrix-analysis/low-rank-approximation.en.md +49 -0
  77. package/knowledge-base/matrix-analysis/low-rank-approximation.md +49 -0
  78. package/knowledge-base/matrix-analysis/matrix-perturbation.en.md +49 -0
  79. package/knowledge-base/matrix-analysis/matrix-perturbation.md +49 -0
  80. package/knowledge-base/matrix-analysis/positive-semidefinite.en.md +50 -0
  81. package/knowledge-base/matrix-analysis/positive-semidefinite.md +50 -0
  82. package/knowledge-base/matrix-analysis/projection.en.md +47 -0
  83. package/knowledge-base/matrix-analysis/projection.md +47 -0
  84. package/knowledge-base/matrix-analysis/spectral-decomposition.en.md +48 -0
  85. package/knowledge-base/matrix-analysis/spectral-decomposition.md +48 -0
  86. package/knowledge-base/optimization/constrained-optimization.en.md +53 -0
  87. package/knowledge-base/optimization/constrained-optimization.md +53 -0
  88. package/knowledge-base/optimization/convex-optimization.en.md +50 -0
  89. package/knowledge-base/optimization/convex-optimization.md +50 -0
  90. package/knowledge-base/optimization/lagrangian-duality.en.md +50 -0
  91. package/knowledge-base/optimization/lagrangian-duality.md +50 -0
  92. package/knowledge-base/optimization/proximal-method.en.md +52 -0
  93. package/knowledge-base/optimization/proximal-method.md +52 -0
  94. package/knowledge-base/optimization/riemannian-optimization.en.md +51 -0
  95. package/knowledge-base/optimization/riemannian-optimization.md +51 -0
  96. package/knowledge-base/overview.en.md +39 -0
  97. package/knowledge-base/overview.md +29 -218
  98. package/knowledge-base/probability/concentration-inequality.en.md +45 -0
  99. package/knowledge-base/probability/concentration-inequality.md +45 -0
  100. package/knowledge-base/probability/entropy.en.md +47 -0
  101. package/knowledge-base/probability/entropy.md +47 -0
  102. package/knowledge-base/probability/fisher-information.en.md +48 -0
  103. package/knowledge-base/probability/fisher-information.md +48 -0
  104. package/knowledge-base/probability/information-bottleneck.en.md +49 -0
  105. package/knowledge-base/probability/information-bottleneck.md +49 -0
  106. package/knowledge-base/probability/kl-divergence.en.md +51 -0
  107. package/knowledge-base/probability/kl-divergence.md +51 -0
  108. package/knowledge-base/topology/euler-characteristic.en.md +54 -0
  109. package/knowledge-base/topology/euler-characteristic.md +54 -0
  110. package/knowledge-base/topology/fundamental-group.en.md +52 -0
  111. package/knowledge-base/topology/fundamental-group.md +52 -0
  112. package/knowledge-base/topology/persistent-homology.en.md +54 -0
  113. package/knowledge-base/topology/persistent-homology.md +54 -0
  114. package/lenses/algorithmic.en.md +49 -0
  115. package/lenses/algorithmic.md +49 -0
  116. package/lenses/axiomatization.en.md +48 -0
  117. package/lenses/axiomatization.md +48 -0
  118. package/lenses/categorical.en.md +48 -0
  119. package/lenses/categorical.md +48 -0
  120. package/lenses/causal.en.md +49 -0
  121. package/lenses/causal.md +49 -0
  122. package/lenses/duality.en.md +47 -0
  123. package/lenses/duality.md +47 -0
  124. package/lenses/game.en.md +50 -0
  125. package/lenses/game.md +50 -0
  126. package/lenses/geometric.en.md +53 -0
  127. package/lenses/geometric.md +53 -0
  128. package/lenses/local-to-global.en.md +52 -0
  129. package/lenses/local-to-global.md +52 -0
  130. package/lenses/perturbation.en.md +51 -0
  131. package/lenses/perturbation.md +51 -0
  132. package/lenses/probabilistic.en.md +48 -0
  133. package/lenses/probabilistic.md +48 -0
  134. package/lenses/projection.en.md +50 -0
  135. package/lenses/projection.md +50 -0
  136. package/lenses/spectral.en.md +52 -0
  137. package/lenses/spectral.md +52 -0
  138. package/lenses/symmetry.en.md +49 -0
  139. package/lenses/symmetry.md +49 -0
  140. package/lenses/topological.en.md +48 -0
  141. package/lenses/topological.md +48 -0
  142. package/lenses/variational.en.md +48 -0
  143. package/lenses/variational.md +48 -0
  144. package/package.json +11 -3
  145. package/references/agentic-workflow.en.md +53 -0
  146. package/references/agentic-workflow.md +2 -0
  147. package/references/books/abstract-algebra.md +9 -5
  148. package/references/books/algebraic-geometry-rising-sea.md +9 -5
  149. package/references/books/differential-geometry.md +7 -3
  150. package/references/books/matrix-analysis.md +8 -4
  151. package/references/books/micro-lie-theory.md +7 -3
  152. package/references/books/optimization-ml.md +5 -1
  153. package/references/books/smooth-manifolds.md +7 -3
  154. package/references/gpu-friendly-math.en.md +65 -0
  155. package/references/gpu-friendly-math.md +4 -2
  156. package/references/inspiration.en.md +113 -0
  157. package/references/inspiration.md +4 -2
  158. package/skills/math-research-activator/SKILL.en.md +187 -0
  159. package/skills/math-research-activator/SKILL.md +162 -108
  160. package/skills/math-research-activator/original-texts.en.md +105 -0
  161. package/skills/math-research-activator/original-texts.md +11 -11
  162. package/commands/abstraction.md +0 -13
  163. package/commands/algorithmic-thinking.md +0 -13
  164. package/commands/axiomatization.md +0 -13
  165. package/commands/causal-inference.md +0 -13
  166. package/commands/discrete-combinatorial.md +0 -13
  167. package/commands/game-theory.md +0 -13
  168. package/commands/induction-analogy.md +0 -13
  169. package/commands/information-theory.md +0 -13
  170. package/commands/logic-deduction.md +0 -13
  171. package/commands/modeling.md +0 -13
  172. package/commands/optimization.md +0 -13
  173. package/commands/probability-statistics.md +0 -13
  174. package/commands/symmetry-invariance.md +0 -13
  175. package/commands/topological-thinking.md +0 -13
  176. package/commands/transformation.md +0 -13
  177. package/skills/abstraction/SKILL.md +0 -119
  178. package/skills/abstraction/original-texts.md +0 -175
  179. package/skills/algorithmic-thinking/SKILL.md +0 -136
  180. package/skills/algorithmic-thinking/original-texts.md +0 -256
  181. package/skills/axiomatization/SKILL.md +0 -149
  182. package/skills/axiomatization/original-texts.md +0 -160
  183. package/skills/causal-inference/SKILL.md +0 -149
  184. package/skills/causal-inference/original-texts.md +0 -167
  185. package/skills/discrete-combinatorial/SKILL.md +0 -129
  186. package/skills/discrete-combinatorial/original-texts.md +0 -185
  187. package/skills/game-theory/SKILL.md +0 -121
  188. package/skills/game-theory/original-texts.md +0 -131
  189. package/skills/induction-analogy/SKILL.md +0 -150
  190. package/skills/induction-analogy/original-texts.md +0 -154
  191. package/skills/information-theory/SKILL.md +0 -138
  192. package/skills/information-theory/original-texts.md +0 -127
  193. package/skills/logic-deduction/SKILL.md +0 -133
  194. package/skills/logic-deduction/original-texts.md +0 -173
  195. package/skills/modeling/SKILL.md +0 -137
  196. package/skills/modeling/original-texts.md +0 -165
  197. package/skills/optimization/SKILL.md +0 -133
  198. package/skills/optimization/original-texts.md +0 -168
  199. package/skills/probability-statistics/SKILL.md +0 -149
  200. package/skills/probability-statistics/original-texts.md +0 -193
  201. package/skills/symmetry-invariance/SKILL.md +0 -137
  202. package/skills/symmetry-invariance/original-texts.md +0 -221
  203. package/skills/topological-thinking/SKILL.md +0 -126
  204. package/skills/topological-thinking/original-texts.md +0 -164
  205. package/skills/transformation/SKILL.md +0 -122
  206. package/skills/transformation/original-texts.md +0 -216
@@ -0,0 +1,90 @@
1
+ # Spectral Clustering Routing(谱聚类路由)
2
+ > **严谨性声明**:本文件中涉及复杂度、显存、FlashAttention 融合、Tensor Core、KV-Cache 压缩的结论均标注为「✅ 已验证 / ⚠️ 可改造需验证 / ❌ 不可行」。未标注的视为理论可行,需工程验证。
3
+
4
+ ## 适用问题
5
+ 当需要基于 token/样本的内在相似性结构进行分组路由时使用。典型场景:
6
+ (1) 无监督专家分配——没有路由标签时,用谱聚类自动发现 token 的自然分簇;
7
+ (2) 自适应专家初始化——训练初期用谱聚类结果初始化专家参数;
8
+ (3) 输入感知的动态聚类——不同 batch 的 token 分布不同,路由应自适应;
9
+ (4) 多粒度聚类——不同层使用不同粒度的谱聚类(粗→细)。
10
+ 核心诉求:**发现数据内在的簇结构,用于路由或专家初始化**。
11
+
12
+ ## 数学思想来源
13
+ - 透镜:lenses/geometric.md(谱图论、拉普拉斯特征映射)、lenses/variational.md(松弛与近似)
14
+ - 知识:knowledge-base/matrix-analysis/projection.md(特征值分解、Rayleigh 商)、
15
+ knowledge-base/differential-geometry/manifold.md(流形学习、图割)
16
+
17
+ ## 需要的数学知识
18
+ - **谱聚类 (Ng-Jordan-Weiss)**:
19
+ 1. 构造相似度图 W_{ij} = exp(-‖x_i - x_j‖² / 2σ²)
20
+ 2. 计算归一化拉普拉斯 L_sym = I - D^{-1/2} W D^{-1/2}
21
+ 3. 取前 k 个最小特征向量 U_k ∈ R^{N×k}
22
+ 4. 对 U_k 的行做 k-means 得到 k 个簇
23
+ - **Nyström 近似**:当 N 太大无法计算完整 W 时,采样 m<<N 个点
24
+ W ≈ C · W_m^{-1} · C^T,将特征分解降维到 m×m
25
+ - **谱松弛连续化**:离散聚类分配 → 连续特征向量 → 可微路由
26
+ 用 softmax(U_k · W_proj) 替代硬 k-means 分配
27
+ - **幂迭代加速**:不需完整特征分解,只需前 k 个特征向量
28
+ 用 Lanczos/Arnoldi 迭代 O(N²·k·iter) 或 randomized SVD O(N·k·log k)
29
+
30
+ ## AI 模块形式
31
+ ```
32
+ 模块:SpectralClusterRouter
33
+ 输入:X ∈ R^{N×d},簇数 K
34
+
35
+ 方法1 - 在线谱聚类路由(训练时周期性更新):
36
+ // 每 M 步更新一次聚类中心,推理时用最近邻
37
+ W = exp(-cdist(X_sample, X_sample) / (2σ²)) // m×m 采样相似度
38
+ L = I - D^{-1/2} W D^{-1/2} // 归一化拉普拉斯
39
+ U_k = eigsh(L, k=K, which='SM') // 前 K 个最小特征向量
40
+ centers = kmeans(U_k, K) // K 个聚类中心
41
+ // 路由:将新 token 投影到谱空间后分配
42
+ proj = X @ W_proj // N→K 维投影(可学习)
43
+ assignment = argmin(cdist(proj, centers)) // 最近中心分配
44
+
45
+ 方法2 - 可微谱路由(端到端):
46
+ // 用 softmax 松弛替代硬分配
47
+ sim_matrix = X @ X^T // N×N(或采样 m×m)
48
+ A = exp(sim_matrix / τ) // 相似度图(可学习 τ)
49
+ D_inv_sqrt = diag(1 / sqrt(sum(A, dim=1) + ε))
50
+ L_norm = I - D_inv_sqrt @ A @ D_inv_sqrt // 归一化拉普拉斯
51
+ // 近似前 K 个特征向量(幂迭代 + 正交化)
52
+ U_k = power_iteration_approx(L_norm, K, steps=5) // N×K
53
+ // 软分配
54
+ cluster_logits = U_k @ W_cluster // N×K(可学习投影)
55
+ route_probs = softmax(cluster_logits / τ_route) // 软路由概率
56
+
57
+ 方法3 - 锚点谱聚类(大规模):
58
+ anchors = kmeans_pp(X, m) // m 个锚点,m << N
59
+ Z = exp(-cdist(X, anchors) / (2σ²)) // N×m 亲和矩阵
60
+ L_anchor = I - D_z^{-1/2} Z^T Z D_z^{-1/2} // m×m 拉普拉斯
61
+ U_k = eigsh(L_anchor, K) // m×K 特征向量
62
+ route = Z @ U_k @ W_proj // N×K 路由分数
63
+ ```
64
+
65
+ ## 可实现结构
66
+ - **周期性离线聚类**:每 N_step 步收集 token 表示 → 离线谱聚类 → 更新路由表
67
+ - **Nyström 采样**:随机采样 m=1024 个代表点,将 N×N 问题降为 m×m
68
+ - **幂迭代实现**:5~10 步幂迭代 + Gram-Schmidt 正交化,GPU 友好
69
+ - **渐进式训练**:初期用 k-means 粗路由 → 中期谱聚类精化 → 后期可微调路由网络
70
+
71
+ ## GPU 可行性
72
+ - **张量化**:相似度矩阵 X@X^T 为 GEMM;拉普拉斯构造为 element-wise + 对角矩阵
73
+ - **GEMM 可映射**:方法3 的 Z^T@Z 为 GEMM (m×N)@(N×m);Z@U_k 为 GEMM (N×m)@(m×K)
74
+ - **复杂度**:完整谱聚类 O(N²·K) 不可扩展;Nyström O(N·m·K+m³);幂迭代 O(N²·K·T)
75
+ - **显存与 KV-Cache**:N×N 相似度矩阵在 N>4096 时 >64MB,必须采样降维
76
+ - **低精度稳定**:特征分解建议 fp32;exp(-dist/σ²) 在 fp16 下需 clip distance
77
+ - **并行与通信**:幂迭代的 matvec 高度并行;k-means 的 assign+update 可批并行
78
+ - **稀疏结构**:k-NN 图替代全连接图,W 稀疏度 >95%,SpMM 加速
79
+ - **算子融合**:D^{-1/2}@A@D^{-1/2} 的对角缩放可融合;cdist+exp+normalize 可融合
80
+
81
+ ## 论文表述方式
82
+ "利用谱聚类的连续松弛实现可微路由:构造 token 相似度图的归一化拉普拉斯,
83
+ 通过 Nyström 近似将 O(N²) 特征分解降为 O(Nm+K³),配合幂迭代实现 GPU 友好的
84
+ 在线谱聚类,聚类质量以 Normalized Cut 衡量保证 O(√(log N/K)) 的近似比。"
85
+
86
+ ## 风险
87
+ - N×N 相似度矩阵的显存和计算在长序列下不可扩展,必须采样或 k-NN 稀疏化
88
+ - 特征分解不可微(特征值重合时梯度未定义),端到端训练需松弛或 stop-gradient
89
+ - 簇数 K 需先验指定,且 K 变化时需重新聚类
90
+ - σ(带宽参数)对聚类质量敏感,过小导致孤立点,过大导致合并
@@ -0,0 +1,51 @@
1
+ # Connection
2
+
3
+ ## Minimal Definition
4
+
5
+ A connection $\nabla$ is a rule specifying "how to transport a vector along a curve from one point to another without extraneous rotation." Formally, $\nabla: \mathfrak{X}(M) \times \mathfrak{X}(M) \to \mathfrak{X}(M)$ satisfies the Leibniz rule. The Levi-Civita connection is the unique connection that is metric-compatible ($\nabla g = 0$) and torsion-free ($T = 0$).
6
+
7
+ ## Core Formulas
8
+
9
+ - Covariant derivative: $\nabla_X Y = \left(X^i \partial_i Y^k + X^i Y^j \Gamma^k_{ij}\right) \partial_k$
10
+ - Christoffel symbols (Levi-Civita): $\Gamma^k_{ij} = \frac{1}{2} g^{kl}(\partial_i g_{jl} + \partial_j g_{il} - \partial_l g_{ij})$
11
+ - Parallel transport equation: $\frac{D V^k}{dt} = \dot V^k + \Gamma^k_{ij} \dot\gamma^i V^j = 0$
12
+ - Connection form (on a principal bundle): $\omega \in \Omega^1(P, \mathfrak{g})$; the gauge field $A_\mu$ is a local connection form
13
+ - Curvature = non-commutativity of the connection: $R(X,Y) = [\nabla_X, \nabla_Y] - \nabla_{[X,Y]}$
14
+
15
+ ## Applicable Problems
16
+
17
+ - Cross-point vector comparison: tangent spaces at different points cannot be directly summed; a connection specifies the "transport rule"
18
+ - Momentum/state transport in optimization: historical gradients in Riemannian Adam must be carried across steps via parallel transport
19
+ - Gauge-equivariant networks: the freedom in choosing local coordinate frames (gauges) is aligned by the connection
20
+ - Physically constrained systems: the electromagnetic field equals the curvature of a U(1) connection; Yang-Mills equals the curvature of a non-Abelian connection
21
+
22
+ ## AI Design Translation
23
+
24
+ - **Vector transport module**: In Riemannian optimizers, transport the momentum $m_k \in T_{x_k}M$ to $T_{x_{k+1}}M$; closed-form transport (e.g., projection on the Stiefel manifold) can be expressed as GEMM
25
+ - **Gauge-equivariant CNN**: On manifolds/meshes, each edge carries a $G$-connection element that aligns the local frames of adjacent points, making convolution kernels invariant to local coordinate choices
26
+ - **Parallel transport regularization**: Penalize non-parallelism of the feature field under the connection $\|\nabla_X f\|^2$, enforcing smooth feature variation along the manifold
27
+ - **Connection learning parameterization**: Parameterize Christoffel symbols as neural network outputs, learning the "optimal transport rule" on the data manifold
28
+
29
+ ## Engineering Feasibility
30
+
31
+ GPU friendliness: the core challenge of connections is "serial ODE integration."
32
+ - **Closed-form parallel transport** (specific manifolds such as SO(3), Stiefel): single-step matrix operations, batchable, GPU-friendly
33
+ - **Parallel transport for general connections**: integrating $\dot{V} + \Gamma \dot\gamma V = 0$ along a curve is a serial ODE with poor parallelism
34
+ - **Christoffel symbol computation**: involves partial derivatives of the metric $g$ and $g^{-1}$; if $g$ has a closed form, $O(n^3)$; otherwise more expensive
35
+ - **Connections in gauge-equivariant CNNs**: one $G$-element action per edge (matrix-times-feature-vector), expressible as sparse matmul or batched small GEMM
36
+ - Key adaptation: use a single-step retraction/closed-form transport instead of step-by-step ODE integration
37
+
38
+ ## Risks and Failure Conditions
39
+
40
+ - **Step-by-step ODE integration for parallel transport**: Serial recurrence kills parallelism; closed-form transport or single-step approximations must be used
41
+ - **Inconsistent left/right connection conventions**: Mixing left-invariant and right-invariant connections leads to misaligned gradients
42
+ - **Improper handling of gauge freedom**: If the connection parameterization in gauge-equivariant networks is incomplete, equivariance silently breaks
43
+ - **Numerical derivatives of Christoffel symbols**: Finite-difference estimation of $\partial_i g_{jk}$ is noisy; analytical formulas or autodiff are preferable
44
+ - **Connection does not imply metric compatibility**: A connection need not admit a compatible metric (non-metric connections); incorrectly assuming compatibility leads to inconsistencies
45
+
46
+ ## Further References
47
+
48
+ - Distillation notes: references/books/differential-geometry.md (Ch 12 Connections and Covariant Derivatives, Section 12.2 Connection Forms, Section 12.4 Ehresmann, Section 12.12 G-Connections)
49
+ - Distillation notes: references/books/differential-geometry.md (Section 6.8 Principal Bundles, Section 9.8 Electromagnetism)
50
+ - Original text: Jeffrey M. Lee, *Manifolds and Differential Geometry*, Ch 12 (Section 12.1--Section 12.12, complete connection theory)
51
+ - Original text: Jeffrey M. Lee, *Manifolds and Differential Geometry*, Section 13.1 Levi-Civita Connection
@@ -0,0 +1,51 @@
1
+ # 联络 (Connection)
2
+
3
+ ## 最小定义
4
+
5
+ 联络 $\nabla$ 是流形上"如何把一个向量沿曲线搬到另一点而不额外旋转"的规则。形式化地,$\nabla: \mathfrak{X}(M) \times \mathfrak{X}(M) \to \mathfrak{X}(M)$ 满足 Leibniz 法则。Levi-Civita 联络是唯一与度量相容($\nabla g = 0$)且无挠($T = 0$)的联络。
6
+
7
+ ## 核心公式
8
+
9
+ - 协变导数:$\nabla_X Y = \left(X^i \partial_i Y^k + X^i Y^j \Gamma^k_{ij}\right) \partial_k$
10
+ - Christoffel 符号(Levi-Civita):$\Gamma^k_{ij} = \frac{1}{2} g^{kl}(\partial_i g_{jl} + \partial_j g_{il} - \partial_l g_{ij})$
11
+ - 平行移动方程:$\frac{D V^k}{dt} = \dot V^k + \Gamma^k_{ij} \dot\gamma^i V^j = 0$
12
+ - 联络形式(主丛):$\omega \in \Omega^1(P, \mathfrak{g})$,规范场 $A_\mu$ 即局部联络形式
13
+ - 曲率 = 联络的不交换性:$R(X,Y) = [\nabla_X, \nabla_Y] - \nabla_{[X,Y]}$
14
+
15
+ ## 适用问题
16
+
17
+ - 跨点向量比较:不同点的切空间无法直接相加,需要联络规定的"搬运规则"
18
+ - 优化中的动量/状态搬运:Riemannian Adam 中历史梯度需要通过 parallel transport 跨步搬运
19
+ - 规范等变网络:局部坐标系(gauge)的选择自由由联络对齐
20
+ - 物理约束系统:电磁场 = U(1) 联络曲率,Yang-Mills = 非阿贝尔联络曲率
21
+
22
+ ## AI 设计翻译
23
+
24
+ - **Vector transport 模块**:Riemannian 优化器中,将动量 $m_k \in T_{x_k}M$ 搬到 $T_{x_{k+1}}M$;闭式 transport(如 Stiefel 上的投影)可 GEMM 化
25
+ - **Gauge-equivariant CNN**:在流形/网格上,每条边携带一个 $G$-联络元素对齐相邻点的局部 frame,使卷积核对局部坐标选择不变
26
+ - **Parallel transport 正则化**:惩罚特征场在联络下的非平行性 $\|\nabla_X f\|^2$,强制特征沿流形平滑变化
27
+ - **联络学习的参数化**:将 Christoffel 符号参数化为神经网络输出,学习数据流形上的"最优搬运规则"
28
+
29
+ ## 工程可行性
30
+
31
+ GPU 友好度:联络的核心挑战是"串行 ODE 积分"。
32
+ - **闭式 parallel transport**(特定流形如 SO(3)、Stiefel):单步矩阵运算,可 batched,GPU 友好
33
+ - **一般联络的平行移动**:沿曲线积分 $\dot{V} + \Gamma \dot\gamma V = 0$ 是串行 ODE,并行性差
34
+ - **Christoffel 符号计算**:涉及度量 $g$ 的偏导和 $g^{-1}$,若 $g$ 有闭式则 $O(n^3)$,否则更贵
35
+ - **规范等变 CNN 中的联络**:每边一个 $G$-元素作用(矩阵乘特征向量),可表达为 sparse matmul 或 batched small GEMM
36
+ - 关键改造:用一步 retraction/closed-form transport 替代逐步 ODE 积分
37
+
38
+ ## 风险与失效条件
39
+
40
+ - **逐步 ODE 积分做平行移动**:串行递推杀死并行度,必须用闭式 transport 或一步近似
41
+ - **左右联络约定不统一**:左不变 vs 右不变联络的选择不一致导致梯度错位
42
+ - **规范自由度未正确处理**:gauge-equivariant 网络中若联络参数化不完备,等变性会悄悄破缺
43
+ - **Christoffel 符号的数值导数**:用有限差分估计 $\partial_i g_{jk}$ 时噪声大,最好有解析式或 autodiff
44
+ - **联络 ≠ 度量**:有联络不一定有相容度量(非度量联络),错误假设相容性会导致不一致
45
+
46
+ ## 深入参考
47
+
48
+ - 蒸馏稿:references/books/differential-geometry.md(Ch 12 Connections and Covariant Derivatives, §12.2 联络形式, §12.4 Ehresmann, §12.12 G-联络)
49
+ - 蒸馏稿:references/books/differential-geometry.md(§6.8 Principal Bundles, §9.8 Electromagnetism)
50
+ - 原书:Jeffrey M. Lee, *Manifolds and Differential Geometry*, Ch 12(§12.1-§12.12 完整联络理论)
51
+ - 原书:Jeffrey M. Lee, *Manifolds and Differential Geometry*, §13.1 Levi-Civita Connection
@@ -0,0 +1,49 @@
1
+ # Curvature
2
+
3
+ ## Minimal Definition
4
+
5
+ Curvature measures whether a manifold "bends" and whether "paths are path-dependent." The Riemann curvature tensor $R(X,Y)Z = \nabla_X\nabla_Y Z - \nabla_Y\nabla_X Z - \nabla_{[X,Y]}Z$ describes the deflection of a vector after parallel transport around an infinitesimal parallelogram. The sectional curvature $K(\sigma)$ is its simplest scalar extraction.
6
+
7
+ ## Core Formulas
8
+
9
+ - Riemann curvature tensor: $R^l_{ijk} = \partial_i \Gamma^l_{jk} - \partial_j \Gamma^l_{ik} + \Gamma^l_{im}\Gamma^m_{jk} - \Gamma^l_{jm}\Gamma^m_{ik}$
10
+ - Ricci curvature (contraction): $R_{ij} = \sum_k R^k_{ikj}$
11
+ - Scalar curvature: $S = \sum_{ij} g^{ij} R_{ij}$
12
+ - Sectional curvature: $K(X,Y) = \frac{\langle R(X,Y)Y, X\rangle}{\|X\|^2\|Y\|^2 - \langle X,Y\rangle^2}$
13
+ - Jacobi equation: $\frac{D^2 J}{dt^2} + R(J, \dot\gamma)\dot\gamma = 0$ (describes divergence/convergence of geodesics)
14
+ - Hessian-vector product: $Hv = \nabla(\nabla L \cdot v)$, $O(N)$ estimation of curvature information
15
+
16
+ ## Applicable Problems
17
+
18
+ - Loss landscape analysis: curvature determines conditioning and sharpness, distinguishing sharp minima from flat minima
19
+ - Optimization trajectory stability: Jacobi fields describe the divergence/convergence of neighboring optimization trajectories
20
+ - Generalization diagnostics: flat minima (low curvature) tend to generalize better
21
+ - Manifold learning: the curvature of the data manifold guides latent space dimension and metric selection
22
+
23
+ ## AI Design Translation
24
+
25
+ - **Curvature regularization (geometric perspective on SAM)**: Use Hessian-vector products to estimate $\max_v v^T H v$, penalizing sharp minima and preferring flat minima
26
+ - **HVP-based diagnostic**: $\|Hv\|/\|v\|$ as a cheap proxy for loss landscape curvature, used for learning rate adaptation and early stopping
27
+ - **Jacobi field trajectory monitoring**: Track the distance evolution between two neighboring optimization trajectories, the discrete analog of $J''(t) + R(J,\dot\gamma)\dot\gamma = 0$, to detect divergence/convergence
28
+ - **Ricci-flow-inspired graph rewiring**: Use discrete Ricci curvature to guide dynamic adjustment of graph/attention structure (negative-curvature edges indicate bottlenecks that need additional connections)
29
+
30
+ ## Engineering Feasibility
31
+
32
+ GPU friendliness: the core difficulty of curvature is that "the full tensor cannot be materialized."
33
+ - **Riemann tensor**: 4th order, $n^4$ components, materializing it exhausts memory -- **explicit computation is prohibited**
34
+ - **Hessian-vector product (HVP)**: Via Pearlmutter's algorithm, one forward pass + one backward pass yields $Hv$, $O(N)$ time and $O(N)$ memory, GPU-friendly
35
+ - **Monte Carlo estimation of Ricci/scalar curvature**: Randomly sample directions $v$, $\mathbb{E}[v^T H v] = \text{tr}(H)$, using Hutchinson's estimator, GPU-friendly
36
+ - **Jacobi fields**: Require integrating a second-order ODE along a trajectory, serial recurrence, GPU-unfriendly; in practice, discrete finite-difference approximations are used
37
+ - Low precision: second-order derivatives in HVP are noisy under fp16, requiring fp32 accumulation
38
+
39
+ ## Risks and Failure Conditions
40
+
41
+ - **Materializing the full Riemann/Hessian tensor**: $O(N^2)$ to $O(N^4)$ memory, impossible when $N \sim 10^9$
42
+ - **Low signal-to-noise ratio in curvature estimation**: Monte Carlo estimation of HVP has high variance; with small batches, the signal may be drowned in noise
43
+ - **Treating curvature regularization as a panacea**: Curvature estimation itself is expensive (requiring additional forward and backward passes); benefits should be validated at small scale first when uncertain
44
+ - **Discrete approximation errors**: When using finite differences to approximate Jacobi fields/HVP, the step size is sensitive -- too large causes truncation error, too small causes floating-point cancellation
45
+
46
+ ## Further References
47
+
48
+ - Distillation notes: references/books/differential-geometry.md (Ch 12 Section 12.5/Section 12.10 Curvature, Ch 13 Section 13.2 Riemann Curvature, Section 13.7 Jacobi Fields, Section 13.11 Rauch Comparison)
49
+ - Original text: Jeffrey M. Lee, *Manifolds and Differential Geometry*, Section 13.2 Riemann Curvature Tensor, Section 13.7 Jacobi Fields
@@ -0,0 +1,49 @@
1
+ # 曲率 (Curvature)
2
+
3
+ ## 最小定义
4
+
5
+ 曲率度量流形"弯不弯"以及"路径依不依赖"。Riemann 曲率张量 $R(X,Y)Z = \nabla_X\nabla_Y Z - \nabla_Y\nabla_X Z - \nabla_{[X,Y]}Z$ 描述向量绕无穷小平行四边形平行移动一圈后的偏转。截面曲率 $K(\sigma)$ 是其最简标量提取。
6
+
7
+ ## 核心公式
8
+
9
+ - Riemann 曲率张量:$R^l_{ijk} = \partial_i \Gamma^l_{jk} - \partial_j \Gamma^l_{ik} + \Gamma^l_{im}\Gamma^m_{jk} - \Gamma^l_{jm}\Gamma^m_{ik}$
10
+ - Ricci 曲率(缩并):$R_{ij} = \sum_k R^k_{ikj}$
11
+ - 标量曲率:$S = \sum_{ij} g^{ij} R_{ij}$
12
+ - 截面曲率:$K(X,Y) = \frac{\langle R(X,Y)Y, X\rangle}{\|X\|^2\|Y\|^2 - \langle X,Y\rangle^2}$
13
+ - Jacobi 方程:$\frac{D^2 J}{dt^2} + R(J, \dot\gamma)\dot\gamma = 0$(描述测地线的发散/汇聚)
14
+ - Hessian-vector product:$Hv = \nabla(\nabla L \cdot v)$,$O(N)$ 估计曲率信息
15
+
16
+ ## 适用问题
17
+
18
+ - 损失地形分析:曲率决定条件数与尖锐度,尖锐极小 vs 平坦极小的区分
19
+ - 优化轨迹稳定性:Jacobi 场描述相邻优化轨迹的发散/汇聚
20
+ - 泛化性诊断:平坦极小(低曲率)往往泛化更好
21
+ - 流形学习:数据流形的曲率指导隐空间维数和度量选择
22
+
23
+ ## AI 设计翻译
24
+
25
+ - **曲率正则化(SAM 的几何视角)**:用 Hessian-vector product 估计 $\max_v v^T H v$,惩罚尖锐极小,偏好 flat minima
26
+ - **HVP-based 诊断器**:$\|Hv\|/\|v\|$ 作为 loss landscape 曲率的廉价代理,用于学习率自适应和早停
27
+ - **Jacobi 场轨迹监控**:跟踪两条相邻优化轨迹的距离变化,$J''(t) + R(J,\dot\gamma)\dot\gamma = 0$ 的离散版,检测发散/收敛
28
+ - **Ricci-flow 启发式图重连**:用离散 Ricci 曲率指导图/注意力结构的动态调整(负曲率边=瓶颈,需增连)
29
+
30
+ ## 工程可行性
31
+
32
+ GPU 友好度:曲率的核心难点是"不能物化全张量"。
33
+ - **Riemann 张量**:4 阶,$n^4$ 个分量,物化即爆显存,**禁止显式计算**
34
+ - **Hessian-vector product (HVP)**:通过 Pearlmutter 算法,一次前向 + 一次反向即可得到 $Hv$,$O(N)$ 时间 $O(N)$ 显存,GPU 友好
35
+ - **Ricci/标量曲率的 Monte Carlo 估计**:随机采样方向 $v$,$\mathbb{E}[v^T H v] = \text{tr}(H)$,用 Hutchinson 估计,GPU 友好
36
+ - **Jacobi 场**:需要沿轨迹积分二阶 ODE,串行递推,GPU 不友好;工程上用离散差分近似
37
+ - 低精度:HVP 中的二阶导在 fp16 下噪声大,需 fp32 累加
38
+
39
+ ## 风险与失效条件
40
+
41
+ - **物化全 Riemann/Hessian 张量**:$O(N^2)$~$O(N^4)$ 显存,$N \sim 10^9$ 时不可能
42
+ - **曲率估计信噪比低**:HVP 的 Monte Carlo 估计方差大,小 batch 下信号可能被噪声淹没
43
+ - **把曲率正则当万能药**:曲率估计本身昂贵(每次需额外前向+反向),收益不确定时需先小规模验证
44
+ - **离散近似误差**:用有限差分近似 Jacobi 场/HVP 时,步长选择敏感——太大截断误差大,太小浮点抵消
45
+
46
+ ## 深入参考
47
+
48
+ - 蒸馏稿:references/books/differential-geometry.md(Ch 12 §12.5/§12.10 Curvature, Ch 13 §13.2 Riemann Curvature, §13.7 Jacobi Fields, §13.11 Rauch Comparison)
49
+ - 原书:Jeffrey M. Lee, *Manifolds and Differential Geometry*, §13.2 Riemann Curvature Tensor, §13.7 Jacobi Fields
@@ -0,0 +1,51 @@
1
+ # Geodesic
2
+
3
+ ## Minimal Definition
4
+
5
+ A geodesic is a locally length-minimizing (or extremal-length) path $\gamma: [0,1] \to M$ on a manifold, satisfying the auto-parallel condition $\nabla_{\dot\gamma} \dot\gamma = 0$ -- that is, the tangent vector is parallel-transported unchanged along itself. The exponential map $\exp_p(v)$ maps a tangent vector $v \in T_pM$ to the endpoint at $t=1$ of the geodesic starting at $p$ with initial velocity $v$.
6
+
7
+ ## Core Formulas
8
+
9
+ - Geodesic equation: $\ddot\gamma^k + \sum_{ij} \Gamma^k_{ij} \dot\gamma^i \dot\gamma^j = 0$
10
+ - Exponential map: $\exp_p(v) = \gamma_v(1)$, where $\gamma_v$ is the geodesic with $\gamma(0)=p, \dot\gamma(0)=v$
11
+ - Logarithmic map: $\log_p(q) = v \in T_pM$ such that $\exp_p(v) = q$ (the inverse of the exponential map)
12
+ - Retraction (engineering approximation): $R_p(v) \approx \exp_p(v)$, requiring only a first-order approximation for optimization
13
+ - Closed-form on the sphere: $\exp_p(v) = \cos(\|v\|) p + \sin(\|v\|) \frac{v}{\|v\|}$
14
+
15
+ ## Applicable Problems
16
+
17
+ - Constrained optimization: performing SGD on SPD/Stiefel/Grassmann/hyperbolic manifolds, where update steps must move along geodesics
18
+ - Latent space interpolation: geodesics between two points in the latent space respect the data manifold structure better than Euclidean straight lines
19
+ - Distance computation on manifolds: $d(p,q) = \|\log_p(q)\|_g$
20
+ - Data augmentation: sampling along geodesics to generate new training samples
21
+
22
+ ## AI Design Translation
23
+
24
+ - **Retraction-based optimizer**: Each step performs $x_{k+1} = R_{x_k}(-\eta \cdot \text{grad})$, using closed-form retractions instead of ODE integration; rotations for the sphere, QR/Cayley for Stiefel, Rodrigues for SO(3)
25
+ - **Geodesic interpolation layer**: Use closed-form geodesics in spherical/hyperbolic latent spaces for mixup and interpolation, $\gamma(t) = \exp_p(t \cdot \log_p(q))$
26
+ - **Momentum on manifolds**: Transport the momentum vector from $T_{x_k}M$ to $T_{x_{k+1}}M$ via vector transport (the discrete analog of parallel transport), then combine with the new gradient
27
+ - **Exponential map output head**: The network makes unconstrained predictions in the tangent space $\mathbb{R}^n$, then projects back to the valid manifold via $\exp_p$, naturally satisfying constraints
28
+
29
+ ## Engineering Feasibility
30
+
31
+ GPU friendliness depends on whether a closed-form retraction exists:
32
+ - **Manifolds with closed forms** (sphere, hyperbolic, SO(3), Stiefel-QR): $\exp_p(v)$ is a finite algebraic expression, $O(1)$/sample, tensorizable in batches, GPU-friendly
33
+ - **Manifolds without closed forms**: require numerical integration of the geodesic equation (second-order ODE), serial recurrence, GPU-unfriendly
34
+ - Closed-form retractions as substitutes for exact exp: QR decomposition, Cayley transform, and other first-order approximations trade a small amount of precision for significant speedup
35
+ - Small-matrix exp for 3x3/4x4 (SO(3)/SE(3)) can be fused into a single kernel, but cannot fully saturate Tensor Cores
36
+ - **Low-precision critical issue**: $\sin\theta/\theta$ and $(1-\cos\theta)/\theta^2$ produce division by zero as $\theta \to 0$; $\log$ is singular as $\theta \to \pi$; fp16 yields NaN directly
37
+
38
+ ## Risks and Failure Conditions
39
+
40
+ - **Step-by-step ODE integration for geodesics**: Serial recurrence kills GPU parallelism; closed-form retractions must be used instead
41
+ - **Small-angle / large-angle singularities**: Numerical instability at $\theta \to 0$ and $\theta \to \pi$ is catastrophically amplified at low precision; Taylor expansion fallbacks are essential
42
+ - **Retraction error accumulation**: First-order approximations may accumulate errors over multiple iterations, requiring occasional exact projection corrections
43
+ - **Cut locus problem**: Beyond the cut locus, the exponential map no longer yields the shortest path; $\log_p(q)$ may not exist or may not be unique
44
+ - **Forced manifold structure for geometric aesthetics**: Applying geodesics to tasks where Euclidean approximations suffice adds complexity and singularity risk
45
+
46
+ ## Further References
47
+
48
+ - Distillation notes: references/books/differential-geometry.md (Ch 13 Section 13.4 Geodesics, Section 13.11 Rauch Comparison)
49
+ - Distillation notes: references/books/smooth-manifolds.md (Ch 20 The Exponential Map)
50
+ - Original text: Jeffrey M. Lee, *Manifolds and Differential Geometry*, Section 13.4 Geodesics
51
+ - Original text: John M. Lee, *Introduction to Smooth Manifolds*, Ch 20 (exponential map, retraction prototype)
@@ -0,0 +1,51 @@
1
+ # 测地线 (Geodesic)
2
+
3
+ ## 最小定义
4
+
5
+ 测地线是流形上局部最短(或极值长度)的路径 $\gamma: [0,1] \to M$,满足自平行条件 $\nabla_{\dot\gamma} \dot\gamma = 0$——即切向量沿自身平行移动不变。指数映射 $\exp_p(v)$ 将切向量 $v \in T_pM$ 映为从 $p$ 出发、初速为 $v$ 的测地线在 $t=1$ 处的终点。
6
+
7
+ ## 核心公式
8
+
9
+ - 测地方程:$\ddot\gamma^k + \sum_{ij} \Gamma^k_{ij} \dot\gamma^i \dot\gamma^j = 0$
10
+ - 指数映射:$\exp_p(v) = \gamma_v(1)$,其中 $\gamma_v$ 是 $\gamma(0)=p, \dot\gamma(0)=v$ 的测地线
11
+ - 对数映射:$\log_p(q) = v \in T_pM$ 使得 $\exp_p(v) = q$(指数映射的逆)
12
+ - Retraction(工程近似):$R_p(v) \approx \exp_p(v)$,只需一阶近似即可用于优化
13
+ - 球面闭式:$\exp_p(v) = \cos(\|v\|) p + \sin(\|v\|) \frac{v}{\|v\|}$
14
+
15
+ ## 适用问题
16
+
17
+ - 约束优化:在 SPD/Stiefel/Grassmann/双曲流形上做 SGD,更新步需要沿测地线移动
18
+ - 隐空间插值:latent space 中两点间的测地线比欧氏直线更尊重数据流形结构
19
+ - 流形上的距离计算:$d(p,q) = \|\log_p(q)\|_g$
20
+ - 数据增广:沿测地线采样生成新训练样本
21
+
22
+ ## AI 设计翻译
23
+
24
+ - **Retraction-based 优化器**:每步做 $x_{k+1} = R_{x_k}(-\eta \cdot \text{grad})$,用闭式 retraction 替代 ODE 积分;球面用旋转、Stiefel 用 QR/Cayley、SO(3) 用 Rodrigues
25
+ - **测地线插值层**:在球面/双曲 latent space 中用闭式测地线做 mixup 和插值,$\gamma(t) = \exp_p(t \cdot \log_p(q))$
26
+ - **流形上的 momentum**:将动量向量通过 vector transport(平行移动的离散版)从 $T_{x_k}M$ 搬到 $T_{x_{k+1}}M$,再与新梯度合成
27
+ - **指数映射输出头**:网络在切空间 $\mathbb{R}^n$ 中自由预测,经 $\exp_p$ 投回合法流形,天然满足约束
28
+
29
+ ## 工程可行性
30
+
31
+ GPU 友好度取决于是否有闭式 retraction:
32
+ - **有闭式的流形**(球面、双曲、SO(3)、Stiefel-QR):$\exp_p(v)$ 是有限项代数表达式,$O(1)$/样本,可 batched 张量化,GPU 友好
33
+ - **无闭式的流形**:需要数值积分测地方程(二阶 ODE),串行递推,GPU 不友好
34
+ - 闭式 retraction 替代精确 exp:QR 分解、Cayley 变换等一阶近似,牺牲少量精度换取大幅加速
35
+ - 3x3/4x4 小矩阵 exp(SO(3)/SE(3))可融进单 kernel,但吃不满 Tensor Core
36
+ - **低精度致命点**:$\sin\theta/\theta$、$(1-\cos\theta)/\theta^2$ 在 $\theta \to 0$ 时除零,$\theta \to \pi$ 时 log 奇异,fp16 直接 NaN
37
+
38
+ ## 风险与失效条件
39
+
40
+ - **逐步 ODE 积分做测地线**:串行递推杀死 GPU 并行度,必须用闭式 retraction 替代
41
+ - **小角/大角奇异点**:$\theta \to 0$ 和 $\theta \to \pi$ 处的数值不稳定在低精度下被灾难性放大,必须做 Taylor 展开兜底
42
+ - **Retraction 误差累积**:一阶近似在多步迭代中误差可能累积,需偶尔做一次精确投影校正
43
+ - **Cut locus 问题**:指数映射在 cut locus 以外不再是最短路径,$\log_p(q)$ 可能不存在或不唯一
44
+ - **为几何美强行流形化**:欧氏近似已足够的任务硬上测地线,增加复杂度和奇异点风险
45
+
46
+ ## 深入参考
47
+
48
+ - 蒸馏稿:references/books/differential-geometry.md(Ch 13 §13.4 Geodesics, §13.11 Rauch Comparison)
49
+ - 蒸馏稿:references/books/smooth-manifolds.md(Ch 20 The Exponential Map)
50
+ - 原书:Jeffrey M. Lee, *Manifolds and Differential Geometry*, §13.4 Geodesics
51
+ - 原书:John M. Lee, *Introduction to Smooth Manifolds*, Ch 20(指数映射、retraction 原型)
@@ -0,0 +1,47 @@
1
+ # Manifold
2
+
3
+ ## Minimal Definition
4
+
5
+ A manifold is a topological space that is locally homeomorphic to Euclidean space $\mathbb{R}^n$. A smooth manifold further requires that the transition maps $\phi_\beta \circ \phi_\alpha^{-1}$ between coordinate charts are $C^\infty$-smooth, enabling calculus to be performed on curved spaces.
6
+
7
+ ## Core Formulas
8
+
9
+ - Coordinate chart: $(U_\alpha, \phi_\alpha)$, where $\phi_\alpha: U_\alpha \to \mathbb{R}^n$ is a homeomorphism
10
+ - Smooth transition map: $\phi_\beta \circ \phi_\alpha^{-1}: \phi_\alpha(U_\alpha \cap U_\beta) \to \phi_\beta(U_\alpha \cap U_\beta) \in C^\infty$
11
+ - Whitney Embedding Theorem: An $n$-dimensional manifold can be embedded in $\mathbb{R}^{2n}$
12
+
13
+ ## Applicable Problems
14
+
15
+ - Data naturally resides in non-Euclidean spaces: rotations SO(3), covariance matrices SPD(n), directional data $S^2$, graphs and meshes
16
+ - Parameters have geometric constraints (orthogonality, unit norm, low rank) that require identifying the constraint set as a submanifold
17
+ - Latent space geometric modeling: interpolation, clustering, and nearest-neighbor search must respect the intrinsic curved structure of the data
18
+ - Dimensionality reduction and embedding: the manifold hypothesis assumes high-dimensional data lies on a low-dimensional manifold
19
+
20
+ ## AI Design Translation
21
+
22
+ - **Manifold optimizer (Riemannian SGD/Adam)**: Project gradients onto the tangent space and retract back to the manifold, replacing the ad-hoc patchwork of projected gradient descent
23
+ - **Latent space geometry module**: Use manifold structure in the VAE/GAN latent space for geodesic interpolation, replacing Euclidean linear interpolation
24
+ - **Constraint reparameterization layer**: Encode orthogonality/SPD/unit-norm constraints as manifold parameterizations (e.g., Cayley transform, matrix exponential), so outputs naturally satisfy constraints
25
+ - **Manifold-hypothesis-driven architecture design**: Use manifold dimension estimation to guide latent space dimension selection, avoiding the curse of dimensionality
26
+
27
+ ## Engineering Feasibility
28
+
29
+ Moderate GPU friendliness. Coordinate chart transformations are element-wise maps (parallelizable), but the core bottleneck lies in:
30
+ - Transition maps with closed-form expressions (sphere, hyperbolic space): can be directly tensorized as batched per-sample computations, GPU-friendly
31
+ - Transition maps requiring iterative solvers (general manifolds): serial dependencies, not tensorizable, GPU-unfriendly
32
+ - Partition of unity involves locally weighted sums, expressible as sparse matmul
33
+ - Key operation complexity depends on the specific manifold: simple manifolds $O(1)$/sample, complex manifolds may be $O(n^3)$
34
+
35
+ ## Risks and Failure Conditions
36
+
37
+ - **Global chart illusion**: Attempting to cover the entire manifold with a single parameterization inevitably introduces singularities (e.g., gimbal lock with Euler angles); an atlas or redundant parameterization is required
38
+ - **Manifold hypothesis abuse**: Applying manifold structure to data that actually lives in flat Euclidean space is pure over-engineering
39
+ - **Low-precision instability**: Matrix exp/log/eig in coordinate transformations becomes catastrophically unstable under fp16/bf16, often silently diverging
40
+ - **Dimension estimation errors**: The Whitney Embedding Theorem provides an upper bound of $2n$; practical embedding dimension selection lacks theoretical guidance
41
+
42
+ ## Further References
43
+
44
+ - Distillation notes: references/books/smooth-manifolds.md (Ch 1-2 Smooth Manifolds / Smooth Maps)
45
+ - Distillation notes: references/books/differential-geometry.md (Ch 1-2 Differentiable Manifolds / The Tangent Structure)
46
+ - Original text: John M. Lee, *Introduction to Smooth Manifolds*, 2nd Edition, Ch 1-2 (topological manifolds, smooth structures, partition of unity)
47
+ - Original text: Jeffrey M. Lee, *Manifolds and Differential Geometry*, Ch 1-2
@@ -0,0 +1,47 @@
1
+ # 流形 (Manifold)
2
+
3
+ ## 最小定义
4
+
5
+ 流形是局部同胚于欧氏空间 $\mathbb{R}^n$ 的拓扑空间。光滑流形进一步要求坐标卡之间的转移映射 $\phi_\beta \circ \phi_\alpha^{-1}$ 是 $C^\infty$ 光滑的,使得微积分可以在弯曲空间上进行。
6
+
7
+ ## 核心公式
8
+
9
+ - 坐标卡:$(U_\alpha, \phi_\alpha)$,其中 $\phi_\alpha: U_\alpha \to \mathbb{R}^n$ 是同胚
10
+ - 光滑转移映射:$\phi_\beta \circ \phi_\alpha^{-1}: \phi_\alpha(U_\alpha \cap U_\beta) \to \phi_\beta(U_\alpha \cap U_\beta) \in C^\infty$
11
+ - Whitney 嵌入定理:$n$ 维流形可嵌入 $\mathbb{R}^{2n}$
12
+
13
+ ## 适用问题
14
+
15
+ - 数据本身住在非欧空间:旋转 SO(3)、协方差矩阵 SPD(n)、方向数据 $S^2$、图与网格
16
+ - 参数有几何约束(正交、单位范数、低秩),需要把约束集识别为子流形
17
+ - 隐空间几何建模:插值、聚类、最近邻需要尊重数据的内在弯曲结构
18
+ - 降维与嵌入:高维数据的低维流形假设(manifold hypothesis)
19
+
20
+ ## AI 设计翻译
21
+
22
+ - **流形优化器(Riemannian SGD/Adam)**:梯度投影到切空间 + retraction 回流形,替代投影梯度下降的 ad-hoc 修补
23
+ - **隐空间几何模块**:在 VAE/GAN 的 latent space 中用流形结构做测地插值,替代欧氏线性插值
24
+ - **约束重参数化层**:将正交/SPD/单位范数约束编码为流形参数化(如 Cayley 变换、矩阵指数),输出天然满足约束
25
+ - **流形假设驱动的架构设计**:用流形维数估计指导隐空间维度选择,避免维度诅咒
26
+
27
+ ## 工程可行性
28
+
29
+ GPU 友好度中等。坐标卡变换本身是逐元素的映射(可并行),但核心瓶颈在于:
30
+ - 转移映射若为闭式(球面、双曲空间):可直接张量化,batched 逐样本计算,GPU 友好
31
+ - 转移映射若需迭代求解(一般流形):串行依赖、不可张量化,GPU 不友好
32
+ - 单位分解(partition of unity)涉及局部加权求和,可表达为稀疏 matmul
33
+ - 关键操作复杂度取决于具体流形:简单流形 $O(1)$/样本,复杂流形可能 $O(n^3)$
34
+
35
+ ## 风险与失效条件
36
+
37
+ - **全局坐标卡幻觉**:试图用单一参数化覆盖整个流形必有奇点(如欧拉角的 gimbal lock),需要 atlas / 冗余参数化
38
+ - **流形假设滥用**:数据实际分布在平坦欧氏空间时硬套流形结构,纯属过度工程
39
+ - **低精度不稳定**:坐标变换中的矩阵 exp/log/eig 在 fp16/bf16 下灾难性不稳定,常静默发散
40
+ - **维数估计错误**:Whitney 嵌入定理给出上界 $2n$,实际嵌入维数选择缺乏理论指导
41
+
42
+ ## 深入参考
43
+
44
+ - 蒸馏稿:references/books/smooth-manifolds.md(Ch 1-2 Smooth Manifolds / Smooth Maps)
45
+ - 蒸馏稿:references/books/differential-geometry.md(Ch 1-2 Differentiable Manifolds / The Tangent Structure)
46
+ - 原书:John M. Lee, *Introduction to Smooth Manifolds*, 2nd Edition, Ch 1-2(拓扑流形、光滑结构、单位分解)
47
+ - 原书:Jeffrey M. Lee, *Manifolds and Differential Geometry*, Ch 1-2
@@ -0,0 +1,50 @@
1
+ # Metric Tensor
2
+
3
+ ## Minimal Definition
4
+
5
+ The metric tensor $g$ is a positive-definite symmetric bilinear form $g_p: T_pM \times T_pM \to \mathbb{R}$ on the tangent space $T_pM$ at each point $p$ of a manifold, defining inner products, lengths, angles, and distances on the manifold. It is the precise quantification of "what is close to what" and "which direction is steepest."
6
+
7
+ ## Core Formulas
8
+
9
+ - Inner product: $\langle u, v \rangle_g = u^T g_p v = \sum_{ij} g_{ij} u^i v^j$
10
+ - Arc length element: $ds^2 = \sum_{ij} g_{ij} dx^i dx^j$
11
+ - Geodesic distance: $d(p,q) = \inf_\gamma \int_0^1 \sqrt{g_{\gamma(t)}(\dot\gamma, \dot\gamma)} \, dt$
12
+ - Musical isomorphism (index raising/lowering): $v^\flat = gv$ (tangent to cotangent), $\omega^\sharp = g^{-1}\omega$ (cotangent to tangent)
13
+ - Fisher-Rao metric: $g_{ij}(\theta) = \mathbb{E}_{p_\theta}\left[\frac{\partial \log p_\theta}{\partial \theta^i} \frac{\partial \log p_\theta}{\partial \theta^j}\right]$
14
+
15
+ ## Applicable Problems
16
+
17
+ - Parameter space is non-flat and non-Euclidean: the natural metric on families of probability distributions is the Fisher information matrix
18
+ - Slow optimization convergence: poor conditioning arises from metric mismatch, remedied by preconditioning with the natural gradient $g^{-1}\nabla L$
19
+ - Distance/similarity must adapt to data geometry: metric learning is essentially learning a $g$
20
+ - Volume computation and density estimation: $\sqrt{\det g}$ gives the volume form on the manifold
21
+
22
+ ## AI Design Translation
23
+
24
+ - **Natural gradient / K-FAC optimizer**: $F^{-1}\nabla L$ where $F$ is the Fisher metric; K-FAC uses the Kronecker factorization $F \approx A \otimes B$ to reduce inversion to two small matrix inversions, and the preconditioning reduces to a GEMM chain
25
+ - **Learnable metric layer**: Parameterize $g = L^T L$ (Cholesky) to learn a task-specific Riemannian metric for metric learning and contrastive learning
26
+ - **Information-geometric regularization**: Replace Euclidean $\|d\theta\|^2$ with the Fisher-Rao distance $\|d\theta\|_F^2 = d\theta^T F d\theta$, making regularization invariant to reparameterization
27
+ - **Fisher-aware learning rate scheduling**: Use $\|g^{-1}\nabla L\|_g$ as the "geometrically correct" gradient norm to guide learning rate selection
28
+
29
+ ## Engineering Feasibility
30
+
31
+ GPU friendliness depends on the structure of the metric:
32
+ - **Diagonal metric** $g = \text{diag}(g_1, \ldots, g_n)$: element-wise multiply/divide, $O(n)$, perfectly GPU-friendly
33
+ - **Kronecker-factored** $g = A \otimes B$: $(A\otimes B)^{-1} = A^{-1}\otimes B^{-1}$, small matrix inversions + GEMM chain, the core trick of K-FAC, GPU-feasible
34
+ - **Block-diagonal metric**: per-block independent inversion, batched small matrix operations, GPU-friendly
35
+ - **Full dense metric**: $n \times n$ matrix inversion $O(n^3)$ + memory $O(n^2)$; with parameter count $N \sim 10^9$, this is immediately ruled out
36
+ - Low-precision risk: the Fisher matrix is often ill-conditioned; inversion under fp16 catastrophically amplifies errors -- **must add damping $F + \lambda I$ and invert in fp32**
37
+
38
+ ## Risks and Failure Conditions
39
+
40
+ - **Ill-conditioned metric matrix**: The condition number of the Fisher matrix can exceed $10^6$; low-precision inversion results are entirely noise
41
+ - **Materializing the full metric matrix**: An $N \times N$ matrix ($N \sim 10^9$) requires $\sim 4$ PB of memory, making materialization impossible
42
+ - **Metric-task mismatch**: The Fisher metric assumes the probabilistic model is correct; under model misspecification, the natural gradient can perform worse than SGD
43
+ - **Dynamic metric update overhead**: The Fisher matrix changes with the parameters; statistical noise from re-estimation at each step may offset preconditioning benefits
44
+
45
+ ## Further References
46
+
47
+ - Distillation notes: references/books/differential-geometry.md (Ch 7 Section 7.6 Metric Tensors, Ch 13 Section 13.1 Levi-Civita)
48
+ - Distillation notes: references/books/smooth-manifolds.md (Ch 13 Riemannian Metrics)
49
+ - Original text: Jeffrey M. Lee, *Manifolds and Differential Geometry*, Section 7.6 Metric Tensors
50
+ - Original text: John M. Lee, *Introduction to Smooth Manifolds*, Ch 13 (Riemannian metrics, musical isomorphisms sharp/flat)
@@ -0,0 +1,50 @@
1
+ # 度量张量 (Metric Tensor)
2
+
3
+ ## 最小定义
4
+
5
+ 度量张量 $g$ 是流形上每点 $p$ 处切空间 $T_pM$ 上的一个正定对称双线性形式 $g_p: T_pM \times T_pM \to \mathbb{R}$,定义了流形上的内积、长度、角度和距离。它是"谁离谁近、什么方向最陡"的精确量化。
6
+
7
+ ## 核心公式
8
+
9
+ - 内积:$\langle u, v \rangle_g = u^T g_p v = \sum_{ij} g_{ij} u^i v^j$
10
+ - 弧长:$ds^2 = \sum_{ij} g_{ij} dx^i dx^j$
11
+ - 测地距离:$d(p,q) = \inf_\gamma \int_0^1 \sqrt{g_{\gamma(t)}(\dot\gamma, \dot\gamma)} \, dt$
12
+ - 升降指标(musical isomorphism):$v^\flat = gv$(切→余切),$\omega^\sharp = g^{-1}\omega$(余切→切)
13
+ - Fisher-Rao 度量:$g_{ij}(\theta) = \mathbb{E}_{p_\theta}\left[\frac{\partial \log p_\theta}{\partial \theta^i} \frac{\partial \log p_\theta}{\partial \theta^j}\right]$
14
+
15
+ ## 适用问题
16
+
17
+ - 参数空间非平坦欧氏:概率分布族上的自然度量是 Fisher 信息矩阵
18
+ - 优化收敛缓慢:条件数差源于度量不匹配,用自然梯度 $g^{-1}\nabla L$ 预条件
19
+ - 距离/相似度需要适应数据几何:度量学习本质是学习一个 $g$
20
+ - 体积计算与密度估计:$\sqrt{\det g}$ 给出流形上的体积形式
21
+
22
+ ## AI 设计翻译
23
+
24
+ - **自然梯度/K-FAC 优化器**:$F^{-1}\nabla L$ 其中 $F$ 是 Fisher 度量;K-FAC 用 Kronecker 因子化 $F \approx A \otimes B$ 使求逆变为两个小矩阵求逆,预条件作用退化为 GEMM 链
25
+ - **可学习度量层**:参数化 $g = L^T L$(Cholesky),学习任务相关的黎曼度量,用于度量学习和对比学习
26
+ - **信息几何正则化**:用 Fisher-Rao 距离 $\|d\theta\|_F^2 = d\theta^T F d\theta$ 替代欧氏 $\|d\theta\|^2$,使正则化对重参数化不变
27
+ - **Fisher 感知的学习率调度**:$\|g^{-1}\nabla L\|_g$ 作为"几何正确"的梯度范数,指导学习率选择
28
+
29
+ ## 工程可行性
30
+
31
+ GPU 友好度取决于度量的结构化程度:
32
+ - **对角度量** $g = \text{diag}(g_1, \ldots, g_n)$:逐元素乘除,$O(n)$,完美 GPU 友好
33
+ - **Kronecker 因子化** $g = A \otimes B$:$(A\otimes B)^{-1} = A^{-1}\otimes B^{-1}$,小矩阵求逆 + GEMM 链,K-FAC 的核心 trick,GPU 可行
34
+ - **块对角度量**:逐块独立求逆,batched 小矩阵运算,GPU 友好
35
+ - **全稠密度量**:$n \times n$ 矩阵求逆 $O(n^3)$ + 显存 $O(n^2)$,参数量 $N \sim 10^9$ 时直接出局
36
+ - 低精度风险:Fisher 矩阵常病态,fp16 下求逆灾难性放大误差,**必须加阻尼 $F + \lambda I$ 并在 fp32 下求逆**
37
+
38
+ ## 风险与失效条件
39
+
40
+ - **度量矩阵病态**:Fisher 矩阵的条件数可达 $10^6$ 以上,低精度求逆结果全是噪声
41
+ - **物化全度量矩阵**:$N \times N$ 矩阵($N \sim 10^9$)需 $\sim 4$ PB 显存,不可能物化
42
+ - **度量与任务不匹配**:Fisher 度量假设概率模型正确,模型误设定时自然梯度可能比 SGD 更差
43
+ - **动态度量更新开销**:Fisher 矩阵随参数变化,每步重新估计的统计噪声可能抵消预条件收益
44
+
45
+ ## 深入参考
46
+
47
+ - 蒸馏稿:references/books/differential-geometry.md(Ch 7 §7.6 Metric Tensors, Ch 13 §13.1 Levi-Civita)
48
+ - 蒸馏稿:references/books/smooth-manifolds.md(Ch 13 Riemannian Metrics)
49
+ - 原书:Jeffrey M. Lee, *Manifolds and Differential Geometry*, §7.6 Metric Tensors
50
+ - 原书:John M. Lee, *Introduction to Smooth Manifolds*, Ch 13(黎曼度量、升降指标 ♯/♭)