Ballistocardiogram Artifact Removal for Concurrent EEG-fMRI Recordings Using Blind Source Separation Based on Dictionary Learning

. Simultaneous recording of electroencephalography (EEG) and functional magnetic resonance imaging (fMRI) have attracted extensive attention and research owing to their high spatial and temporal resolution. However, EEG data are easily influenced by physiological causes, gradient artifact (GA) and ballistocardiogram (BCG) artifact. In this paper, a new blind source separation technique based on dictionary learning is proposed to remove BCG artifact. The dictionary is learned from original data which represents the features of clean EEG signals and BCG artifact. Then, the dictionary atoms are classified according to a list of standards. Finally, clean EEG signals are obtained from the linear combination of the modified dictionary. The proposed method, ICA, AAS, and OBS are tested and compared using simulated data and real simultaneous EEG–fMRI data. The results suggest the efficacy and advantages of the proposed method in the removal of BCG artifacts.


Introduction
With the development of brain science, the simultaneous acquisition of electroencephalography (EEG) and functional magnetic resonance imaging (fMRI) has attracted extensive attention and research.EEG signal has the characteristics of low spatial resolution and high temporal resolution, while fMRI has the characteristics of low temporal resolution and high spatial resolution.Therefore, the combination of EEG signal and fMRI is very important for the study of brain function, pathogenesis and cognition of mental diseases [1,2].
However, we face a serious problem: during recording data in an MRI scanner, EEG singals are easy to be affected by the gradient artifact (GA) and ballistocardiogram (BCG) artifact [9].Fortunately, the gradient artifacts have time-shift invariance and their amplitude is more than 100 times higher than that of an EEG signal [3].Thus, gradient artifacts are easily discernable, and they can be removed from the EEG signal by average artifact subtraction (AAS) [4].However, the original EEG signals still contain another serious artifact, the most difficult one to remove: the ballistocardiogram artifact (BCG).The BCG artifact is caused by the hall effect induced by blood flow and the scalp twitch caused by blood pulsation, which mainly covers the frequency of EEG signal at alpha  and below [15].Therefore, BCG artifacts are a major bottleneck to the wide application of simultaneous EEG-fMRI, and it is also a problem we need to solve urgently.
To ensure the quality of EEG signals in EEG-fMRI, many methods have been used to remove BCG artifact.1) Averaged artifact subtraction (AAS), in which a BCG artifact template is estimated by averaging over the intervals of EEG signal that are corrupted by the artifact and subsequent subtraction of the template from the corrupted segments to obtain clean signal [4].2) Independent Component Analysis (ICA) [5] and 3) Principal Component Analysis (PCA) [6] are used to separate the original EEG signal into different components, identify the artifact components and then remove them.The primary challenge with these two methods is the definition of a consistent standard for artificial component selection.There is always a trade-off in selecting the number of components: removing a small number of components may still leave traces in the EEG signal, on the contrary, removing a large number of components may lead to the loss of important information in the EEG signal [14].
In recent years, dictionary learning has been widely used in multiple denoising study areas concerning images and medical signals.Abolghasemi et al. [11] firstly used dictionary learning to removal BCG artifact.The method first learns a sparse dictionary from the given data.Then, the dictionary is used to model the existing BCG contribution in the original signals.The obtained BCG model is then subtracted from the original EEG to achieve clean EEG.The results show that dictionary learning is better than AAS, OBS and DHT.However, their methods lack the study of learned dictionary.Their work proved the reliability from the evaluation index while not from the dictionary itself.The features of atoms is not fully explained and the amount of useful EEG features in the dictionary on real-time data has not been proved.
In this paper, based on the previous approach, we propose a new blind source separation technique for removing BCG artifacts EEG data based on dictionary learning.The original EEG signal is firstly studied through dictionary learning, while the dictionary contains all of the features of the dictionary, which can ensure that all BCG features are fitted.Then, the atoms contain only EEG features are sorted according to a list of standards.Finally, the clean EEG signal is obtained by the linear combination of the sorted atoms.The proposed method is tested on synthetic data and real data, and compared with ICA, AAS and OBS, which proves that the method has better performance.

Dictionary Learning
Dictionary learning is a machine learning algorithm that has been widely used in image denoising and classification, medical signals analyzing and so on.The learning method aims at inferring a dictionary matrix which can give a sparse representation of the input image or signal.Each column of the dictionary is called an atom which describes the features of the input data.The method is also called sparse dictionary learning which means the signal is represented as a sparse linear combination of atoms [16].Assume a dictionary matrix 1 . Each column of the dataset matrix which usually represents a signal i x can be represented as the linear combination of the atoms of dictionary, i.e. need to achieve enough sparsity.Thus, the learning process can be formulated as: The 0 -norm l in the minimization problem (1) is commonly substituted by 1 -norm l for measurement of sparsity since the previous is NP-hard [20].A usual procedure to solve the problem is iteratively updating dictionary and sparse coding, i.e. solve dictionary D while sparse coding S is fixed then solve S when D is fixed, until convergence.K-SVD [22] is a powerful algorithm that based on Singular Value Decomposition (SVD) and K-means implements the alternating minimization process well.And the paper uses K-SVD as the core algorithm to complete the dictionary learning.

The Proposed Technique
The input data for the proposed method which is the original signal added by all 32 EEG channels with length L is segmented into m pieces of data of length n.Each segment is a piece of continues EEG signal.
The segments would increase the amount of the input which means there would be a certain amount of duplication between each other.This strategy, which is quite similar to the one between decision tree and random forest, has the advantage that it can capture most of the local features in the giving sample [11].Each segment can be formulated as: i P is a binary matrix.Hence the proposed method can be formulated as: After solving the above problem, a dictionary D with atoms contain all of the features will be obtained.
Since the dictionary matrix contains the features of signals, it has similar mathematical forms with Independent Component Analysis (ICA), which the signals can be represented by the linear combination of independent components.Compared with ICA, dictionary learning has allows the dimensionality of atoms to be higher than the number of EEG channels.This property leads to adding redundant information to the dictionary, in this case, which mainly is the clean EEG but also provides an improvement in sparsity and precision of the signal features representation.To improve the result of dictionary learning, we modify the dimension of the dictionary and sort the atoms we learned through atoms classification and rearrangement.For the learned redundant dictionary matrix D, principal component analysis (PCA) is first applied to reduce the dimensionality and figure out the number k of the first principal components.After identifying the atoms which describe the features of the BCG, we return BCG atoms to zero to avoid returning artifact information.Therefore, the dictionary would have a better description of clean EEG signal.

Experiment
In this section, we divide the experimental results into two parts: synthetic and real data.The performance of AAS, OBS, ICA and dictionary learning was evaluated by experiments on these two data types, which proved the superiority of the proposed method.

Data Acquisition 1) Synthetic Data
The synthetic EEG data used in this paper was proposed by Vahid abolghasemi et al in 2015 [11].The synthesized signal consists of two parts: 1/f noise (also known as pink noise) and BCG artifact.For the non-uniform characteristics of EEG signal, 1/f noise is selected as EEG signal [17], while the synthesized BCG artifact is overlapped by cosine waves of different frequencies and amplitudes.For the final synthesized signal, 10dB white Gaussian noise is added, and these signals and their synthesized signals are shown in Figure 3.1.It can be seen from Figure 3.1 that as the 1/f noise of EEG signal, its amplitude is significantly smaller than that of simulated BCG signal.In fact, the EEG signal of synchronous acquisition of EEG and fMRI obtained from actual acquisition is the same.

2) Public Data
The experimental data is provided by FMRIB research center of Oxford University.The test data (FMRIB_Data.set)downloaded from the open-source EEG signal processing website includes EEG signal, BCG artifact, gradient artifact and other noises [6].

3) Clinical Data
The experimental data were provided by the sleep neuroimaging center of Southwest University.Two healthy subjects with no history of nerve or heart disease participated in the experiment.Indicates that the object rests in the scanner with the eyes open and closed, without performing any specific tasks.
Among the 32 channels recorded, 30 EEG channels used 10-20 international system.In addition, there are two bipolar channels for EMG and ECG.The data sampling rate is 5000Hz.

Data Preprocessing
The premise of effectively suppressing BCG artifacts is that there are no gradient artifacts that seriously distort EEG signals in EEG signals, so it is necessary to preprocess the collected EEG data first.Matlab (2017b; MathWorks, Inc., Natick, MA) and eeglab 12.0 (Delorme and makeig, 2004) were used to preprocess the EEG data.Firstly, the FMRIB 2.1 plug-in in EEGLAB [18] is used to remove the gradient artifact in EEG data of MRI scan.In order to save memory, we can down sample the data to 250Hz.All EEG data are then bandpass filtered at 0.5-50Hz.The data is finally converted to a 32 channel EEG with sampling frequency of 250 Hz.

Evaluation
Various evaluation indexes are used to determine the ability to inhibit BCG artifacts (more importantly, to retain brain activity), and to compare the experimental results of various methods.
The evaluation is mainly carried out from the following aspects: a) the amplitude changes; b) Power Spectral Density [19]; c) Improvement in Normalized Power Spectral Density Ratio (INPS) [12]; d) Peak-to-peak Value (PPV) [13].In addition, INPS and PPV can be calculated as follows: before, i 10 log 1 before, 1 before,  According to the method in section 3.3, the normalized power spectral density (INPS) is 13.16, the peak-to-peak value is 8.44.The values of the two performance evaluation indexes are much higher than 1, indicating that dictionary learning plays an important role in inhibiting BCG artifacts.Visualization results of EEG records before and after BCG artifact correction (AAS, OBS, ICA, proposed) of DATA 3.With the application of BCG removal method, BCG artifact contribution is attenuated, so EEG signal strongly decreases.However, it is obvious that the BCG residuals significantly decreased when using the proposed method.     1 compares the performance of AAS, OBS, and the proposed method in terms of INPS ratio.It can be concluded that the proposed method is obviously superior to other methods.

Conclusion
In this study, a set of new algorithms based on dictionary learning has been developed to remove BCG artifact from EEG recordings simultaneously acquired with continuous fMRI scanners.This method can reduce the BCG related artifacts in EEG fMRI records without using additional hardware.The validity of this method is verified by the experiments of synthetic data and real data.
It is gratifying that we still have a lot of room for progress.The classification standards still need further study, such as classifying the atoms through mutual information, the features of ECG are considered to sort m atoms with the largest mutual information with the ECG [21].At present, most of the EEG-fMRI researches are on-line data artifact removal, only a few involve real-time correction.We can consider applying the proposed method to real-time detection and removal of artifacts.One of the limitations of our study is that we did not compare the results with the data recorded outside the MRI.In fact, we can design specific paradigms involving in and out of scanner experiments, so that we can how BCG was considered affecting EEG signals from these recorded data.

Fig. 2 . 1 .
Fig. 2.1.The single values of PCA.In this situation, the first 3 single values mainly describe the features of dictionary.

Fig. 2 . 3 .
Fig. 2.3.Schematic diagram for dictionary atom selection and rearrangement.The green atoms in the dictionary are clean EEG atoms.The grey atoms are BCG atoms.White girds in sparse code matrix are 0.

Fig. 4 .
Fig. 4.1 gives the results using AAS, OBS and ICA and dictionary learning methods to the synthetic signal.

Figure 4 .
Figure 4.2 shows a segment of EEG data with a length of 10s processed by various methods.

Figure 4 .
Figure 4.3 shows averaged power spectral density of EEG data using different artifact removal algorithms.It shows that our procedure can effectively reduce the artifacts in different EEG frequency bands without alpha (8-13 Hz) and beta (13-15 Hz in Fig. 4.3) activities.

Figure 4 .
Figure 4.4 and Table1compares the performance of AAS, OBS, and the proposed method in

Fig. 4 . 4 .
Fig.4.4.INPS Reduction for all channels using different artifact correction methods versus the EEG data after removing imaging artifacts.

Table 1 .
INPSs for all channels using different artifact removal methods.

Table 2
compares the peak-to-peak ratio of BCG artifact suppression methods based on AAS, OBS and ICA, and the results prove the advantages of dictionary learning.

Table 2 .
Peak-to-peak value for all channels using different artifact removal methods.

TABLE 3
is the averaged INPS and peak-to-peak ratio using different method.it is obvious that the performance indexes of dictionary-based learning methods are allmost higher than those of ICA, AAS and OBS methods, except DATA 3, so it can be considered that the dictionary-based learning method studied in this paper is more effective in inhibiting BCG artifacts.

TABLE 3 .
Averaged INPS and peak-to-peak ratio.