Shogo-Noguchi commited on
Commit
1a09654
·
verified ·
1 Parent(s): ebc1828

Upload folder using huggingface_hub

Browse files
PredANNpp_SongAcc_Acoustic_Pretrain10kEpoch_Finetune3.5kEpoch_seed42.ckpt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7eab2d06d24b62044a368525ae6cd168e99559cf405d064e407079bc632952a9
3
+ size 78942947
README.md CHANGED
@@ -1,3 +1,65 @@
1
- ---
2
- license: cc-by-nc-4.0
3
- ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ language: en
3
+ license: cc-by-nc-4.0
4
+ tags:
5
+ - eeg
6
+ - music
7
+ - representation-learning
8
+ - pytorch-lightning
9
+ - transformer
10
+ ---
11
+
12
+ # PredANN++ (Acoustic, MuQ) — Encoder-only checkpoint
13
+
14
+ ## Model description
15
+
16
+ This model is a finetuned **EEG encoder** for **song identification from 3-second EEG segments** (10-class classification) on the NMED-T dataset.
17
+
18
+ The encoder is trained with multitask pretraining (masked prediction) using **MuQ acoustic embeddings**, followed by finetuning with cross-entropy on song ID.
19
+
20
+ - Input: EEG (128 channels, 125 Hz, 3 seconds)
21
+ - Output: Song ID logits (10 classes)
22
+
23
+ ## Intended use
24
+
25
+ - Research use for EEG-based music recognition
26
+ - Comparing the effect of acoustic neural representations (MuQ) with expectation-related representations
27
+
28
+ ## Not intended use
29
+
30
+ - Medical diagnosis
31
+ - Any clinical decision making
32
+ - Commercial usage without verifying upstream non-commercial licenses
33
+
34
+ ## Training data
35
+
36
+ - NMED-T (Naturalistic Music EEG Dataset – Tempo), 10 songs, 20 subjects, trial=1.
37
+
38
+ ## Training procedure (high-level)
39
+
40
+ 1. Multitask pretraining: encoder-decoder masked prediction of MuQ acoustic tokens (50% masking)
41
+ 2. Finetuning: encoder-only training for Song ID classification
42
+
43
+ ## Evaluation
44
+
45
+ The repository contains an evaluation script:
46
+ - `codes_3s/analysis/evaluate.py`
47
+
48
+ ## License and upstream dependencies (IMPORTANT)
49
+
50
+ This checkpoint is trained using features computed with:
51
+ - MuQ model weights (CC-BY-NC 4.0)
52
+
53
+ Therefore, this checkpoint is distributed as **CC-BY-NC 4.0**.
54
+
55
+ ## Citation
56
+
57
+ If you use this model, please cite:
58
+ - The PredANN++ paper (to be added upon publication)
59
+ - NMED-T dataset paper
60
+ - MuQ model
61
+
62
+ ## Acknowledgements
63
+
64
+ We borrow and adapt code from multiple repositories. Please see:
65
+ - `THIRD_PARTY_NOTICES.md` in the GitHub repository.