cs.LGMar 12, 2026

A Geometrically-Grounded Drive for MDL-Based Optimization in Deep Learning

Authors: Ming Lei, Shufan Wu, Christophe Baehr

Organizations: School of Aeronautics and Astronautics, Shanghai JiaoTong University, Shanghai China · Météo-France/CNRS CNRM/GAME UMR 3589 and the Mathematical Institute of Toulouse, France

Abstract

This paper introduces a novel optimization framework that fundamentally integrates the Minimum Description Length (MDL) principle into the training dynamics of deep neural networks. Moving beyond its conventional role as a model selection criterion, we reformulate MDL as an active, adaptive driving force within the optimization process itself. The core of our method is a geometrically-grounded cognitive manifold whose evolution is governed by a coupled Ricci flow, enriched with a novel MDL Drive term derived from first principles. This drive, modulated by the task-loss gradient, creates a seamless harmony between data fidelity and model simplification, actively compressing the internal representation during training. We establish a comprehensive theoretical foundation, proving key properties including the monotonic decrease of description length (Theorem~\ref{thm:convergence}), a finite number of topological phase transitions via a geometric surgery protocol (Theorems~\ref{thm:surgery}, \ref{thm:ultimate_fate}), and the emergence of universal critical behavior (Theorem~\ref{thm:universality}). Furthermore, we provide a practical, computationally efficient algorithm with O(Nlog⁡N)O(N \log N) per-iteration complexity (Theorem~\ref{thm:complexity}), alongside guarantees for numerical stability (Theorem~\ref{thm:stability}) and exponential convergence under convexity assumptions (Theorem~\ref{thm:convergence_rate}). Empirical validation on synthetic regression and classification tasks confirms the theoretical predictions, demonstrating the algorithm's efficacy in achieving robust generalization and autonomous model simplification. This work provides a principled path toward more autonomous, generalizable, and interpretable AI systems by unifying geometric deep learning with information-theoretic principles.

Figures & tables

Explore similar work

CardsList
  1. Does Compression Imply Generalization? A Minimum Description Length Perspective

    Oct 15, 2026Lincen Yang, Jiayang Shi, Nan Pu +4Neural Network GeneralizationGeneralization Bounds

  2. Unified convergence analysis for gradient descent optimization methods in the training of deep neural networks

    Jul 5, 2026Shokhrukh Ibragimov, Arnulf JentzenDeep Learning OptimizationNeural Network Optimization

  3. Conjugate Learning Theory: Uncovering the Mechanisms of Trainability and Generalization in Deep Neural Networks

    Feb 18, 2026Binchuan QiNeural Network GeneralizationNeural Network Optimization