groupICA

This site accompanies the following manuscript on an extension to the well-known independent component analysis (ICA) for blind source separation. The extension demixes signals that have been created as follows:

$$X_i = A \cdot (S_i + H_i)$$ where
Python implementation
An open-source scikit-learn compatible Python implementation of this algorithm can be installed from PyPI. The Python source is available on github; please report issues there or fire a pull request if you wish to contribute. Please refer to the Python documentation, getting started in Python, as well as the minimalistic example to get started.
The experimental results presented in the manuscript can be reproduced using this code archive.
R implementation
An open-source R implementation of this algorithm can be installed from CRAN. The R source is available on github; please report issues there or fire a pull request if you wish to contribute. The documentation ships with the CRAN package.
Matlab implementation
An open-source Matlab implementation of this algorithm can be installed from the Matlab source available on github; please report issues there or fire a pull request if you wish to contribute. A basic docstring is provided to get you started.

Contents

"America's Got Talent Duet Problem" -- an audible example

Consider we are recording from two microphones in a setting as it is sketched below, where two singers perform a duet on stage while we may want to judge the individual's performance. schematic

Original signal recorded at microphones (real scrambled)

Let's first see how the signals recorded in such a setting may sound like.
Signal Audience noise Open window (birds) Mower Applause
1


2




The signals are mixed to an extent where it is sometimes hard to follow one of the two speeches, which are interwovenly recorded at both microphones.

Signals recovered by pooled ICA (still scrambled)

As a first classical attempt we may apply ICA in order to do blind source separation and hopefully recover the two speeches from the microphone recordings. Here is what pooled ICA can do for us:
Signal Audience noise Open window (birds) Mower Applause
1


2




Signals recovered by groupICA (much better)

Since the data has a grouped structure, e.g., there is periods where someone opened the window and we can hear a clear mower sound etc., it falls into the regime of the groupICA model. Applying groupICA that properly accounts for this grouped structure we are able to recover the two speeches as follows:
Signal Audience noise Open window (birds) Mower Applause
1


2




EEG Data -- an example comparing to pooledICA

A common application of ICA is in the analysis of EEG (Electroencephalography) data. To illustrate a potential use of groupICA, we apply it to the publicly available multi-subject data set Covert shifts of attention, which is preprocessed as in our manuscript (Data set 2).

In this illustration, we select one subject and learn an unmixing matrix with groupICA, pooledICA and a random projection on the remaining 7 subjects. For each of the 3 unmixing matrices, we then construct the following two types of topographic maps on the left-out subject:

  1. A topographic map of $a_j$, where $a_j$ is the $j$-th column of the mixing matrix. This topographic map illustrates the mixing of the $j$-th source at each electrode position.
  2. A topographic map of $\operatorname{cov}(\mathbf{X_t})v_j^\top$, where $v_j$ is the $j$-th row of the unmixing matrix and $\operatorname{cov}(\mathbf{X_t})$ is the covariance matrix of the observed data at time $t$ estimated with a moving-window estimator. This topographic map illustrates the time-dependent source activation of the recovered $j$-th source at each electrode.

The resulting plots are illustrated in the following video. Here, the first row corresponds to the topographic map (1) and the second row to the one described in (2).

Given an underlying ICA model and a good estimation the time changing topographic maps in the second row should correspond to the first row. In this particular example, one can see that the source recovered by groupICA remains more stable across time and is able to capture the overall structure more consistently. See also Section 3.1 in our manuscript for more details on this.

Getting started in Python

We have made our code available as a scikit-learn compatible package. The groupICA package can be installed from PyPI using the following command:
pip install groupICA
The default usage requires a data matrix $X\in\mathbb{R}^{n\times d}$, an array $\operatorname{groups}\in\mathbb{R}^{n}$ specifing to which group each of the $n$ observations belongs and a second array $\operatorname{partition}\in\mathbb{R}^{d}$ specifing the partition. The sources can then be recovered using the following commands:
from groupICA import GroupICA

g = GroupICA()
g.fit(Xtrain, group_index=groups, partition_index=partition)
# g.V_ holds the unmixing matrix

recovered_sources = g.transform(Xtest)
For a more in-depth example and further details on the package, see the documentation.
The presented educational audible example combines parts of these sounds from freesound "Crowd ambience, eating popcorn.wav" by IllusiaProductions (https://freesound.org/people/IllusiaProductions/sounds/249940/), "Ambience, Food Court, B.wav" by InspectorJ (https://freesound.org/people/InspectorJ/sounds/421715/), "rbh Applause 01 big.WAV" by RHumphries (https://freesound.org/people/RHumphries/sounds/1921/), "Birds awaking" by arnaud coutancier (https://freesound.org/people/arnaud%20coutancier/sounds/427335/), "lawn mower (from in house)" by rayjensen (https://freesound.org/people/rayjensen/sounds/347227/); as well as parts of the audio of the Final Presidential State of the Union Address 2016 (http://www.americanrhetoric.com/speeches/stateoftheunion2016.htm) and 2018 (http://www.americanrhetoric.com/speeches/stateoftheunion2018.htm) which are in the public domain or property of AmericanRhetoric.com (http://www.americanrhetoric.com/copyrightinformation.htm).