Tensor Decompositions Meet Control Theory: Learning General Mixtures of Linear Dynamical Systems
Ainesh Bakshi, Allen Liu, Ankur Moitra, Morris Yau
Abstract
Recently Chen and Poor initiated the study of learning mixtures of linear dynamical systems. While linear dynamical systems already have wide-ranging applications in modeling time-series data, using mixture models can lead to a better fit or even a richer understanding of underlying subpopulations represented in the data. In this work we give a new approach to learning mixtures of linear dynamical systems that is based on tensor decompositions. As a result, our algorithm succeeds without strong separation conditions on the components, and can be used to compete with the Bayes optimal clustering of the trajectories. Moreover our algorithm works in the challenging partially-observed setting. Our starting point is the simple but powerful observation that the classic Ho-Kalman algorithm is a close relative of modern tensor decomposition methods for learning latent variable models. This gives us a playbook for how to extend it to work with more complicated generative models.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext db2cdbd3-33ad-440c-a365-b05c7de70263Cited by top-tier papers8
- PURE: Prompt Evolution with Graph ODE for Out-of-distribution Fluid Dynamics ModelingHao Wu, Changhu Wang, Fan Xu, Jinbao Xue et al.NeurIPS 2024 · 23 citations
- Structure Learning of Hamiltonians from Real-Time EvolutionAinesh Bakshi, Allen Liu, Ankur Moitra, Ewin TangFOCS 2024 · 7 citations
- Universal Learning of Nonlinear DynamicsEvan Dogariu, Anand Brahmbhatt, Elad HazanICML 2026 · 5 citations
- A New Approach to Controlling Linear Dynamical SystemsAnand Paresh Brahmbhatt, Gon Buzaglo, Sofiia Druchyna, Elad HazanICLR 2026 · 4 citations
- Model Stealing for Any Low-Rank Language ModelAllen Liu, Ankur MoitraSTOC 2025 · 1 citation
Builds on7
- Meta-learning for Mixed Linear RegressionWeihao Kong, Raghav Somani, Zhao Song, Sham M. Kakade et al.ICML 2020 · 70 citations
- Robust Meta-learning for Mixed Linear Regression with Small BatchesWeihao Kong, Raghav Somani, Sham M. Kakade, Sewoong OhNeurIPS 2020 · 38 citations
- Learning Mixtures of Linear Dynamical SystemsYanxi Chen, H. Vincent PoorICML 2022 · 22 citations
- Robustly learning mixtures of k arbitrary GaussiansAinesh Bakshi, Ilias Diakonikolas, He Jia, Daniel M. Kane et al.STOC 2022 · 21 citations
- Algorithmic foundations for the diffraction limitSitan Chen, Ankur MoitraSTOC 2021 · 16 citations
Related papers
- Learning Mixtures of Linear Dynamical Systems via Hybrid Tensor-EM MethodLulu Gong, Shreya SaxenaICLR 2026
- A New Approach to Learning Linear Dynamical SystemsAinesh Bakshi, Allen Liu, Ankur Moitra, Morris YauSTOC 2023 · 10 citations
- Bayesian Continuous-Time Tucker DecompositionShikai Fang, Akil Narayan, Robert M. Kirby, Shandian ZheICML 2022 · 20 citations
- Nonparametric Factor Trajectory Learning for Dynamic Tensor DecompositionZheng Wang, Shandian ZheICML 2022 · 8 citations
- Streaming Factor Trajectory Learning for Temporal Tensor DecompositionShikai Fang, Xin Yu, Shibo Li, Zheng Wang et al.NeurIPS 2023 · 12 citations
