Remove personalised READMEs
Browse files- personalised/offline/README.md +0 -98
- personalised/online/README.md +0 -108
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.
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|