SpectraLDS: Provable Distillation for Linear Dynamical Systems

Sofiia Druchyna (Princeton University) · Elad Hazan (Princeton University) · Devan Shah (Princeton University) · Shlomo Fortgang (Princeton University)
accuracy guaranteesconstant-space inferenceconstant-time inferenceconvex optimizationconvolutionseffective memoryinference efficiencyinvert representationpredictive accuracysequence prediction architecturesspectral transformspectral transformationsstate dimensionsymmetric linear dynamical systems

We present the first provable method for identifying symmetric linear dynamical systems (LDS) with accuracy guarantees that are independent of the system’s state dimension or effective memory. Our approach builds upon recent work that represents symmetric LDSs as convolutions learnable via fixed spectral transformations. We show how to invert this representation—recovering an LDS model from its spectral transform—yielding an end-to-end convex optimization procedure. This distillation preserves predictive accuracy while enabling constant-time and constant-space inference per token, independent of sequence length. We evaluate our method, SpectraLDS, as a component in sequence prediction architectures and demonstrate that accuracy is preserved while inference efficiency is improved on tasks such as language modeling.