Latent mixed-effect models for high-dimensional longitudinal data

Loading...
Thumbnail Image

Access rights

openAccess
CC BY
publishedVersion

URL

Journal Title

Journal ISSN

Volume Title

A1 Alkuperäisartikkeli tieteellisessä aikakauslehdessä

Major/Subject

Mcode

Degree programme

Language

en

Pages

30

Series

Transactions on Machine Learning Research, Volume 2025, issue May

Abstract

Modelling longitudinal data is an important yet challenging task. These datasets can be high-dimensional, contain non-linear effects and feature time-varying covariates. Gaussian process (GP) prior-based variational autoencoders (VAEs) have emerged as a promising approach due to their ability to model time-series data. However, they are costly to train and struggle to fully exploit the rich covariates characteristic of longitudinal data, making them difficult for practitioners to use effectively. In this work, we leverage linear mixed models (LMMs) and amortized variational inference to provide conditional priors for VAEs, and propose LMM-VAE, a scalable, interpretable and identifiable model. We highlight theoretical connections between it and GP-based techniques, providing a unified framework for this class of methods. Our proposal performs competitively compared to existing approaches across simulated and real-world datasets.

Description

Publisher Copyright: © 2025, Transactions on Machine Learning Research. All rights reserved.

Keywords

Other note

Citation

Ong, P, Haußmann, M, Lönnroth, O & Lähdesmäki, H 2025, 'Latent mixed-effect models for high-dimensional longitudinal data', Transactions on Machine Learning Research, vol. 2025, no. May. < https://openreview.net/forum?id=7A96yteeF9 >