On Admissible Rank-based Input Normalization Operators
Taeyun Kim
TL;DR
We address the stability and invariance of rank-based input normalization by imposing three axioms: rank-level invariance under strictly increasing feature-wise transforms (C1), batch independence (C2), and monotone Lipschitz-stable scalarization (C3). The main result proves that any admissible operator must factor through a feature-wise rank representation and a monotone Lipschitz scalarization, i.e., $Q(x)=\Phi\bigl(s(r(x))\bigr)$; thus differentiable sorting methods based on value gaps or batch interactions lie outside this class. The paper introduces a minimal realization called QNorm and an explicit rank-based representation layer, plus empirical validation across operator stability, model robustness, and real-world datasets demonstrating nontrivial, stable behavior under monotone transformations and batch changes. This work delineates the design space for rank-based input normalization, offering a principled framework to analyze, compare, and design admissible normalization operators with provable invariance and stability properties.
Abstract
Rank-based input normalization is a workhorse of modern machine learning, prized for its robustness to scale, monotone transformations, and batch-to-batch variation. In many real systems, the ordering of feature values matters far more than their raw magnitudes - yet the structural conditions that a rank-based normalization operator must satisfy to remain stable under these invariances have never been formally pinned down. We show that widely used differentiable sorting and ranking operators fundamentally fail these criteria. Because they rely on value gaps and batch-level pairwise interactions, they are intrinsically unstable under strictly monotone transformations, shifts in mini-batch composition, and even tiny input perturbations. Crucially, these failures stem from the operators' structural design, not from incidental implementation choices. To address this, we propose three axioms that formalize the minimal invariance and stability properties required of rank-based input normalization. We prove that any operator satisfying these axioms must factor into (i) a feature-wise rank representation and (ii) a scalarization map that is both monotone and Lipschitz-continuous. We then construct a minimal operator that meets these criteria and empirically show that the resulting constraints are non-trivial in realistic setups. Together, our results sharply delineate the design space of valid rank-based normalization operators and formally separate them from existing continuous-relaxation-based sorting methods.
