[edit]
Scalable Algorithms for Learning High-Dimensional Linear Mixed Models
Proceedings of the 34th Conference on Uncertainty in Artificial Intelligence, PMLR R16:258-267, 2018.
Abstract
Linear mixed models (LMMs) are used exten- sively to model observations that are not in- dependent. Parameter estimation for LMMs can be computationally prohibitive on big data. State-of-the-art learning algorithms require computational complexity which depends at least linearly on the dimension p of the co- variates, and often use heuristics that do not offer theoretical guarantees. We present scal- able algorithms for learning high-dimensional LMMs with sublinear computational complex- ity dependence on p. Key to our approach are novel dual estimators which use only kernel functions of the data, and fast computational techniques based on the subsampled random- ized Hadamard transform. We provide theo- retical guarantees for our learning algorithms, demonstrating the robustness of parameter es- timation. Finally, we complement the theory with experiments on large synthetic and real data.