hanseeul commited on
Commit
2e6dc6d
·
verified ·
1 Parent(s): b3a9ec0

Remove personalised READMEs

Browse files
personalised/offline/README.md DELETED
@@ -1,98 +0,0 @@
1
- # CaReDiff Personalised Models, Offline Track (REACT 2026)
2
-
3
- Three personalised models for the offline MAFRG track. Each model is the same
4
- frozen generic offline backbone plus a Personalised Residual Adapter (PRA)
5
- trained under a different listener condition. The backbone weights are shared
6
- by all three and are identical to the generic offline submission.
7
-
8
- ## Layout
9
-
10
- ```
11
- offline/
12
- backbone/ frozen generic backbone (shared by all conditions)
13
- CausalTransformerDenoiser/checkpoint_120.pth
14
- DiffusionPriorNetwork/checkpoint_120.pth
15
- EEGPredictionHead/checkpoint_120.pth
16
- adapters/
17
- personality/ModifierNetwork/checkpoint_best.pth
18
- lhfb/ModifierNetwork/checkpoint_best.pth
19
- both/ModifierNetwork/checkpoint_best.pth
20
- ```
21
-
22
- The adapter file also contains the fine-tuned EEG head, which overwrites the
23
- backbone EEG head at load time.
24
-
25
- ## Checksums (SHA-256)
26
-
27
- | File | SHA-256 |
28
- |---|---|
29
- | backbone/CausalTransformerDenoiser/checkpoint_120.pth | 68faca9700415c949eecbe7bd3e381877a76b5e1b24bdab9c30e6fd5b628faa2 |
30
- | backbone/DiffusionPriorNetwork/checkpoint_120.pth | d1b66e87f51afd9bb93bdcef1b9e350e6366aa8f995920e400d7e7dd4e299357 |
31
- | backbone/EEGPredictionHead/checkpoint_120.pth | 750c49999a180cda330b88d771f99d1dca0fd94a810470ea77a45561cfd58780 |
32
- | adapters/personality/ModifierNetwork/checkpoint_best.pth | 8e0a501237c9b80b8c9e9524bd089fa5ca54ad747bdf9ed65dd97b8d883bf928 |
33
- | adapters/lhfb/ModifierNetwork/checkpoint_best.pth | 0ddfde5284c580c3cc461006b2b7cd4df73d2715a8700d3838d8d5e5db8eb7f4 |
34
- | adapters/both/ModifierNetwork/checkpoint_best.pth | 73434669c633bc6384acc9845e62c0c4302c9322be27f04e30005e55dda3ab92 |
35
-
36
- ## Conditions
37
-
38
- | Folder | Listener condition | Config value |
39
- |---|---|---|
40
- | adapters/personality | Big-Five personality (5-d) | `personality_only` |
41
- | adapters/lhfb | Listener historical facial behaviour (3DMM) | `3dmm_only` |
42
- | adapters/both | Both, gated fusion | `3dmm_personality` |
43
-
44
- Training: AdamW, learning rate 2e-4, weight decay 1e-4, gradient clipping 1.0,
45
- 30 epochs, batch size 32, seed 1234, counterfactual listener-swap loss
46
- (weight 0.5, margin 0.05). The backbone stays frozen throughout.
47
-
48
- ## Test performance (MARS test set, official evaluation code, num_gts=10)
49
-
50
- | Condition | FRCorr | FRDist | FRDiv | FRVar | FRRea | FRSyn |
51
- |---|---|---|---|---|---|---|
52
- | personality | 0.7786 | 173.63 | 0.1221 | 0.0782 | 50.91 | 48.37 |
53
- | lhfb | 0.7824 | 173.11 | 0.1200 | 0.0766 | 51.23 | 48.26 |
54
- | both | 0.7822 | 171.41 | 0.1187 | 0.0761 | 50.82 | 48.28 |
55
-
56
- FRRea is the FID between rendered generated frames and ground-truth frames
57
- (56,100 frames per side, frame stride 30).
58
-
59
- ## How to run
60
-
61
- The source code is in the CaReDiff GitHub repository
62
- (https://github.com/smu-ivpl/CaReDiff, `personalised/code/`). Example for
63
- the personality condition (set `PKG` to the absolute path of the
64
- `personalised` folder containing the checkpoints):
65
-
66
- ```bash
67
- cd code
68
- python main.py --config-name g2p_delta stage=test task=offline \
69
- data_dir=<MARS_ROOT> run_id=eval_offline_personality \
70
- trainer.batch_size=4 num_gts=10 \
71
- trainer.generic.eval_condition_mode=matched \
72
- trainer.generic.eval_eeg=false \
73
- trainer.main_model.args.personal_condition_mode=personality_only \
74
- resume_id=personality \
75
- trainer.ckpt_dir=$PKG/offline/adapters \
76
- trainer.pretrained.diffusion_decoder=$PKG/offline/backbone/CausalTransformerDenoiser/checkpoint_120.pth \
77
- trainer.pretrained.diffusion_prior=$PKG/offline/backbone/DiffusionPriorNetwork/checkpoint_120.pth \
78
- trainer.pretrained.eeg_head_checkpoint=$PKG/offline/backbone/EEGPredictionHead/checkpoint_120.pth
79
- ```
80
-
81
- The adapter is loaded from `<trainer.ckpt_dir>/<resume_id>/ModifierNetwork/`,
82
- which maps directly onto the `adapters/` layout above. For the other two
83
- conditions, change `personal_condition_mode` and `resume_id` (`lhfb` or
84
- `both`) according to the table. The loader verifies that the checkpoint was
85
- trained with the configured condition mode and stops with an error on a
86
- mismatch.
87
-
88
- ## Notes
89
-
90
- - Large assets shared with the official baseline are not duplicated here.
91
- The post-processor EmotionVAE checkpoint (517 MB) is required for
92
- evaluation and must be placed at
93
- `code/pretrained_models/post_processor/checkpoint.pth`. The PIRender
94
- renderer (234 MB) is needed only for FRRea rendering. Take both from the
95
- official baseline_react2026 repository.
96
- - Python dependencies: `code/requirements.txt`.
97
- - The MARS dataset is not included and must be obtained through the
98
- challenge organisers.
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
personalised/online/README.md DELETED
@@ -1,108 +0,0 @@
1
- # CaReDiff Personalised Models, Online Track (REACT 2026)
2
-
3
- Three personalised models for the online MAFRG track. Each model is the same
4
- frozen generic online backbone plus a Personalised Residual Adapter (PRA)
5
- trained under a different listener condition. Each online adapter was
6
- warm-started from its offline counterpart of the same condition and adapted
7
- with scheduled sampling (probability ramping to 0.5 over 25 epochs). The
8
- backbone weights are shared by all three and are identical to the generic
9
- online submission.
10
-
11
- ## Layout
12
-
13
- ```
14
- online/
15
- backbone/ frozen generic backbone (shared by all conditions)
16
- CausalTransformerDenoiser/checkpoint_120.pth
17
- DiffusionPriorNetwork/checkpoint_120.pth
18
- EEGPredictionHead/checkpoint_120.pth
19
- adapters/
20
- personality/ModifierNetwork/checkpoint_best.pth
21
- lhfb/ModifierNetwork/checkpoint_best.pth
22
- both/ModifierNetwork/checkpoint_best.pth
23
- ```
24
-
25
- The adapter file also contains the fine-tuned EEG head, which overwrites the
26
- backbone EEG head at load time.
27
-
28
- ## Checksums (SHA-256)
29
-
30
- | File | SHA-256 |
31
- |---|---|
32
- | backbone/CausalTransformerDenoiser/checkpoint_120.pth | f4fc53506fc94a65e86b52bfe1491669a73ca3429ad0b1ab51c62488854242f0 |
33
- | backbone/DiffusionPriorNetwork/checkpoint_120.pth | 8b717d619cd37fc793f80f37a4af607bda5e9709c83b82f17916b2467a4380a6 |
34
- | backbone/EEGPredictionHead/checkpoint_120.pth | 60c7a7ae4e6a233fdb59c0ee1e099daf1158931d876a5f46386a781fa2a52a52 |
35
- | adapters/personality/ModifierNetwork/checkpoint_best.pth | fad7691aea8c1895a11fcc7d40873d83ea44e0692e6dfb75f736d89cdd9e62d0 |
36
- | adapters/lhfb/ModifierNetwork/checkpoint_best.pth | c5d8c774774b8a1994be3149c3bb475384326e24a9af64edbf15cda0292da868 |
37
- | adapters/both/ModifierNetwork/checkpoint_best.pth | 77a744f486c46484dc7a357c484bd5c3582345ea5f6947aea1aa625ba00e660a |
38
-
39
- ## Conditions
40
-
41
- | Folder | Listener condition | Config value |
42
- |---|---|---|
43
- | adapters/personality | Big-Five personality (5-d) | `personality_only` |
44
- | adapters/lhfb | Listener historical facial behaviour (3DMM) | `3dmm_only` |
45
- | adapters/both | Both, gated fusion | `3dmm_personality` |
46
-
47
- Training: AdamW, learning rate 2e-4, weight decay 1e-4, gradient clipping 1.0,
48
- 30 epochs, batch size 32, seed 1234, counterfactual listener-swap loss
49
- (weight 0.5, margin 0.05). The backbone stays frozen throughout. Generation
50
- is autoregressive over 30-frame windows; each of the 10 predictions
51
- conditions on its own previous window.
52
-
53
- ## Test performance (MARS test set, official evaluation code, num_gts=10)
54
-
55
- | Condition | FRCorr | FRDist | FRDiv | FRVar | FRRea | FRSyn |
56
- |---|---|---|---|---|---|---|
57
- | personality | 0.6485 | 185.17 | 0.1521 | 0.0831 | 50.58 | 47.92 |
58
- | lhfb | 0.6481 | 191.88 | 0.1521 | 0.0828 | 50.89 | 47.92 |
59
- | both | 0.6355 | 181.11 | 0.1451 | 0.0790 | 52.09 | 48.16 |
60
-
61
- FRRea is the FID between rendered generated frames and ground-truth frames
62
- (56,100 frames per side, frame stride 30).
63
-
64
- ## How to run
65
-
66
- The source code is in the CaReDiff GitHub repository
67
- (https://github.com/smu-ivpl/CaReDiff, `personalised/code/`). Example for
68
- the personality condition (set `PKG` to the absolute path of the
69
- `personalised` folder containing the checkpoints):
70
-
71
- ```bash
72
- cd code
73
- python main.py --config-name g2p_delta_online stage=test task=online \
74
- data_dir=<MARS_ROOT> run_id=eval_online_personality \
75
- trainer.batch_size=4 num_gts=10 \
76
- trainer.generic.eval_condition_mode=matched \
77
- trainer.generic.eval_eeg=false \
78
- trainer.main_model.args.personal_condition_mode=personality_only \
79
- trainer.model.diff_model.diffusion_decoder.args.past_l_emotion_drop_prob=0.2 \
80
- resume_id=personality \
81
- trainer.ckpt_dir=$PKG/online/adapters \
82
- trainer.pretrained.diffusion_decoder=$PKG/online/backbone/CausalTransformerDenoiser/checkpoint_120.pth \
83
- trainer.pretrained.diffusion_prior=$PKG/online/backbone/DiffusionPriorNetwork/checkpoint_120.pth \
84
- trainer.pretrained.eeg_head_checkpoint=$PKG/online/backbone/EEGPredictionHead/checkpoint_120.pth
85
- ```
86
-
87
- The adapter is loaded from `<trainer.ckpt_dir>/<resume_id>/ModifierNetwork/`,
88
- which maps directly onto the `adapters/` layout above. For the other two
89
- conditions, change `personal_condition_mode` and `resume_id` (`lhfb` or
90
- `both`) according to the table. The loader verifies that the checkpoint was
91
- trained with the configured condition mode and stops with an error on a
92
- mismatch.
93
-
94
- Keep `past_l_emotion_drop_prob=0.2`. This flag enables the past-listener
95
- conditioning pathway; the config default disables it and does not reproduce
96
- the reported numbers.
97
-
98
- ## Notes
99
-
100
- - Large assets shared with the official baseline are not duplicated here.
101
- The post-processor EmotionVAE checkpoint (517 MB) is required for
102
- evaluation and must be placed at
103
- `code/pretrained_models/post_processor/checkpoint.pth`. The PIRender
104
- renderer (234 MB) is needed only for FRRea rendering. Take both from the
105
- official baseline_react2026 repository.
106
- - Python dependencies: `code/requirements.txt`.
107
- - The MARS dataset is not included and must be obtained through the
108
- challenge organisers.