arXiv:2608.22746stat.MLcs.LG2026-08

用生成模型提升分布鲁棒检测的效率与精度

Generative Neural Networks for Sinkhorn Distributionally Robust Hypothesis Testing

论文配图:Generative Neural Networks for Sinkhorn Distributionally Robust Hypothesis Testing
图 1 · 摘自论文原文
  • 通过生成对抗分布学习最不利情形,避免求解大规模优化问题
  • 在不同样本量和维度下均实现更高准确率与鲁棒性
  • 适合需要高效、可扩展鲁棒检测的机器学习应用

本文研究基于Sinkhorn差异的分布鲁棒假设检验(SDRHT)问题,旨在寻找对经验分布为中心的模糊集内最不利分布具有鲁棒性的检测器。现有方法需求解大规模锥规划,难以扩展。为此,我们提出一种生成式框架,可学习最不利分布,支持高效训练与端到端采样。针对基于Sinkhorn差异的模糊集,我们首次推导出其等价的条件KL散度表示,基于核平滑参考分布。该性质使我们证明了约束与无约束极小极大SDRHT形式的强对偶性。结合闭式最优检测器与Brenier定理,我们将极大极小对偶形式重写为凸势函数最大化问题,其梯度表征核平滑分布与其最不利对应物之间的可逆映射。我们使用带有随机梯度估计器的超输入凸神经网络(HyCNNs)高效逼近这些势函数,并证明了HyCNNs的表示能力及其诱导映射的分布普适性。数值实验表明,该方法在不同样本规模与维度下均优于传统SDRHT方法,且避免了计算瓶颈。

原文摘要 · Abstract (English)

This paper studies the Sinkhorn distributionally robust hypothesis testing (SDRHT) problem, seeking a robust detector against least-favorable distributions in Sinkhorn discrepancy-based ambiguity sets centered at the empirical distributions. Existing approaches solve this problem by solving large-scale conic programs, which are not scalable. To overcome this, we propose a generative framework that learns least-favorable distributions and supports efficient training and end-to-end sampling. For the Sinkhorn discrepancy-based ambiguity sets, we first derive an equivalent conditional-KL-divergence representation with respect to kernel-smoothed reference distributions. This property allows us to prove strong duality for both constrained and unconstrained minimax SDRHT formulations. Based on the closed-form optimal detector and Brenier's theorem, we reformulate the max-min dual formulation as a maximization problem over convex potentials whose gradients characterize invertible transport maps between kernel-smoothed distributions and their least-favorable counterparts. We efficiently approximate these potentials using Hyper Input Convex Neural Networks (HyCNNs) equipped with stochastic gradient estimators and prove the representation power of HyCNNs and the distributional universality of their induced transport maps. Numerical results show that the proposed method achieves superior accuracy and robustness across different sample sizes and dimensions, while avoiding the scalability limitations of classical SDRHT methods.

生成模型分布鲁棒假设检验神经网络

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