4D Tensor Multi-task Continual Learning for Disease Dynamic Prediction

Yu Zhang, Xulong Wang, Vitaveska Lanfranchi, Po Yang

Primary Area: applications to neuroscience & cognitive science

Code Of Ethics: I acknowledge that I and all co-authors of this work have read and commit to adhering to the ICLR Code of Ethics.

Keywords: Alzheimer’s disease progression, tensor multi-task learning, continual learning, amalgamated magnitude-direction quantification, brain structure variation

Submission Guidelines: I certify that this submission complies with the submission instructions as described on https://iclr.cc/Conferences/2024/AuthorGuide.

Abstract: Machine learning techniques for predicting Alzheimer's disease (AD) progression can substantially help researchers and clinicians establish strong AD preventive and treatment strategies. However, current research on AD prediction algorithms encounters challenges with monotonic data form, small dataset and scarcity of time-continuous data. To address all three of these problems at once, we propose a novel machine learning approach that implements the 4D tensor multi-task continual learning algorithm to predict AD progression by quantifying multi-dimensional information on brain structural variation and knowledge sharing between patients. To meet real-world application scenarios, the method can integrate knowledge from all available data as patient data increases to continuously update and optimise prediction results. To evaluate the performance of the proposed approach, we conducted extensive experiments utilising data from the Alzheimer's Disease Neuroimaging Initiative (ADNI). The results demonstrate that the proposed approach has superior accuracy and stability in predicting various cognitive scores of AD progression compared to single-task learning, benchmarks and state-of-the-art multi-task regression methods. The proposed approach identifies structural brain variations in patients and utilises it to accurately predict and diagnose AD progression from magnetic resonance imaging (MRI) data alone, and the performance of the model improves as the MRI data increases.

Anonymous Url: I certify that there is no URL (e.g., github page) that could be used to find authors' identity.

No Acknowledgement Section: I certify that there is no acknowledgement section in this submission for double blind review.

Submission Number: 5477