ECG Synthesis using MIMIC-IV ECGs

[sanketahegde]

Hello @juanlopezcode ,

I have been trying to process the MIMIC-IV ECGs to use for training your SSSD-ECG model with the intention of generating ECGs. I also created multi-labels and ended up with a label length of 20. Currently, there are around 200,000 ECGs which I use for training which is quite large compared to PTB-XL dataset. At the end, I have my training set with data shape - (220000, 12, 5000) and label shape - (220000, 20). I already tried training once with these ECGs but the model did not learn properly and also started over-fitting after 6 epochs.

Now, my question is what modifications must I do to the currently available SSSD-ECG model setup to train and generate with MIMIC-IV ECGs?

Any suggestions are appreciated. Thank you!