Hugging Face's logo

GY2233
/
R2R_router_qwen3-4b

Text Classification
r2r
router
efficiency
language-model
Model card Files
xet
 Community
R2R_router_qwen3-4b
File size: 2,417 Bytes
22584f6
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
{
  "model": {
    "model_type": "HiddenStatesTokenLMHeadLogitsClassifier",
    "init_args": {
      "hidden_states_size": 2560,
      "logits_size": 100,
      "hidden_dims": [
        1024,
        1024,
        1024,
        1024,
        1024,
        1024
      ],
      "expansion_factor": 4,
      "dropout_rate": 0.1,
      "use_position_embedding": false,
      "freeze_lm_head": true,
      "normalize_input": false,
      "pretrained_model_name": "/share/public/public_models/Qwen3-4B"
    },
    "model_specific_args": {},
    "input_type": [
      "hidden_states",
      "token",
      "logits"
    ],
    "output_type": "binary"
  },
  "data": {
    "train": {
      "path": [
        "local:output_qwen3_4b/query_dataset_train/LLM_response/SLM_prefill/LLM_continuation_verify/divergent_label_dataset"
      ],
      "type": "divergent",
      "input_prefix": "small_"
    },
    "test": {
      "path": [
        "local:output_qwen3_4b/query_dataset_validation/LLM_response/SLM_prefill/LLM_continuation_verify/divergent_label_dataset"
      ],
      "type": "divergent",
      "input_prefix": "small_"
    }
  },
  "training": {
    "optimizer": {
      "lr": 5e-05,
      "weight_decay": 0.0005
    },
    "params": {
      "num_epochs": 50,
      "batch_size": 1024,
      "patience": 10,
      "device": "cuda"
    },
    "loss": {
      "type": "BCEWithLogitsLoss",
      "recall_factor": 1.0
    },
    "validation": {
      "valid_freq": 2
    },
    "dtype": "float32"
  },
  "optimizing": {
    "type": "threshold",
    "min_recall": 0.95
  },
  "output": {
    "output_dir": "resource/default_router_qwen3_4b.pt",
    "checkpoint_dir": "output/checkpoint_qwen3_4b",
    "model_name": null
  },
  "result": {
    "model_path": "resource/default_router_qwen3_4b.pt/classifier_20250722_162811.pt",
    "results": {
      "threshold": 0.44555555555555554,
      "best_epoch": 10,
      "best_val_loss": 0.5960861395445372,
      "final_metrics": {
        "accuracy": 0.7977646101727843,
        "precision": 0.15479111283151836,
        "recall": 0.9502874416816649,
        "f1": 0.2662182701980336,
        "positive_rate": 0.2370020493434411
      },
      "pre_opt_metrics": {
        "accuracy": 0.5716250266165094,
        "precision": 0.5454684069285192,
        "recall": 0.9376539866151776,
        "f1": 0.689707914270107,
        "positive_rate": 0.8728085740648733
      }
    }
  }
}