精读笔记
Problem Setting
【Riemannian Archetypal Analysis: Interpretable non-linear data analysis on deformed star distributions】（arXiv preprint / 2026）
这篇论文不是简单提出一个 nonlinear AA 变体，而是在处理一个更具体的问题：如何让 AA 的 archetype、convex combination、projection、weight interpretation 在非线性数据流形上仍然有几何语义。经典 AA 的解释性来自欧氏凸包：archetypes 是极端点，样本是凸组合，权重有清楚含义。但这个结构一旦遇到强非线性数据 manifold，就变成错误几何。神经 AA 的问题相反：表达力有了，但 latent convexity 往往只是 architecture 约束，不保证对应原空间中的合理插值或高密度路径。
真正困难点是同时满足三件事：非线性表达、archetypal 可解释性、几何一致性。以前方法通常只能满足其中两项。AAnet / Deep AA 这类方法可以学习非线性表示，但 latent space 的距离和插值没有统计几何保证；normalizing-flow pullback geometry 可以提供 geodesic，但已有理论主要对应 deformed Gaussian，更接近非线性 PCA，而不是 AA 所需的 star / polytope geometry。本文的关键矛盾是：AA 需要 extremal / star-shaped inductive bias，而现有 Riemannian flow geometry 默认的是 Gaussian-like central geometry。
Motivation
作者的动机很明确：如果 PCA 的 Riemannian analogue 对应 deformed Gaussian，那么 AA 的 Riemannian analogue 应该对应 deformed star distribution。这个观察是论文最重要的出发点。AA 本质上不是在找一个低维线性子空间，而是在找由极端点张成的 polytope；在概率几何上，这更接近沿若干方向延展的 star-shaped density，而不是各向同性 Gaussian。
已有路线缺的是“分布—几何—archetype”三者的一致绑定。神经 AA 缺分布几何：decoder 能生成路径，但路径是否走在高似然区域没有保证。现有 pullback Riemannian learning 缺 archetypal geometry：flow 可以 flatten manifold，但 flatten 到 Gaussian 时，archetype corners 被抹掉。本文的核心缺口就是：没有一个可计算的 Riemannian structure，使得 archetypal convex hull 在原空间中变成 geodesically meaningful 的对象。
Core Idea
论文的核心思想是把 archetypal convexity 放到一个学习得到的 pullback chart 里：在 φ-space 中，archetypes 的 convex hull 是普通欧氏凸包；在原空间中，它被 φ^{-1} 拉回成一个弯曲的 geodesically convex manifold with corners。于是非线性 AA 不再是“latent decoder 上的凸组合”，而是“pullback Riemannian geometry 下的 barycentric hull”。这个建模方式改变了 AA 的基本几何对象：从 Euclidean polytope 变为 deformed polytope / star manifold。
新的 inductive bias 是 deformed star distribution。ϕ 负责把原数据拉到一个 latent star；ρ 负责描述不同方向的 radial extension；χ_ρ 把这个 star 规整化；ψ_v 用 concave radial transform 把 geodesics 往中心拉，避免在 star arms 之间插值时向外甩。理论上，这样构造出的 geodesics 上 negative log-likelihood 强凸，因此路径不会穿过低密度区域。这一点使它区别于普通 neural latent interpolation：这里的 interpolation 是由 density geometry 支撑的，而不是由 decoder 的视觉平滑性支撑的。
Method
方法的机制层面可以压缩为四个必要部件。
1. Deformed star density：论文定义 p_{ϕ,ρ}，其中 ϕ 是常 Jacobian determinant diffeomorphism，ρ 是球面上的正 radial function。它解决的是 AA 缺少统计模型的问题。ρ 大的方向对应 star arms / archetypal directions，因此 archetypes 不再只是优化得到的极端样本，而被解释为高径向延展方向附近的代表点。
2. Pullback geometry φ = ψ_v ∘ χ_ρ ∘ ϕ：χ_ρ 负责把方向尺度归一化，使 latent star 在几何上可被当作规整对象；ψ_v 是额外 radial reparameterization，用来减少 geodesic swinging。Theorem 1 的作用不是给出一个漂亮性质，而是保证该几何下 archetype-to-archetype 的路径具有高似然解释。
3. RAM projection：论文把 AA 的 convex hull 替换为 archetypes 的 Riemannian barycentric hull，即所有加权 Riemannian barycentres 的集合。由于 Euclidean pullback 下 barycentre 可写成 φ^{-1}(Σ g_j φ(v_j))，RAM 最终变成 simplex 上的非凸优化。这个转换是可计算性的关键，否则一般 Riemannian barycentre 会使问题不可用。
4. Relaxed RAM + refinement + iso correction：relaxed RAM 在 φ-space 中做凸投影，给非凸 RAM 提供初始化；RAM refinement 则用原空间 ℓ2 reconstruction objective 修正 relaxed projection 的偏差。iso correction 不是 cosmetic，它修正了 pullback log length 与原空间 arc length 不一致的问题，使 weight 更接近可解释的 archetypal proximity。
Key Insight / Why It Works
这篇论文真正有效的原因不是“用了 Riemannian geometry”，而是找到了一个让 AA 结构在非线性空间中仍然保持凸可计算性的 chart。只要 φ 能把数据整理成 star-like latent distribution，那么 archetypal hull 就退化为 φ-space 中的普通 convex hull；所有 geodesic / barycentric / projection 操作都可以借这个 chart 实现。核心贡献在于把 nonlinear AA 的解释性问题转成了 pullback geometry 的选择问题。
最实质的 insight 是：AA 的非线性推广不应该从 autoencoder latent simplex 开始，而应该从适合 AA 的 density family 开始。deformed Gaussian 对 PCA 合理，但对 AA 不合理；deformed star 才对应 archetypes。这个视角可迁移性很强：很多 factorization 方法的 Riemannian extension 可能都需要先找其“自然分布族”，而不是直接把线性代数操作塞进 neural latent space。
ψ_v 和 iso correction 是重要辅助，但不是主创新。ψ_v 主要解决 geodesic outward swing，属于几何 regularization；iso correction 主要修正时间参数和权重解释，属于解释层面的 calibration。真正的机制是 χ_ρ + pullback convex hull。学习流程中的 ellipsoid radial functions 更像 engineering：它让系统跑起来，但不是最优统计学习。MNIST 上的收益也很可能混合了三类来源：normalizing flow flattening、star-shaped inductive bias、latent AA 的数据覆盖。文中没有充分 disentangle 这些来源。
需要直说的是：full MNIST 上优化失败说明 RAM 目前不是一个成熟 scalable solver，而是一个几何 formulation 加初步数值方案。分类结果好坏很大程度受 projection 是否收敛影响，因此不能把分类提升完全归因于 geometry-aware weights。
Relation To Prior Work
最接近的三条线是 classical AA / neural AA、normalizing-flow pullback Riemannian geometry、Riemannian/iso-Riemannian optimization。
相对 classical AA，本文的真正新增不是“把凸组合换成 geodesic convex combination”这么简单，而是证明在 Euclidean pullback 下这个集合可写成 φ^{-1} of latent convex hull，因此仍然保留 AA 的 barycentric 结构。相对 AAnet / Deep AA，本文最大的差异是 latent geometry 不是任意 learned representation，而由显式 density 和 pullback metric 诱导；因此 interpolation 和 weights 有更强的几何语义。
相对已有 flow-based Riemannian geometry，本文从 deformed Gaussian 扩展到 deformed star。这个扩展不是纯形式变化，因为 χ_ρ 的 Jacobian 非常数，带来 geodesic stability、速度参数化和优化问题；也正因此需要 ψ_v 和 iso correction。可以说它属于“data-driven pullback geometry + interpretable factorization”的技术谱系。
看似新的部分中，有些是已有思想重组：flow flattening、latent AA、ellipsoidal radial approximation、proximal simplex optimization 都不是新东西。实质创新在于把它们组织成一个具有 AA 语义的 Riemannian model，并给出 high-likelihood geodesic 与 RAM constraint set 的理论闭合。
Dataset / Evaluation
实验覆盖 synthetic deformed star、single-digit MNIST、full MNIST。synthetic 部分最有说服力，因为数据确实符合模型假设，能直接观察 geodesic 是否贴合 high-density star arms、RAM 是否优于 relaxed RAM、iso correction 是否修正权重。MNIST 的作用更多是 sanity check：证明该机制在高维图像上至少能产生视觉上合理的插值和 denoising。
但 evaluation 对核心 claim 的支持是不完整的。论文 claim 是 interpretable nonlinear data analysis；实验主要展示插值图、投影图和简单分类权重，没有系统比较 nonlinear AA baselines，也没有量化几何 fidelity / likelihood path / denoising robustness。full MNIST 中 RAM 未能稳定收敛，这反而说明当前方法的计算部分还不成熟。没有真实世界科学数据或需要 archetype 解释的任务，例如 single-cell / hyperspectral，这使得“解释性”的验证偏弱。
benchmark 没有明显 leakage 问题，但存在 evaluation bias：很多结论来自视觉判断，尤其是 MNIST interpolation 是否“更合理”主观性较强。分类实验也不是强 benchmark，因为 projection convergence 本身成为 confounder。
Limitation
第一，方法强依赖 deformed star assumption。数据必须能被常 Jacobian flow ϕ 整理成 latent star，再由 ρ 描述方向延展。如果真实数据是多中心、多拓扑分支、环状或层级结构，star geometry 可能只是强行解释。文中未充分说明这种假设在复杂数据上的适用边界。
第二，ρ 的学习目前不是 principled。作者明确指出直接 NLL 训练会遇到高维球面积分和 ρ^d 数值问题，于是采用 flow + latent AA + ellipsoid radial construction。这是合理工程方案，但统计上是 suboptimal。换言之，论文最漂亮的理论建立在 p_{ϕ,ρ} 已知或学得很好的前提上，而实际学习部分并没有达到同等理论强度。
第三，RAM 优化是实际瓶颈。非凸 RAM objective 通过 relaxed RAM 初始化，但在 high-dimensional / many-archetype setting 下 line search 步长会衰减到不可用。这个问题不是调参能根治的；它反映出通过 φ^{-1} 优化时 Lipschitz constants 和局部几何扭曲非常差。作者提出 constrained iso-Riemannian optimization 是未来方向，基本等于承认当前 solver 不够。
第四，权重解释存在 gauge ambiguity。φ、ρ、archetypes 之间并非唯一；同一个数据分布可能有多种 latent star parametrization。iso correction 修正了长度解释，但没有解决 identifiability。某些 archetype weights 是否反映真实语义，还是 flow chart 的坐标效应，文中没有给出强证据。
第五，增益归因不清。普通 flow geodesic 已经显著优于线性插值；star geometry 在此基础上的增益有多少来自 radial model，有多少来自 ψ_v 的 inward bias，有多少来自 archetype selection，实验没有系统 ablation。
Takeaway
- 1. 非线性 AA 的关键不应是更强 decoder，而是给 convex combination 找到正确的几何 chart；这篇论文把这个问题表述得很清楚。
- 2. deformed star distribution 是 AA 的自然概率几何模型。
- 这个 insight 比具体算法更值得迁移：不同 interpretable factorization 方法可能都需要自己的 natural pullback density family。
- 3. RAM 的理论形式很优雅，但当前数值优化是短板。
一句话总结
这篇论文把 archetypal analysis 从神经 latent heuristic 推向了 deformed-star pullback geometry，是一次把 nonlinear AA 重新建立在统计几何模型上的尝试，但当前主要贡献是 formulation 和 insight，scalable learning / optimization 仍未真正解决。