MADHAV-SAMDANI commited on
Commit
5170c6f
·
verified ·
1 Parent(s): 980225b

Upload README.md with huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +66 -0
README.md ADDED
@@ -0,0 +1,66 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: mit
3
+ tags:
4
+ - keyword-spotting
5
+ - speech
6
+ - pytorch
7
+ - onnx
8
+ - bc-resnet
9
+ language:
10
+ - en
11
+ datasets:
12
+ - google/speech_commands
13
+ metrics:
14
+ - accuracy
15
+ ---
16
+
17
+ # SpeechGuard KWS — BC-ResNet-8 Keyword Spotter
18
+
19
+ Part of the SpeechGuard AI system submitted to Samsung EnnovateX AX Hackathon 2026.
20
+
21
+ ## Model Description
22
+
23
+ BC-ResNet-8 keyword spotter trained on Google Speech Commands v2 with noise augmentation.
24
+ Uses PCEN (Per-Channel Energy Normalization) frontend for robust noise handling.
25
+
26
+ ## Performance
27
+
28
+ | Metric | Value |
29
+ |---|---|
30
+ | TA Clean | 99.0% |
31
+ | TA Noisy (-5 to +30 dB) | 98.5% |
32
+ | Parameters | 2,444 |
33
+ | Latency (CPU) | 1.1ms |
34
+
35
+ ## Usage
36
+
37
+ ```python
38
+ import torch
39
+ from huggingface_hub import hf_hub_download
40
+
41
+ # Download checkpoint
42
+ ckpt_path = hf_hub_download(
43
+ repo_id="MADHAV-SAMDANI/speechguard-kws",
44
+ filename="best_kws.pt"
45
+ )
46
+
47
+ # Load model
48
+ from speechguard.kws.bc_resnet import BCResNet8
49
+ ckpt = torch.load(ckpt_path, map_location="cpu", weights_only=False)
50
+ model = BCResNet8(num_classes=len(ckpt["classes"]), n_mels=80)
51
+ model.load_state_dict(ckpt["model_state"])
52
+ model.eval()
53
+ ```
54
+
55
+ ## Training
56
+
57
+ - Dataset: Google Speech Commands v2 (2000 samples/class)
58
+ - Epochs: 35
59
+ - Optimizer: AdamW with cosine LR annealing
60
+ - Noise augmentation: ESC-50 + synthetic (white, pink, babble)
61
+ - Hardware: MacBook Air CPU (~70 minutes)
62
+
63
+ ## Citation
64
+
65
+ Samsung EnnovateX AX Hackathon 2026 — Problem #04
66
+ Team: Placecomm Prophets (IIT Kharagpur)