Added "picard-extended_infomax" ICA algorithm

mne-tools · Oct 30, 2023 · a811ac0 · a811ac0
1 parent 327cb94
commit a811ac0
Show file tree

Hide file tree

Showing 4 changed files with 11 additions and 2 deletions.
diff --git a/docs/source/v1.5.md.inc b/docs/source/v1.5.md.inc
@@ -11,6 +11,7 @@
 - Improved logging of coregistration distances (#769 by @larsoner)
 - Input validation has been improved by leveraging [pydantic](https://docs.pydantic.dev) (#779 by @larsoner)
 - Reduced logging when reports are created and saved (#799 by @hoechenberger)
+- Added [`"picard-extended_infomax"`][mne_bids_pipeline._config.ica_algorithm] ICA algorithm to perform "extended Infomax"-like ICA decomposition using Picard (#801 by @hoechenberger)
 
 [//]: # (### :warning: Behavior changes)
 

diff --git a/mne_bids_pipeline/_config.py b/mne_bids_pipeline/_config.py
@@ -1258,9 +1258,13 @@
     ```
 """
 
-ica_algorithm: Literal["picard", "fastica", "extended_infomax"] = "picard"
+ica_algorithm: Literal[
+    "picard", "fastica", "extended_infomax", "picard-extended_infomax"
+] = "picard"
 """
-The ICA algorithm to use.
+The ICA algorithm to use. `"picard-extended_infomax"` operates `picard` such that the
+generated ICA decomposition is identical to the one generated by the extended Infomax
+algorithm (but may converge in less time).
 """
 
 ica_l_freq: Optional[float] = 1.0

diff --git a/mne_bids_pipeline/steps/preprocessing/_06a_run_ica.py b/mne_bids_pipeline/steps/preprocessing/_06a_run_ica.py
@@ -70,6 +70,9 @@ def fit_ica(
 
     if algorithm == "picard":
         fit_params = dict(fastica_it=5)
+    elif algorithm == "picard-extended_infomax":
+        algorithm = "picard"
+        fit_params = dict(ortho=False, extended=True)
     elif algorithm == "extended_infomax":
         algorithm = "infomax"
         fit_params = dict(extended=True)

diff --git a/mne_bids_pipeline/tests/configs/config_ds003392.py b/mne_bids_pipeline/tests/configs/config_ds003392.py
@@ -20,6 +20,7 @@
 
 # Artifact correction.
 spatial_filter = "ica"
+ica_algorithm = "picard-extended_infomax"
 ica_max_iterations = 500
 ica_l_freq = 1.0
 ica_n_components = 0.99