Geometric numerical integration of the assignment flow
Abstract
The assignment flow is a smooth dynamical system that evolves on an elementary statistical manifold and performs contextual data labeling on a graph. We derive and introduce the linear assignment flow that evolves nonlinearly on the manifold, but is governed by a linear ODE on the tangent space. Various numerical schemes adapted to the mathematical structure of these two models are designed and studied, for the geometric numerical integration of both flows: embedded Runge-Kutta-Munthe-Kaas schemes for the nonlinear flow, adaptive Runge-Kutta schemes and exponential integrators for the linear flow. All algorithms are parameter free, except for setting a tolerance value that specifies adaptive step size selection by monitoring the local integration error, or fixing the dimension of the Krylov subspace approximation. These algorithms provide a basis for applying the assignment flow to machine learning scenarios beyond supervised labeling, including unsupervised labeling and learning from controlled assignment flows.
- Publication:
-
Inverse Problems
- Pub Date:
- March 2020
- DOI:
- 10.1088/1361-6420/ab2772
- arXiv:
- arXiv:1810.06970
- Bibcode:
- 2020InvPr..36c4003Z
- Keywords:
-
- image labeling;
- assignment flow;
- assignment manifold;
- geometric integration;
- adaptive step size selection;
- Krylov subspace approximation;
- Mathematics - Numerical Analysis;
- Mathematics - Dynamical Systems;
- 62H35;
- 62M40;
- 65K10;
- 68U10
- E-Print:
- doi:10.1088/1361-6420/ab2772