Sample-Weighted End-to-End Trace-Norm Geometry for Multitask Learning
Not on the map. Why?
Abstract (arXiv)
Multitask models combine a shared representation with task-specific outputs, but generalization bounds often control the two components separately. Such products can discard relative orientation and cancellation and can change under equivalent transformations of intermediate coordinates even when the represented predictors are unchanged. We study instead the sample-size-weighted trace norm of the end-to-end map from task coefficients to input-space predictors. For its fixed-radius class, we derive the exact empirical Rademacher complexity. The same quantity is characterized by eliminating a positive-definite task covariance after the representation acts and, in finite-dimensional intermediate spaces, by optimizing the separated product over all equivalent invertible refactorizations. Explicit constructions show unbounded orientation and factorization gaps and an exponential depth gap for cancelling linear layers. As a geometric application, finite-to-one Lipschitz shared maps yield an exact Sobolev task Gram matrix determined by multiplicity and local directional distortion. We evaluate the corresponding convex regularizer in two protocol-locked unseen suites. Across 252 paired held-out comparisons, weighted joint nuclear regularization improves average population excess over unweighted nuclear regularization by 0.00764, with a stratified-bootstrap 95% interval [0.00465, 0.01110]. Correct task counts improve average and least-sampled-quartile excess over shifted counts by 0.01072 and 0.02847; all 15 imbalanced rank-suite cells are positive and the balanced effect is zero. Weighted joint nuclear also outperforms weighted Frobenius and independent ridge. The least-sampled-quartile comparison with unweighted nuclear remains unresolved, delimiting rather than contradicting the average advantage. All seven predeclared gates pass.
Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 23 of its 42 references that are in the map. First posted on arXiv 25 Aug 2026.
Builds on
23 of 42 references.
References · 23
- 2017Spectrally-normalized margin bounds for neural networkscited by 681
- 2017Sharp Minima Can Generalize For Deep Netscited by 462
- 2019Implicit Regularization in Deep Matrix Factorizationcited by 402
- 2017Implicit Regularization in Matrix Factorizationcited by 318
- 2018Stronger generalization bounds for deep nets via a compression approachcited by 306
- 2015Norm-Based Capacity Control in Neural Networkscited by 299
- 2017Size-Independent Sample Complexity of Neural Networkscited by 289
- 2015The Benefit of Multitask Representation Learningcited by 194
- 2015Path-SGD: Path-Normalized Optimization in Deep Neural Networkscited by 117
- 2019How do infinite width bounded norm networks look in function space?cited by 105
- 2020On Linear Identifiability of Learned Representationscited by 103
- 2017Fisher-Rao Metric, Geometry, and Complexity of Neural Networkscited by 99
11 more
- 2022Learning Dynamical Systems via Koopman Operator Regression in Reproducing Kernel Hilbert Spacescited by 66
- 2022The Role of Linear Layers in Nonlinear Interpolating Networkscited by 15
- 2019On Dropout and Nuclear Norm Regularizationcited by 12
- 2022Trace norm regularization for multi-task learning with scarce datacited by 10
- 2022Representer Theorem for Learning Koopman Operatorscited by 9
- 2023Koopman-based generalization bound: New aspect for full-rank weightscited by 9
- 2014An Inequality with Applications to Structured Sparsity and Multitask Dictionary Learningcited by 5
- 2024A Rescaling-Invariant Lipschitz Bound Based on Path-Metrics for Modern ReLU Network Parameterizationscited by 4
- 2025Operator-Based Generalization Bound for Deep Learning: Insights on Multi-Task Learningcited by 1
- 2025On the Koopman-Based Generalization Bounds for Multi-Task Deep Learningcited by 1
- 2026Operator-Theoretic Generalization Bounds for Multitask Deep Learningcited by 1
Built on by
Not cited yet.