English

Avoiding Premature Collapse: Adaptive Annealing for Entropy-Regularized Structural Inference

Machine Learning 2026-02-06 v3 Artificial Intelligence

Abstract

Differentiable matching layers and residual connection paradigms, often implemented via entropy-regularized Optimal Transport (OT), serve as critical mechanisms in structural prediction and architectural scaling. However, recovering discrete permutations or maintaining identity mappings via annealing ϵ0\epsilon \to 0 is notoriously unstable. In this work, we identify a fundamental mechanism for this failure: \textbf{Premature Mode Collapse}. By analyzing the non-normal dynamics of the Sinkhorn fixed-point map, we reveal a theoretical thermodynamic speed limit: standard exponential cooling outpaces the contraction rate of the inference operator, which degrades as O(1/ϵ)O(1/\epsilon). To address this, we propose \textbf{Efficient Piecewise Hybrid Adaptive Stability Control (EPH-ASC)}, an adaptive scheduling algorithm that monitors the stability of the inference process. We demonstrate that EPH-ASC is essential for stabilizing Manifold-Constrained Hyper-Connections (mHC) during large-scale training on the FineWeb-Edu dataset, effectively preventing late-stage gradient explosions by enforcing a linear stability law.

Keywords

Cite

@article{arxiv.2601.23039,
  title  = {Avoiding Premature Collapse: Adaptive Annealing for Entropy-Regularized Structural Inference},
  author = {Yizhi Liu},
  journal= {arXiv preprint arXiv:2601.23039},
  year   = {2026}
}
R2 v1 2026-07-01T09:27:52.739Z