bakqui / ST-MEM

[ICLR2024] Guiding Masked Representation Learning to Capture Spatio-Temporal Relationship of Electrocardiogram
Other
32 stars 5 forks source link

Question about t-SNE experiment #9

Closed na-fi closed 3 weeks ago

na-fi commented 3 weeks ago

Thanks for the intriguing work!

I have a question about the t-SNE experiment. For generating the embeddings, did you fine-tune on a downstream task, like classification, first? Or did you simply use the encoder with the pre-trained MAE weights?

Thank you

YunwonTae commented 3 weeks ago

Thank you for getting interested in our paper.

We used the pre-trained encoder weights and generated the embeddings. (didn't fine-tune on a downstream task)

na-fi commented 3 weeks ago

Thank you :)