Group-constrained manifold learning: Application to AD risk assessment

Abstract The exploration of high-dimensional medical imaging datasets often requires the use of dimensionality reduction techniques. Generally, individual instances in longitudinal studies are not independent as several measurements of the same subjects, acquired at several time points, are included. Most non-linear dimensionality reduction techniques, that rely on defining a neighborhood graph to uncover local geometry, neglect potentially valuable grouping information and thus can lead to a poor estimation of the underlying manifold. Incorporating a-priori grouping knowledge of the data can lead to an improved low-dimensional representation. In this work, an approach to leverage such a-priori information is proposed, with the main interest in generating longitudinal biomarkers to assess a subject's risk of developing Alzheimer's disease (AD), and hence estimate time-to-conversion (TTC) and the mini mental state examination (MMSE) of subjects that develop AD. This is achieved via a novel formulation of the Laplacian Eigenmaps and Isomap manifold learning algorithms, that allows the incorporation of group-constraints. The potential of the proposed formulation was demonstrated on a swiss-roll toy example that clearly illustrates the problems faced by traditional techniques when additional structure is observed in the data. A real data experiment showed that the proposed constrained methods yielded improvements of ∼16% and ∼6% over the original unconstrained methods in kNN estimation of the TTC and MMSE on the ADNI database.