Neural Multivariate Regression with Multi-Task Learning and Target Preprocessing
Abstract
The Unconstrained Feature Model (UFM) enables closed-form approximations for training loss in deep neural networks (DNNs). We use the UFM to motivate testable hypotheses about neural multivariate regression—fundamental to imitation learning, robotics, and reinforcement learning. Specifically, we analyze multi-task versus single-task models and the impact of target preprocessing. The UFM predicts advantages for multi-task models under comparable or stronger regularization and identifies regimes where whitening or normalization reduces training error. Experiments across four robotic and autonomous driving datasets consistently support these qualitative trends. This work illustrates how simplified analytical models can structure empirical investigation by generating theoretical predictions that are then validated through practice.