File size: 7,729 Bytes
309cbe2
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
5cdabc9
 
4b828bc
 
 
0dd3b45
4b828bc
 
 
5cdabc9
 
309cbe2
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
---
language:
- ru
- en
- de
- pl
- it
- fr
- es
- bn
license: mit
library_name: transformers
pipeline_tag: audio-classification
base_model: Aniemore/unispeech-sat-emotion-russian-resd
base_model_relation: finetune
datasets:
- Aniemore/resd
- Aniemore/resd_annotated
- amu-cai/CAMEO
tags:
- audio-classification
- emotion-recognition
- speech-emotion-recognition
- speech
- multilingual
- russian
- quantized
- compressed-tensors
- int8
- fp8
- int4
metrics:
- f1
- accuracy
- recall
model-index:
- name: unispeech-sat-emotion-v1-crosslingual
  results:
  - task:
      name: Speech Emotion Recognition
      type: audio-classification
    dataset:
      name: RESD test
      type: Aniemore/resd
    metrics:
    - name: Macro F1
      type: f1
      value: 0.5697
    - name: Unweighted accuracy
      type: recall
      value: 0.6158
  - task:
      name: Speech Emotion Recognition
      type: audio-classification
    dataset:
      name: Dusha podcast test
      type: dusha
    metrics:
    - name: Macro F1
      type: f1
      value: 0.3705
    - name: Unweighted accuracy
      type: recall
      value: 0.6992
  - task:
      name: Speech Emotion Recognition
      type: audio-classification
    dataset:
      name: CAMEO test
      type: amu-cai/CAMEO
    metrics:
    - name: Macro F1
      type: f1
      value: 0.5689
    - name: Unweighted accuracy
      type: recall
      value: 0.5706
---

<img src="assets/banner.svg" alt="unispeech-sat-emotion-v1-crosslingual" width="100%">

# unispeech-sat-emotion-v1-crosslingual

Speech emotion recognition over seven classes &mdash; `anger`, `disgust`, `enthusiasm`, `fear`, `happiness`, `neutral`, `sadness`.

Same architecture as [`Aniemore/unispeech-sat-emotion-russian-resd`](https://huggingface.co/Aniemore/unispeech-sat-emotion-russian-resd), retrained on a mix of 27,939 clips spanning eight languages and three speaking registers instead of one acted Russian corpus. The point of the change is spontaneous speech: the previous release was trained only on acted dialogue, where every class is equally frequent and every utterance is performed, and real speech is neither.

Quantized builds ship in the same repository under `int8/`, `fp8/` and `int4/`.

## Results

| test set | what it is | macro-F1 | UA | WA | previous release |
|---|---|---:|---:|---:|---:|
| RESD test | acted Russian, 7 balanced classes | **0.5697** | 0.6158 | 0.6179 | 0.7027 |
| Dusha podcast test | spontaneous Russian, majority neutral | **0.3705** | 0.6992 | 0.6993 | 0.1000 |
| CAMEO test | 7 non-Russian languages | **0.5689** | 0.5706 | 0.6832 | 0.1946 |

<img src="assets/panel.svg" alt="Panel results" width="760">

Read the first two rows together. The acted score goes down and the spontaneous score goes up; both follow from the same change, and which one matters is a deployment question. If your audio is read or performed speech, the previous release may still suit you better.

<details><summary>About the CAMEO row</summary>

CAMEO ships no train/test partition, and the usual way to make one &mdash; a random split over clips &mdash; puts nearly every test speaker into training as well: three of its twelve constituent corpora contain a single speaker each, so no clip-level split of them can be speaker-disjoint even in principle. The number above is reported for completeness. Treat it as an in-domain figure, not as evidence of cross-lingual transfer.

</details>

<details>
<summary><b>Per class, across the panel</b> &mdash; recall and F1 for every class on every test set</summary>

<img src="assets/classes.svg" alt="Per-class recall and F1" width="100%">

Each card leads with the macro-F1 for that set; the rows are the detail behind it. Both per-class numbers are shown because they disagree in a way that matters: recall rewards a class the model over-predicts, so on the spontaneous set the minority classes reach decent recall at poor F1 &mdash; most clips called `sad` there are not sad. If you are going to act on one class, read its F1.

Each set keeps its own class list: the spontaneous corpus has four classes and the other two have seven, and there is no correspondence between `positive` and any single one of `happiness`/`enthusiasm` to line them up with.

</details>

### Per-class recall on spontaneous speech

| class | this model | previous release |
|---|---:|---:|
| `angry` | 0.5689 | 0.5629 |
| `neutral` | 0.6958 | 0.0583 |
| `positive` | 0.8138 | 0.5029 |
| `sad` | 0.7184 | 0.2718 |

`neutral` carries most of real speech and is the class the previous release missed.

## Quantized variants

| subfolder | scheme | weights | vs fp32 | macro-F1 | UA | WA |
|---|---|---:|---:|---:|---:|---:|
| _(root)_ | fp32 | 1206 MiB | 1.0x | 0.5697 | 0.6158 | 0.6179 |
| `int8` | W8A16 | 351 MiB | 3.4x smaller | 0.5697 | 0.6158 | 0.6179 |
| `fp8` | W8A16-float | 343 MiB | 3.5x smaller | 0.5663 | 0.6126 | 0.6143 |
| `int4` | W4A16_ASYM | 209 MiB | 5.8x smaller | 0.5714 | 0.6163 | 0.6179 |

<img src="assets/quality.svg" alt="Quality after quantization" width="760">

<img src="assets/size.svg" alt="Weights on disk" width="760">

Weight-only, round-to-nearest, no calibration. Every variant lands within the seed spread of the fp32 parent on RESD test, so the choice is about download size rather than about quality.

## Training data

| corpus | clips | language | register |
|---|---:|---|---|
| RESD | 948 | Russian | acted dialogue |
| Dusha crowd | 6,800 | Russian | acted, crowd-sourced |
| CAMEO | 6,800 | 7 languages | 12 corpora, no Russian |
| Dusha podcast | 6,060 | Russian | spontaneous podcast speech |
| IEMOCAP | 4,735 | English | elicited dyadic sessions |
| ASVP-ESD | 2,596 | multilingual | mixed register |
| **total** | **27,939** | 8 languages | 3 registers |

A slice is held out of every corpus in the mix, in the same proportion, and model selection is on that held-out split &mdash; never on any of the test sets above. Labels are unified to seven classes; four-class corpora are mapped upward and scored on the classes they actually contain.

## Usage

```python
import torch, librosa
from transformers import AutoModelForAudioClassification, AutoFeatureExtractor

repo = "Aniemore/unispeech-sat-emotion-v1-crosslingual"
model = AutoModelForAudioClassification.from_pretrained(repo).eval()
fe = AutoFeatureExtractor.from_pretrained(repo)

# Resample to 16 kHz. Do not skip it: RESD itself ships at 44.1 kHz,
# and handing the model 44.1 kHz audio while telling the extractor it
# is 16 kHz stretches time 2.8x and silently changes the answer.
wav, _ = librosa.load("clip.wav", sr=16000, mono=True)
x = fe(wav, sampling_rate=16000, return_tensors="pt", padding=True)
with torch.no_grad():
    probs = model(**x).logits.softmax(-1)[0]
print({model.config.id2label[i]: round(p.item(), 3) for i, p in enumerate(probs)})
```

For a quantized build, name the subfolder &mdash; only that subfolder is downloaded:

```python
model = AutoModelForAudioClassification.from_pretrained(
    repo, subfolder="int8").eval()          # or "fp8", "int4"
fe = AutoFeatureExtractor.from_pretrained(repo, subfolder="int8")
```

## Limitations

- Scores are the mean of two seeds; the seed spread on the 280-clip RESD split is ±0.03&ndash;0.05, so differences smaller than that are not differences.
- The spontaneous set has four classes where the model has seven, so its numbers are computed over a mapped label space and are not comparable to seven-class figures.
- Spontaneous scores are at zero decision bias. Calibrating the `neutral` threshold on your own development split will move them.
- Inherited from [`microsoft/unispeech-sat-large`](https://huggingface.co/microsoft/unispeech-sat-large); the licence follows the base model.