Codecfake-W2V2-AASIST-Codec / implementation-notes.md
korallll's picture
Wrapper code (speech-spoof-bench)
9b965b9 verified
|
Raw History Blame Contribute Delete
1.91 kB
# Codecfake-W2V2-AASIST-Codec — implementation notes
Codec-trained **W2V2-AASIST** of the Codecfake paper (Xie et al., arXiv 2405.04880), plan model #8, built
2026-11-01. Everything except the checkpoint is identical to `../Codecfake-W2V2-AASIST-CoTrain/` (same `_net.py`
copy; see that dir's notes for the port, XLS-R truncation bit-identity proof, allow-list unpickler, score
definition, TF32-off). Helper scripts (`run_sets.sh`, `trt_export_both.sh`, `trt_jspaw.sh`, `parity.py`) live in
the CoTrain dir and cover both models.
- Code: xieyuankun/Codecfake @ `72c32e2840eef95fbeade349425fa9c5391fa11e`, **no LICENSE** → linked, not re-hosted.
- Checkpoint: `pretrained_model/codec_w2v2aasist/anti-spoofing_feat_model.pt` (sha256 `5a606d1f…c25a3`), trained on
Codecfake train only; fetched at load time from the pinned raw GitHub URL → `head_codec_w2v2aasist.safetensors`.
Strict load 245 keys; pickled temperatures/pool ratios identical to the port.
- Params 76,554,634. Score = logits[:,0] − logits[:,1] (class 0 = bona fide). trained_on: none.
## Reproduction vs authors' `result/ITW_result.txt` (no 19LA file shipped for this model)
Full InTheWild (31 779): EER ours 9.274% vs authors 9.609%; Pearson (logit) 0.9989, Spearman 0.9990,
median |Δ logit| 0.044 — the same 16-bit-FLAC packaging residual measured for the CoTrain sibling (float
source wavs: median |Δ| 0.0035).
## Throughput / TRT (2026-11-01, shared GPUs)
PyTorch: same arch as CoTrain (288 / 104 utt/s on 4070 Ti SUPER / 5060 Ti). TRT sidecar
`trt_codecfake-w2v2-aasist-codec.json`:
| card | precision | best batch | utt/s | build Spearman | J-SPAW_LA EER PT → TRT | Spearman vs PT |
|---|---|---|---|---|---|---|
| RTX 4070 Ti SUPER | FP32 (FP16 0.9997 < 0.9999) | 32 | 395 | 1.0000 | 0.1252 → 0.1252 | 0.999999 |
| RTX 5060 Ti | FP16 (0.99995) | 16 | 784 | 0.99995 | 0.1252 → 0.1252 | 0.99996 |
Both accepted.