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

摘要

arXiv:2502.01889v1 Announce Type: 多领域交叉 摘要:最优传输(OT)理论旨在确定从空间X到空间Y的传输映射$T:X \to Y$,使源测度$P$传输到目标测度$Q$,同时最小化$x$与其像$T(\mathbf{x})$之间的成本$c(\mathbf{x}, T(\mathbf{x}))$。基于输入凸神经网络OT求解器,并结合位移稀疏映射的概念,我们引入了最小最大韦尔斯特拉斯形式的稀疏惩罚,促进位移向量$\Delta(\mathbf{x}) := T(\mathbf{x}) - \mathbf{x}$的稀疏性,并增强了该映射的可解释性。然而,增加稀疏性往往降低了可行性,导致$T_{\#}(P)$与目标测度的偏离更大。在低维设置中,我们提出了一种启发式框架来平衡稀疏性和可行性之间的权衡,在训练过程中动态调整稀疏性强度参数。在高维设置中,我们直接通过限制位移向量的维数来约束,即强制$\dim(\Delta(\mathbf{x})) \leq l$,其中对于$X \subseteq \mathbb{R}^d$,$l < d$。在满足此约束的映射中,我们旨在识别最可行的一个。这一目标可以通过适应我们低维启发式框架来有效实现,而无需进行维数降解。我们在合成sc-RNA和真实4i细胞扰动数据集上验证了该方法,表明其优于现有方法。