Riemannian Bilevel Optimization with Gradient Aggregation

Abstract

Bilevel optimization (BLO) offers a principled framework for hierarchical decision-making and has been widely applied in machine learning tasks such as hyperparameter optimization and meta-learning. While existing BLO methods are mostly developed in Euclidean spaces, many real-world problems involve structural constraints. In this paper, we propose a Riemannian bilevel optimization (RBLO) algorithm that incorporates a bilevel descent aggregation (BDA) scheme to jointly coordinate upper- and lower-level updates. Concretely, first we abstract the constraints in the BLO to a manifold structure and then transform the constrained BLO be a unconstrained RBLO problem. Second, to address limitations of existing RBLO methods, particularly the restrictive assumptions required for convergence, we reformulate the bilevel problem using smooth manifold mappings and provide a convergence analysis under the conditions of geodesic convexity and Lipschitz smoothness. Finally, we recall the multi-view hypergraph spectral clustering task, and evaluate the proposed approach on 3sources data sets. The numerical results validate the superior performance over Euclidean and manifold-based baselines.