LLM2D
位移稀疏神经最优传输
Displacement-Sparse Neural Optimal Transport
作者: Peter Chen, Yue Xie, Qingpeng Zhang
发布日期: 2/5/2025
arXiv ID: 2502.01889

摘要

arXiv:2502.01889v1 宣告类型: cross 摘要: 最优传输(OT)理论旨在确定从源测度 \(P\) 转运到目标测度 \(Q\) 的映射 \(T: X \to Y\),以最小化 \(\mathbf{x}\) 与其像点 \(T(\mathbf{x})\) 之间的代价 \(c(\mathbf{x}, T(\mathbf{x}))\)。基于输入可凸神经网络 OT 解决器,并结合位移稀疏映射的概念,我们在最小最大 Wasserstein 表述中引入稀疏性惩罚,促进位移向量 \(\Delta(\mathbf{x}) := T(\mathbf{x}) - \mathbf{x}\) 的稀疏性,并增强所得映射的可解释性。然而,增加稀疏性通常会降低可行性,导致 \(T_{\#}(P)\) 更显著地偏离目标测度。在低维情况下,我们提出了一种启发式框架来平衡稀疏性和可行性之间的权衡,通过在训练过程中动态调整稀疏性强度参数来实现。在高维情况下,我们直接通过约束位移向量的维度来限制,即 \(\dim(\Delta(\mathbf{x})) \leq l\),其中 \(l < d\) 且 \(X \subseteq \mathbb{R}^d\)。在满足此约束的映射中,我们旨在识别最可行的一个。这一目标可以通过适应我们的低维启发式框架来有效实现,而无需降低维度。我们在合成 sc-RNA 和真实 4i 细胞扰动数据集上验证了该方法,展示了相对于现有方法的改进。