An AO-ADMM Approach to Constraining PARAFAC2 on All Modes

Research output: Contribution to journalJournal articleResearchpeer-review


  • Fulltext

    Accepted author manuscript, 8.72 MB, PDF document

  • Marie Roald
  • Carla Schenker
  • Vince D. Calhoun
  • Tülay Adali
  • Bro, Rasmus
  • Jeremy E. Cohen
  • Evrim Acar
Analyzing multiway measurements with variations across one mode of the dataset is a challenge in various fields including data mining, neuroscience, and chemometrics. For example, measurements may evolve over time or have unaligned time profiles. The PARAFAC2 model has been successfully used to analyze such data by allowing the underlying factor matrices in one mode (i.e., the evolving mode) to change across slices. The traditional approach to fit a PARAFAC2 model is to use an alternating least squares–based algorithm, which handles the constant cross-product constraint of the PARAFAC2 model by implicitly estimating the evolving factor matrices. This approach makes imposing regularization on these factor matrices challenging. There is currently no algorithm to flexibly impose such regularization with general penalty functions and hard constraints. In order to address this challenge and to avoid the implicit estimation, in this paper, we propose an algorithm for fitting PARAFAC2 based on alternating optimization with the alternating direction method of multipliers (AO-ADMM). With numerical experiments on simulated data, we show that the proposed PARAFAC2 AO-ADMM approach allows for flexible constraints, recovers the underlying patterns accurately, and is computationally efficient compared to the state-of-the-art. We also apply our model to two real-world datasets from neuroscience and chemometrics, and show that constraining the evolving mode improves the interpretability of the extracted patterns.
Original languageEnglish
JournalSIAM Journal on Mathematics of Data Science
Issue number3
Pages (from-to)1191-1222
Publication statusPublished - 2022

ID: 338064674