Learning Multi-Frequency Partial Correlation Graphs
Gabriele D'Acunto, Paolo Di Lorenzo, Francesco Bonchi, Stefania Sardellitti, Sergio Barbarossa
TL;DR
The paper tackles learning conditional dependencies among multiple time series across different frequency bands by introducing a multi-frequency, block-sparse partial-correlation graph ($K$-PCG). It develops two nonconvex methods: a closed-form CF approach that requires per-band sparsity and an iterative IA approach that jointly learns the CSD and its inverse via a successive convex approximation framework with inner ADMM. CF recovers per-band arc sets by selecting the top $s_m$ columns of a flattened inverse CSD, while IA solves a sequence of strongly convex surrogates to converge to stationary points. Across synthetic experiments, IA and especially CF with full sparsity knowledge outperform baselines, and a financial case study demonstrates that partial correlations concentrate in a few frequency bands, underscoring the practical value of frequency-aware learning. Public JAX code is provided to enable results reproduction and broader adoption across finance, climate, industrial, and biomedical domains.
Abstract
Despite the large research effort devoted to learning dependencies between time series, the state of the art still faces a major limitation: existing methods learn partial correlations but fail to discriminate across distinct frequency bands. Motivated by many applications in which this differentiation is pivotal, we overcome this limitation by learning a block-sparse, frequency-dependent, partial correlation graph, in which layers correspond to different frequency bands, and partial correlations can occur over just a few layers. To this aim, we formulate and solve two nonconvex learning problems: the first has a closed-form solution and is suitable when there is prior knowledge about the number of partial correlations; the second hinges on an iterative solution based on successive convex approximation, and is effective for the general case where no prior knowledge is available. Numerical results on synthetic data show that the proposed methods outperform the current state of the art. Finally, the analysis of financial time series confirms that partial correlations exist only within a few frequency bands, underscoring how our methods enable the gaining of valuable insights that would be undetected without discriminating along the frequency domain.
