Search papers, labs, and topics across Lattice.
This paper derives and empirically validates scaling laws for neural Koopman operators, connecting sample size, latent space dimension, and downstream control quality by decomposing the Koopman approximation error into sampling and projection error terms. The authors introduce a covariance loss and an inverse control loss as regularizers to improve model fitting and closed-loop control. Experiments across six robotic environments confirm the derived scaling laws and demonstrate the effectiveness of the proposed regularizers in enhancing dynamic model fitting fidelity and control performance.
Forget trial-and-error: this work provides a theoretical recipe for scaling neural Koopman operators, showing how to optimally allocate effort between data collection and model capacity for robotic control.
Data-driven neural Koopman operator theory has emerged as a powerful tool for linearizing and controlling nonlinear robotic systems. However, the performance of these data-driven models fundamentally depends on the trade-off between sample size and model dimensions, a relationship for which the scaling laws have remained unclear. This paper establishes a rigorous framework to address this challenge by deriving and empirically validating scaling laws that connect sample size, latent space dimension, and downstream control quality. We derive a theoretical upper bound on the Koopman approximation error, explicitly decomposing it into sampling error and projection error. We show that these terms decay at specific rates relative to dataset size and latent dimension, providing a rigorous basis for the scaling law. Based on the theoretical results, we introduce two lightweight regularizers for the neural Koopman operator: a covariance loss to help stabilize the learned latent features and an inverse control loss to ensure the model aligns with physical actuation. The results from systematic experiments across six robotic environments confirm that model fitting error follows the derived scaling laws, and the regularizers improve dynamic model fitting fidelity, with enhanced closed-loop control performance. Together, our results provide a simple recipe for allocating effort between data collection and model capacity when learning Koopman dynamics for control.
CMU ML