Nurgali Kadyrbek commited on
Commit ·
69d7c8e
0
Parent(s):
R0 BLSP cohort: 5 frozen-frozen adapter checkpoints
Browse files- .gitattributes +3 -0
- A_BLSP_seed1234.pt +3 -0
- A_BLSP_seed1235.pt +3 -0
- A_BLSP_seed1236.pt +3 -0
- A_BLSP_seed1237.pt +3 -0
- A_BLSP_seed1238.pt +3 -0
- README.md +64 -0
- adapter_init.json +28 -0
.gitattributes
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
*.pt filter=lfs diff=lfs merge=lfs -text
|
| 2 |
+
*.bin filter=lfs diff=lfs merge=lfs -text
|
| 3 |
+
*.safetensors filter=lfs diff=lfs merge=lfs -text
|
A_BLSP_seed1234.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:f5cc55d352a3bcd7a147896f49286bae6f2719465d71174c161673377b2fb2ee
|
| 3 |
+
size 335649312
|
A_BLSP_seed1235.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:6342f7b6e2406ced8a887917f8a5ffa7c1b4aa57ed7d76f16df94f25e035e6d2
|
| 3 |
+
size 335648160
|
A_BLSP_seed1236.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:9ebd1ddd9d0abc197436741c91d097ba431eb8d84ba1b47d4e61f88dfba9368a
|
| 3 |
+
size 335648160
|
A_BLSP_seed1237.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:d31deafaa33e1f33bcd9f959de8d14c64f403bc6153e30267eb8c60426b593a6
|
| 3 |
+
size 335648160
|
A_BLSP_seed1238.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:d25ca9da1fe215bc06523fa40ee013804786a79a24a1003d827239dbdfce8a3a
|
| 3 |
+
size 335648160
|
README.md
ADDED
|
@@ -0,0 +1,64 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
---
|
| 2 |
+
license: cc-by-nc-4.0
|
| 3 |
+
tags:
|
| 4 |
+
- speech-language-models
|
| 5 |
+
- sentence-stress
|
| 6 |
+
- frozen-frozen
|
| 7 |
+
- blsp
|
| 8 |
+
- paralinguistic
|
| 9 |
+
base_model: Qwen/Qwen3-8B
|
| 10 |
+
---
|
| 11 |
+
|
| 12 |
+
# BTA — R0 BLSP cohort (5 seeds)
|
| 13 |
+
|
| 14 |
+
5-seed cohort of the BLSP-class baseline adapter for the
|
| 15 |
+
Beyond Transcript Alignment research project.
|
| 16 |
+
|
| 17 |
+
**Architecture:** Conv1d(1024→4096, k=4, s=4) → MLP-2 → RMSNorm → +modality token (84M trainable params)
|
| 18 |
+
**Encoder:** WavLM-Large layer 16 (`microsoft/wavlm-large`, frozen)
|
| 19 |
+
**LLM:** Qwen3-8B-Instruct (`Qwen/Qwen3-8B`, frozen, `enable_thinking=False`)
|
| 20 |
+
**Loss:** $\mathcal{L}_{\mathrm{BLSP}} = \mathcal{L}_{\mathrm{task}} + \lambda_{\mathrm{KL}} \mathcal{L}_{\mathrm{KL}}$
|
| 21 |
+
**Optimizer:** AdamW lr=5e-5, weight_decay=0.01, max_norm=1.0; cosine to 1e-6
|
| 22 |
+
**Schedule:** seed 1234 = 600 steps + warmup 500; seeds 1235-1238 = 300 steps + warmup 250
|
| 23 |
+
**Code:** <https://github.com/Nurgali-Kadyrbek/frozen-speech-llm-stress>
|
| 24 |
+
|
| 25 |
+
## Files
|
| 26 |
+
|
| 27 |
+
| File | Size |
|
| 28 |
+
|---|---|
|
| 29 |
+
| `A_BLSP_seed1234.pt` | 336 MB |
|
| 30 |
+
| `A_BLSP_seed1235.pt` | 336 MB |
|
| 31 |
+
| `A_BLSP_seed1236.pt` | 336 MB |
|
| 32 |
+
| `A_BLSP_seed1237.pt` | 336 MB |
|
| 33 |
+
| `A_BLSP_seed1238.pt` | 336 MB |
|
| 34 |
+
| `adapter_init.json` | calibrated init (std_8B=0.02205, RMSNorm scale=0.0215) |
|
| 35 |
+
|
| 36 |
+
## Reported metrics (5-seed cohort)
|
| 37 |
+
|
| 38 |
+
| Metric | Mean ± σ |
|
| 39 |
+
|---|---|
|
| 40 |
+
| Probe-G total | 0.6306 ± 0.0095 |
|
| 41 |
+
| **Probe-G$_{\mathrm{neutral}}$** | **0.5122 ± 0.0108** |
|
| 42 |
+
| Probe-G$_{\mathrm{explicit}}$ | 0.7491 ± 0.0120 |
|
| 43 |
+
| Probe-K linear eval_full | 0.2105 ± 0.0128 |
|
| 44 |
+
| Probe-K MLP-2 eval_full | 0.2446 ± 0.0339 |
|
| 45 |
+
|
| 46 |
+
**Sub-cascade collapse:** every seed's Probe-K linear is below the
|
| 47 |
+
$K_T$ text-only baseline of 0.290.
|
| 48 |
+
|
| 49 |
+
## Usage
|
| 50 |
+
|
| 51 |
+
```python
|
| 52 |
+
import torch
|
| 53 |
+
from huggingface_hub import hf_hub_download
|
| 54 |
+
|
| 55 |
+
ckpt = hf_hub_download("nur-dev/frozen-stress-r0-blsp", "A_BLSP_seed1234.pt")
|
| 56 |
+
state = torch.load(ckpt, map_location="cpu")
|
| 57 |
+
# Adapter loading — see scripts/stage2_eval.py in the GitHub repo
|
| 58 |
+
```
|
| 59 |
+
|
| 60 |
+
## License
|
| 61 |
+
|
| 62 |
+
CC-BY-NC-4.0 (weights inherit non-commercial restriction from the
|
| 63 |
+
training datasets Stress-17K-raw, StressPresso, Expresso). Permitted
|
| 64 |
+
for academic research and ablation studies.
|
adapter_init.json
ADDED
|
@@ -0,0 +1,28 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"qwen3_8b": {
|
| 3 |
+
"model_id": "Qwen/Qwen3-8B",
|
| 4 |
+
"d_llm": 4096,
|
| 5 |
+
"embed_tokens_std": 0.022050147876143456,
|
| 6 |
+
"embed_row_norm_mean": 1.3758326768875122
|
| 7 |
+
},
|
| 8 |
+
"adapter_init": {
|
| 9 |
+
"d_enc": 1024,
|
| 10 |
+
"d_llm": 4096,
|
| 11 |
+
"conv_kernel": 4,
|
| 12 |
+
"conv_stride": 4,
|
| 13 |
+
"mlp_hidden_mult": 2,
|
| 14 |
+
"last_linear_std": 0.0003445335605647415,
|
| 15 |
+
"rmsnorm_init_scale": 0.021490697426189264,
|
| 16 |
+
"modality_token_std": 0.022050147876143456
|
| 17 |
+
},
|
| 18 |
+
"smoke": {
|
| 19 |
+
"median_ratio": 1.4409220218658447,
|
| 20 |
+
"p10_ratio": 1.4286571741104126,
|
| 21 |
+
"p90_ratio": 1.4487676620483398,
|
| 22 |
+
"target_band": [
|
| 23 |
+
0.5,
|
| 24 |
+
2.0
|
| 25 |
+
],
|
| 26 |
+
"n_trainable_params": 83910656
|
| 27 |
+
}
|
| 28 |
+
}
|