stat.MLOct 6, 2026

Trust-Region Optimization for Smooth Potential-Interaction Energies in Wasserstein Space

Authors: You Wan, Ting Gao, Jinqiao Duan

Organizations: School of Mathematics, Nanjing University, Nanjing, 210093, China · School of Sciences, Great Bay University, Dongguan 523000, China · Guangdong Provincial Key Laboratory of Mathematical and Neural Dynamical Systems, Great Bay University, Dongguan 523000, China · School of Mathematics and Statistics, Huazhong University of Science and Technology, Wuhan 430074, China · Center for Mathematical Science, Huazhong University of Science and Technology, Wuhan 430074, China · Steklov-Wuhan Institute for Mathematical Exploration, Huazhong University of Science and Technology, Wuhan 430074, China

Abstract

Finding low-energy configurations of interacting particles and approximating probability distributions lead to the minimization of potential-interaction energies in Wasserstein space. These energies can be nonconvex, making it important to exploit second-order information while controlling the reliability of local approximations. We study trust-region optimization of smooth potential-interaction energies on the Wasserstein space of probability measures with finite second moment. The method uses a quadratic model along pushforward curves, an L2(ρ)L^2(ρ) step radius, and a Steihaug-Toint subsolver with an explicit self-adjoint second-variation operator. A ratio test determines acceptance and guides the radius update. Under a lower energy bound and globally bounded Hessians of the potential and interaction kernel, we prove that the objective is nonincreasing, the Wasserstein-gradient norms converge to zero, and an ε\varepsilon-stationary iterate is reached within O(ε−2)O(\varepsilon^{-2}) total outer trials, including rejected trials. If the potential is quadratically coercive, every weak accumulation point is stationary. The analysis applies to arbitrary initial measures with finite second moment. For empirical measures, the iteration is a finite-dimensional trust-region method in the L2(ρN)L^2(ρ_N) inner product, with complexity constants independent of particle number and dimension when the initial objective gaps are uniformly bounded. Numerical experiments include a smooth soft-particle energy, maximum-mean-discrepancy minimization for non-Gaussian targets, component ablations, and scaling studies in particle number and dimension.

Figures & tables

Explore similar work

CardsList
  1. A stochastic smoothing framework for nonconvex-nonconcave minEmax problems with applications to Wasserstein distributionally robust optimization

    Feb 24, 2025Wei Liu, Muhammad Khan, Gabriel Mancino-Ball +1Stochastic Convex OptimizationDistributionally-Robust Optimization

  2. Difference of Convex Programming in the Wasserstein Space with Applications to MMD Optimization

    Jun 26, 2026Clément Bonet, Pierre-Cyril Aubin-Frankowski, Youssef MrouehWasserstein DistanceMaximum Mean Discrepancy