Computing Augustin Information via Hybrid Geodesically Convex Optimization
Guan-Ren Wang, Chung-En Tsai, Hao-Chung Cheng, Yen-Huan Li
TL;DR
The paper tackles computing the order-$\alpha$ Augustin information $I_\alpha(P_X, P_{Y|X}) = \min_{Q_Y} \mathbb{E}_X D_\alpha(P_{Y|X}(\cdot|X) \| Q_Y)$ by embedding the problem in a Riemannian framework with the Poincaré metric. It introduces a hybrid analysis where the objective is geodesically convex under the Euclidean structure but geodesically smooth under the Poincaré metric, and proves a non-asymptotic $\mathcal{O}(1/T)$ convergence rate for Riemannian gradient descent with step $\eta=1/L$. The approach yields the first finite-time guarantee for all $\alpha>0$ and is validated numerically, showing empirical efficiency relative to fixed-point and other gradient-based methods. The results open pathways to extending the framework to quantum Rényi divergences and other problems with similar geometric structures, enabling robust non-asymptotic guarantees in geodesically non-Euclidean settings.
Abstract
We propose a Riemannian gradient descent with the Poincaré metric to compute the order-$α$ Augustin information, a widely used quantity for characterizing exponential error behaviors in information theory. We prove that the algorithm converges to the optimum at a rate of $\mathcal{O}(1 / T)$. As far as we know, this is the first algorithm with a non-asymptotic optimization error guarantee for all positive orders. Numerical experimental results demonstrate the empirical efficiency of the algorithm. Our result is based on a novel hybrid analysis of Riemannian gradient descent for functions that are geodesically convex in a Riemannian metric and geodesically smooth in another.
