← Back to Research

Star Geometry + Regularizer Learning

Overview

In optimization-based approaches to inverse problems and statistical inference, regularizers are added to an objective to promote desired structural properties in solutions. Convex regularizers—such as the ℓ1 norm for sparsity or the nuclear norm for low rank—are computationally attractive but limited in the structures they can represent. My research studies the broader class of regularizers defined by Minkowski functionals of star bodies: compact sets that include both convex bodies and nonconvex shapes. This class is rich enough to capture complex data structure while remaining amenable to geometric analysis.

star
Figure 1: Optimal star regularizers for different Gaussian mixture distributions

Learning regularizers from data

A central question is: given data drawn from an unknown distribution, what is the optimal regularizer for recovering signals from that distribution? In joint work with Oscar Leong, Yong Sheng Soh, and Venkat Chandrasekaran, we characterize the optimal star-body regularizer for a data source and identify the geometric properties of the data distribution that determine whether the optimal regularizer is convex or nonconvex. We also study how well optimal regularizers can be learned from data using critic-based methods that use observed data and the known forward map to help guide the choice of regularizer.

Robustness to distributional uncertainty

In practice, the true data distribution is unknown and may shift between training and deployment. We study distributionally robust variants of the regularizer learning problem, in which the regularizer is chosen to perform well across an ambiguity set of distributions. We characterize optimal robust regularizers and relate robustness considerations to the regularity and convexity of the regularizer.

Related Publications

Optimal regularization under uncertainty: Distributional robustness and convexity constraints
Oscar Leong, Eliza O'Reilly, Yong Sheng Soh
Preprint, 2025
Optimal regularization for a data source
Oscar Leong, Eliza O'Reilly, Yong Sheng Soh, Venkat Chandrasekaran
Foundations of Computational Mathematics, 2025
The star geometry of critic-based regularizer learning
Oscar Leong, Eliza O'Reilly, Yong Sheng Soh
The Thirty-eighth Annual Conference on Neural Information Processing Systems (NeurIPS), 2024