arXiv Science⌕ Search
Home/Archive/arXiv

arXiv · 2609.00870

Stochastic Optimization of Tree Tensor Networks

Marius Willner·Maximilian Scharf·André Uschmajew·Timo Felser·Marco Trenti
View original source ↗Read source document ↗Export metadata ↓

Abstract

Tensor networks, originally developed for quantum many-body physics, are promising models for machine learning. We derive stochastic Riemannian optimizers for tree tensor networks (TTNs) on both their parameter and quotient manifolds, including adaptive and learning-rate-free schemes suitable for minibatch training. Using a hybrid CNN-TTN architecture, we evaluate the methods on Fashion-MNIST, CIFAR10, and Imagenette. The proposed optimizers achieve predictive performance comparable to unconstrained optimization while enabling numerically stable downstream compression.

Explore related subjects

math.OCcs.CVphysics.comp-ph

Keep this discovery

BibTeXRIS

Marius Willner, Maximilian Scharf, André Uschmajew, Timo Felser, Marco Trenti. 2026-09-01. Stochastic Optimization of Tree Tensor Networks. https://arxiv.org/abs/2609.00870

Cite the original work for its findings. Save a collection to share your selection of sources.

Discover connections

CIFAR-10 datasetcs.CV subjectmath.OC subjectphysics.comp-ph subject

Connections use source metadata and explicit phrase matches, not verified experimental comparisons.

KEEP EXPLORING

Related papers

arXiv2026

Constrained dynamics for searching saddle points on embedded Riemannian submanifolds of Euclidean space

Finding constrained saddle points on embedded Riemannian submanifolds of Euclidean space is significant for analyzing energy landscapes arising in physics and chemistry. Existing works exploit explicit global/local regular level-set representations of manifolds, which may be unavailable or computationally inconvenient for manifolds represented through, e.g., projectors, factorizations, or rank constraints. In this paper, we develop a constrained saddle dynamic based on embedded-submanifold geometric primitives, completely avoiding the use of explicit representations. In particular, our dynamic is formulated compactly on the Grassmann bundle of the tangent bundle. By analyzing the Grassmann bundle geometry, we rigorously establish the local linear stability of the dynamic and the local linear convergence of the resulting algorithms. Remarkably, our analysis provides the first iterate convergence result for discretized algorithms to saddle points of prescribed indices in embedded-submanifold settings. Moreover, by virtue of the Grassmann bundle formulation, we remove unnecessary nondegeneracy assumptions on the eigenvalues of the Riemannian Hessian that are present in existing works. We also point out that locating saddle points can be more ill-conditioned than finding local minimizers, and requires using nonredundant parametrizations. Finally, numerical experiments on linear eigenvalue problems and electronic excited-state calculations showcase the effectiveness of the proposed algorithms and corroborate the established local theory.

math.NA↗
arXiv2026

Efficient primal--dual splitting methods for a Poisson-constrained JKO scheme for Poisson-Nernst-Planck models

The Poisson--Nernst--Planck (PNP) equations strongly couple ionic transport and electrostatic interactions through the Poisson equation, posing substantial numerical challenges under small permittivity and complex potential boundary conditions. Underlying these equations is a natural Wasserstein gradient-flow structure, in which the Poisson equation serves as a local realization of the nonlocal electrostatic interaction energy. Exploiting this structure, we formulate each time step as a constrained convex minimization problem where the ionic continuity equations and the Poisson equation are incorporated as linear constraints, allowing the concentrations, fluxes, and electrostatic potential to be updated simultaneously. The variational structure of the scheme intrinsically guarantees the dissipation of the original free energy, mass conservation, and nonnegativity of ionic concentrations under general electrostatic boundary conditions. Moreover, the framework is structurally modular: extending from classical to modified PNP models with steric interactions and concentration-gradient corrections requires only modifying the energy functional, while all structure-preserving properties are automatically retained. To efficiently solve the resulting large-scale constrained problems, we develop preconditioned and transformed primal--dual algorithms equipped with tailored fast dual solvers, namely DCT-based direct and Schur-complement iterative methods, that exploit the coupled block structure of the PDE constraints. Numerical experiments on classical and modified PNP systems demonstrate the accuracy and structure-preserving properties of the scheme, and show that the proposed algorithms converge reliably in strongly coupled small-permittivity regimes without significant growth in computational cost.

math.NA↗
arXiv2026

Geometric integrators for adiabatically closed simple thermodynamic systems

A variational formulation for non-equilibrium thermodynamics was developed by Gay-Balmaz and Yoshimura. In a recent article, the first two authors of the present paper introduced partially cosymplectic structures as a geometric framework for thermodynamic systems, recovering the evolution equations obtained variationally. In this paper, we develop a discrete variational principle for adiabatically closed simple thermodynamic systems, which can be utilised to construct numerical integrators for the dynamics of such systems. The effectiveness of our method is illustrated with several examples.

math-ph↗