| { | |
| "model_type": "BADClassifier", | |
| "input_dim": 2048, | |
| "output_dim": 2, | |
| "hidden_size": 2048, | |
| "target_layer": 15, | |
| "num_layers": 22, | |
| "base_model_name": "TinyLlama/TinyLlama-1.1B-Chat-v1.0", | |
| "label_mapping": { | |
| "biased": 0, | |
| "unbiased": 1 | |
| }, | |
| "dataset_info": { | |
| "bbq_samples": 58942, | |
| "mmlu_samples": 20266, | |
| "bbq_source": "bitlabsdb/BBQ_dataset", | |
| "mmlu_source": "cais/mmlu" | |
| }, | |
| "layer_selection": { | |
| "candidates": [ | |
| 11, | |
| 12, | |
| 13, | |
| 14, | |
| 15, | |
| 16 | |
| ], | |
| "selected": 15, | |
| "accuracies": { | |
| "11": 0.8152321577072144, | |
| "12": 0.8395493626594543, | |
| "13": 0.8271411657333374, | |
| "14": 0.8292092084884644, | |
| "15": 0.8414747714996338, | |
| "16": 0.8393354415893555 | |
| } | |
| }, | |
| "performance": { | |
| "best_epoch": 50 | |
| }, | |
| "trained_date": "2025-11-04 13:03:45" | |
| } |